add summary

app.py

@@ -2,13 +2,12 @@ import gradio as gr
 import yt_dlp
 import os
 from openai import OpenAI
+import anthropic
 
-client = OpenAI(api_key=os.environ['OPENAI_API_KEY'])
-
-# from faster_whisper import WhisperModel
-# tiny, tiny.en, base, base.en, small, small.en, medium, medium.en, large-v1, large-v2, large-v3, or large
-# model_name = 'base'
-# model = WhisperModel(model_name, device="cpu", download_root="./models")
+client = OpenAI(
+    api_key=os.getenv("OPENAI_API_KEY"),
+)
+claude_client = anthropic.Anthropic()
 
 ydl_opts = {
     'outtmpl': 'demo.m4a',
@@ -27,6 +26,7 @@ def download_audio(url):
         code = ydl.download([url])
     assert code == 0, "Failed to download audio"
 
+text = ""
 def generate_text(url):
     download_audio(url)
     with open("demo.m4a", "rb") as f:
@@ -35,20 +35,45 @@ def generate_text(url):
             file=f, 
             response_format="text"
         )
-    return transcription.text
+    global text
+    text = transcription.text
+
+def summarize_text():
+    print(text)
+    prompt = f"Please summarize the following article in 5 sentences or less: [{text}]"
+    message = claude_client.messages.create(
+        model="claude-3-haiku-20240307",
+        max_tokens=1000,
+        temperature=0.0,
+        # system="Respond only in mandarin",
+        messages=[
+            {"role": "user", "content": prompt}
+        ]
+    )
+    return message.content[0].text
 
 with gr.Blocks() as demo:
     with gr.Column():
         name = gr.Textbox(label="Enter your youtube url")
-        button = gr.Button("Download")
+        button_download = gr.Button("Download")
 
-    with gr.Column():
-        output = gr.TextArea(label="Output")
+    with gr.Row():
+        output = gr.TextArea(label="Output")        
 
-    button.click(
+        with gr.Column():
+            button_summary = gr.Button("Summary")
+            summary = gr.TextArea(label="Summary")
+
+    button_download.click(
         generate_text,
         inputs=[name],
         outputs=[output],
     )
 
+    button_summary.click(
+        summarize_text,
+        inputs=[],
+        outputs=[summary],
+    )
+
 demo.launch()

requirements.txt

@@ -1,2 +1,3 @@
 yt-dlp
-faster-whisper
+openai
+anthropic