Upload app.py

app.py

@@ -13,32 +13,68 @@ def extract_last_frame(video_file):
         # 創建固定名稱的輸出檔案
         temp_output = tempfile.mktemp(suffix='.jpg')
         
-        # 使用 ffmpeg 獲取影片總幀數並提取最後一幀
-        # 先獲取影片時長
-        duration_cmd = [
+        # 獲取影片總幀數
+        frame_count_cmd = [
             'ffprobe',
             '-v', 'error',
-            '-show_entries', 'format=duration',
-            '-of', 'default=noprint_wrappers=1:nokey=1',
+            '-select_streams', 'v:0',
+            '-count_frames',
+            '-show_entries', 'stream=nb_read_frames',
+            '-of', 'default=nokey=1:noprint_wrappers=1',
             video_file
         ]
         
-        duration_result = subprocess.run(duration_cmd, capture_output=True, text=True)
-        duration = float(duration_result.stdout.strip())
+        frame_result = subprocess.run(frame_count_cmd, capture_output=True, text=True)
         
-        # 提取最後一幀 (從最後0.1秒處提取)
-        extract_cmd = [
-            'ffmpeg',
-            '-sseof', '-0.1',
-            '-i', video_file,
-            '-update', '1',
-            '-frames:v', '1',
-            '-q:v', '2',
-            temp_output,
-            '-y'
-        ]
-        
-        subprocess.run(extract_cmd, check=True, capture_output=True)
+        # 如果無法獲取幀數，使用反向讀取方法
+        if not frame_result.stdout.strip() or frame_result.returncode != 0:
+            # 方法2: 使用 select 過濾器選擇最後一幀
+            extract_cmd = [
+                'ffmpeg',
+                '-i', video_file,
+                '-vf', 'select=eq(n\\,0)',
+                '-vsync', '0',
+                '-q:v', '2',
+                '-frames:v', '1',
+                temp_output,
+                '-y'
+            ]
+            
+            # 先嘗試從尾部讀取
+            extract_cmd_reverse = [
+                'ffmpeg',
+                '-sseof', '-3',
+                '-i', video_file,
+                '-vframes', '1',
+                '-vf', 'select=eq(pict_type\\,I)',
+                '-vsync', '0',
+                '-q:v', '2',
+                temp_output,
+                '-y'
+            ]
+            
+            result = subprocess.run(extract_cmd_reverse, capture_output=True, text=True)
+            
+            # 如果失敗，使用標準方法
+            if result.returncode != 0 or not os.path.exists(temp_output):
+                subprocess.run(extract_cmd, capture_output=True, text=True)
+        else:
+            # 方法1: 使用幀數選擇最後一幀
+            total_frames = int(frame_result.stdout.strip())
+            last_frame = total_frames - 1
+            
+            extract_cmd = [
+                'ffmpeg',
+                '-i', video_file,
+                '-vf', f'select=eq(n\\,{last_frame})',
+                '-vsync', '0',
+                '-q:v', '2',
+                '-frames:v', '1',
+                temp_output,
+                '-y'
+            ]
+            
+            subprocess.run(extract_cmd, check=True, capture_output=True)
         
         if os.path.exists(temp_output) and os.path.getsize(temp_output) > 0:
             # 重新命名為 lastframe.jpg
@@ -73,8 +109,8 @@ def merge_videos_wrapper(video_files):
         return None, message
 
 # 創建 Gradio 界面
-with gr.Blocks(title="Grok影片工具箱") as demo:
-    gr.Markdown("# Grok影片工具箱")
+with gr.Blocks(title="影片工具箱") as demo:
+    gr.Markdown("# 影片工具箱")
     gr.Markdown("提供影片最後一幀提取和多段影片合併功能")
     
     with gr.Tabs():