Update app.py

app.py

@@ -5,7 +5,6 @@ import base64
 from openai import OpenAI
 import os
 
-# 參考: https://cookbook.openai.com/examples/gpt4o/introduction_to_gpt4o
 def process_video(video_path, seconds_per_frame=2):
     base64Frames = []
     base_video_path, _ = os.path.splitext(video_path)
@@ -33,8 +32,7 @@ def process_video(video_path, seconds_per_frame=2):
 
     return base64Frames, audio_path
 
-
-def summarize_video(file_path):
+def summarize_video(file_path, summary_detail):
     api_key = os.getenv("OPENAI_API_KEY")
     client = OpenAI(api_key=api_key)
 
@@ -46,13 +44,21 @@ def summarize_video(file_path):
         model="whisper-1", file=open(audio_path, "rb")
     )
 
+    # 根據用戶選擇的詳細程度設置提示
+    if summary_detail == "簡短":
+        summary_prompt = "生成一個簡短的摘要，不超過50字。"
+    elif summary_detail == "中等":
+        summary_prompt = "生成一個中等長度的摘要，概述影片的主要內容。"
+    else:  # 詳細
+        summary_prompt = "生成一個詳細的摘要，包括影片的主要內容和關鍵細節。"
+
     # 使用GPT-4o生成摘要
     response = client.chat.completions.create(
         model="gpt-4o",
         messages=[
             {
                 "role": "system",
-                "content": """您是一名優秀的摘要專家，請根據提供的影片和其轉錄內容生成Markdown格式的摘要。""",
+                "content": f"""您是一名優秀的摘要專家，請根據提供的影片和其轉錄內容生成Markdown格式的摘要。{summary_prompt}""",
             },
             {
                 "role": "user",
@@ -82,7 +88,10 @@ def summarize_video(file_path):
 
 demo = gr.Interface(
     fn=summarize_video,
-    inputs=[gr.File(label="上傳影片 (mp4)")],
+    inputs=[
+        gr.File(label="上傳影片 (mp4)"),
+        gr.Radio(["簡短", "中等", "詳細"], label="摘要詳細程度", value="中等")
+    ],
     outputs="markdown",
     title="影片摘要生成器",
     description="上傳影片，將生成影片的摘要。",