Update app.py

app.py

@@ -1,3 +1,8 @@
+import warnings
+
+# 過濾 SyntaxWarning 警告
+warnings.filterwarnings("ignore", category=SyntaxWarning)
+
 import gradio as gr
 from moviepy.editor import VideoFileClip
 import cv2
@@ -5,9 +10,8 @@ import base64
 from openai import OpenAI
 import os
 
-def process_video(video_path, seconds_per_frame=2):
+def extract_frames(video_path, seconds_per_frame):
     base64Frames = []
-    base_video_path, _ = os.path.splitext(video_path)
     video = cv2.VideoCapture(video_path)
     total_frames = int(video.get(cv2.CAP_PROP_FRAME_COUNT))
     fps = video.get(cv2.CAP_PROP_FPS)
@@ -23,21 +27,37 @@ def process_video(video_path, seconds_per_frame=2):
         base64Frames.append(base64.b64encode(buffer).decode("utf-8"))
         curr_frame += frames_to_skip
     video.release()
+    return base64Frames
 
+def extract_audio(video_path):
+    base_video_path, _ = os.path.splitext(video_path)
     audio_path = f"{base_video_path}.mp3"
     clip = VideoFileClip(video_path)
-    clip.audio.write_audiofile(audio_path, bitrate="32k")
-    clip.audio.close()
+    clip.audio.write_audiofile(filename=audio_path, codec="mp3", bitrate="32k")
+    # Remove the line that closes the audio
     clip.close()
+    return audio_path
 
+def process_video(video_path, seconds_per_frame=2):
+    base64Frames = extract_frames(video_path, seconds_per_frame)
+    audio_path = extract_audio(video_path)
     return base64Frames, audio_path
 
+def get_summary_prompt(summary_detail):
+    if summary_detail == "簡短":
+        return "生成一個簡短的摘要，不超過50字。"
+    elif summary_detail == "中等":
+        return "生成一個中等長度的摘要，概述影片的主要內容。"
+    else:  # 詳細
+        return "生成一個詳細的摘要，包括影片的主要內容和關鍵細節。"
+
 def summarize_video(file_path, summary_detail):
-    api_key = os.getenv("OPENAI_API_KEY")
+    api_key = "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"
+    # api_key = os.getenv("OPENAI_API_KEY")
     client = OpenAI(api_key=api_key)
 
     # 抽取幀和音頻（每0.5秒一幀）
-    base64Frames, audio_path = process_video(file_path, seconds_per_frame=0.5)
+    base64Frames, audio_path = process_video(file_path, seconds_per_frame=int(0.5))
 
     # 使用Whisper進行音頻轉錄
     transcription = client.audio.transcriptions.create(
@@ -45,12 +65,7 @@ def summarize_video(file_path, summary_detail):
     )
 
     # 根據用戶選擇的詳細程度設置提示
-    if summary_detail == "簡短":
-        summary_prompt = "生成一個簡短的摘要，不超過50字。"
-    elif summary_detail == "中等":
-        summary_prompt = "生成一個中等長度的摘要，概述影片的主要內容。"
-    else:  # 詳細
-        summary_prompt = "生成一個詳細的摘要，包括影片的主要內容和關鍵細節。"
+    summary_prompt = get_summary_prompt(summary_detail)
 
     # 使用GPT-4o生成摘要
     response = client.chat.completions.create(
@@ -95,7 +110,8 @@ demo = gr.Interface(
     outputs="markdown",
     title="影片摘要生成器",
     description="上傳影片，將生成影片的摘要。",
+    flagging_dir=""  # 這會移除 Flag 功能
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()