summary

app.py

@@ -102,6 +102,45 @@ def generate_transcript(youtube_link):
         print(f"\n生成逐字稿時發生錯誤: {str(e)}")
         raise
 
+def generate_summary(transcript):
+    """Generate a summary from the transcript using Gemini."""
+    try:
+        print("\n開始生成摘要...")
+        model = "gemini-2.0-flash-exp"
+        contents = [
+            types.Content(
+                role="user",
+                parts=[
+                    types.Part.from_text(
+                        f"""請根據以下逐字稿生成重點摘要，以條列方式呈現主要觀點：
+
+{transcript}
+
+請以下列格式輸出：
+# 主要觀點：
+1. [重點1]
+2. [重點2]
+...
+
+# 結論：
+[整體結論]
+"""
+                    )
+                ]
+            )
+        ]
+        
+        response = GENAI_CLIENT.models.generate_content(
+            model=model,
+            contents=contents,
+        )
+        
+        print("摘要生成完成！")
+        return response.text
+    except Exception as e:
+        print(f"\n生成摘要時發生錯誤: {str(e)}")
+        raise
+
 def process_all_files(file_list):
     print("\n=== 開始處理檔案 ===")
     print(f"待處理檔案數量: {len(file_list)}")
@@ -176,12 +215,12 @@ with gr.Blocks() as demo:
             gr.Markdown("### 功能卡片")
             with gr.Tab("摘要生成"):
                 summary_button = gr.Button("生成摘要")
-                summary = gr.Textbox(label="摘要", interactive=False)
+                summary_output = gr.Markdown(label="摘要")
             with gr.Tab("逐字稿"):
                 transcript_display = gr.Textbox(
                     label="YouTube 逐字稿", 
                     interactive=False, 
-                    lines=15,
+                    lines=10,
                     placeholder="處理 YouTube 影片後，逐字稿將顯示在這裡..."
                 )
             with gr.Tab("其他功能"):
@@ -200,6 +239,17 @@ with gr.Blocks() as demo:
 
     history = gr.State([])
     ask_button.click(mock_question_answer, inputs=[question, history], outputs=[chatbot, question])
-    summary_button.click(mock_summary, outputs=summary)
+    summary_button.click(
+        fn=lambda transcript: on_summary_click(transcript),
+        inputs=[transcript_display],
+        outputs=[summary_output]
+    )
+
+    def on_summary_click(transcript):
+        if not transcript:
+            return "請先上傳文件或輸入 YouTube 連結並處理完成後再生成摘要。"
+        
+        summary = generate_summary(transcript)
+        return summary
 
 demo.launch(share=True)