Update app.py

app.py

@@ -1,7 +1,11 @@
 import gradio as gr
 from langchain_community.document_loaders import YoutubeLoader
 from langchain.text_splitter import TokenTextSplitter
+from groq import Groq
 
+client = Groq(
+    api_key=os.environ.get("api_key"),
+)
 max_textboxes = 5
 
 def process_youtube_url(url, language):
@@ -22,14 +26,37 @@ def process_youtube_url(url, language):
         
         output_textboxes = [chunk for i, chunk in enumerate(chunks)]
         output_textboxes += ["" for _ in range(max_textboxes - len(chunks))]
+
+        completion = client.chat.completions.create(
+            model="llama3-70b-8192",
+            messages=[
+                {
+                    "role": "system",
+                    "content": "lang:日本語"
+                },
+                {
+                    "role": "user",
+                    "content": f"lang:日本語　解説して\n\n## trascript \n{text}"
+                },
+            ],
+            temperature=0.7,
+            max_tokens=1024,
+            top_p=1,
+            stream=True,
+            stop=None,
+        )
+        summirizedtext = ""
+        for chunk in completion:
+            summirizedtext += chunk.choices[0].delta.content or ""
+            # print(chunk.choices[0].delta.content or "", end="")
         
-        return *output_textboxes, [], text, token_count
+            yield *output_textboxes, [], text, token_count,summirizedtext
     
     except Exception as e:
         error_msg = str(e)
         available_languages = extract_available_languages(error_msg)
         recommended_language = extract_recommended_language(error_msg)
-        return *[error_msg for _ in range(max_textboxes)], available_languages, recommended_language, 0
+        return *[error_msg for _ in range(max_textboxes)], available_languages, recommended_language, 0,""
 
 
 def extract_available_languages(error_msg):
@@ -65,6 +92,7 @@ iface = gr.Interface(
         gr.Dropdown(label="Available Languages", allow_custom_value=True),
         gr.Textbox(label="Recommended Language",show_copy_button=True),
         gr.Number(label="Character Count")
+        gr.Markdown(label='summirized output')
     ],
     examples = [["https://youtu.be/6Af6b_wyiwI?si=zqD9-kjw24lpRJw3","ja"],["https://youtu.be/9kxL9Cf46VM?si=ADgUmDXb6riA-lgb","ja"]],
     title="YouTube Transcript Loader",
@@ -72,4 +100,5 @@ iface = gr.Interface(
 )
 
 if __name__ == "__main__":
+    iface.queue()
     iface.launch(share=True)