Upload 3 files

app.py

@@ -2,14 +2,16 @@ import gradio as gr
 import subprocess
 import os
 import tempfile
+import shutil
+from video_merger import merge_videos
 
 def extract_last_frame(video_file):
     if video_file is None:
         return None, "請上傳影片檔案"
     
     try:
-        # 創建臨時輸出檔案
-        output_path = tempfile.mktemp(suffix='.jpg')
+        # 創建固定名稱的輸出檔案
+        temp_output = tempfile.mktemp(suffix='.jpg')
         
         # 使用 ffmpeg 獲取影片總幀數並提取最後一幀
         # 先獲取影片時長
@@ -32,14 +34,19 @@ def extract_last_frame(video_file):
             '-update', '1',
             '-frames:v', '1',
             '-q:v', '2',
-            output_path,
+            temp_output,
             '-y'
         ]
         
         subprocess.run(extract_cmd, check=True, capture_output=True)
         
-        if os.path.exists(output_path) and os.path.getsize(output_path) > 0:
-            return output_path, "成功提取最後一幀！"
+        if os.path.exists(temp_output) and os.path.getsize(temp_output) > 0:
+            # 重新命名為 lastframe.jpg
+            output_dir = tempfile.gettempdir()
+            final_output = os.path.join(output_dir, 'lastframe.jpg')
+            shutil.copy2(temp_output, final_output)
+            os.remove(temp_output)
+            return final_output, "成功提取最後一幀！"
         else:
             return None, "提取失敗，請確認影片格式正確"
             
@@ -48,33 +55,80 @@ def extract_last_frame(video_file):
     except Exception as e:
         return None, f"發生錯誤: {str(e)}"
 
+def merge_videos_wrapper(video_files):
+    """包裝合併函數以處理檔名"""
+    if not video_files:
+        return None, "請上傳影片檔案"
+    
+    temp_output, message = merge_videos(video_files)
+    
+    if temp_output and os.path.exists(temp_output):
+        # 重新命名為 final.mp4
+        output_dir = tempfile.gettempdir()
+        final_output = os.path.join(output_dir, 'final.mp4')
+        shutil.copy2(temp_output, final_output)
+        os.remove(temp_output)
+        return final_output, message
+    else:
+        return None, message
+
 # 創建 Gradio 界面
-with gr.Blocks(title="影片最後一幀提取器") as demo:
-    gr.Markdown("# 影片最後一幀提取器")
-    gr.Markdown("上傳影片（最大 100MB），自動提取最後一幀為 JPEG 圖片")
+with gr.Blocks(title="影片工具箱") as demo:
+    gr.Markdown("# 影片工具箱")
+    gr.Markdown("提供影片最後一幀提取和多段影片合併功能")
     
-    with gr.Row():
-        with gr.Column():
-            video_input = gr.Video(
-                label="上傳影片",
-                max_length=None,
-                height=400
+    with gr.Tabs():
+        # Tab 1: 提取最後一幀
+        with gr.Tab("提取最後一幀"):
+            gr.Markdown("上傳影片（最大 100MB），自動提取最後一幀為 JPEG 圖片")
+            with gr.Row():
+                with gr.Column():
+                    video_input = gr.Video(
+                        label="上傳影片",
+                        max_length=None,
+                        height=400
+                    )
+                    extract_btn = gr.Button("提取最後一幀", variant="primary")
+                
+                with gr.Column():
+                    image_output = gr.Image(
+                        label="最後一幀 (lastframe.jpg)",
+                        type="filepath",
+                        height=400
+                    )
+                    status_output = gr.Textbox(label="狀態", interactive=False)
+            
+            extract_btn.click(
+                fn=extract_last_frame,
+                inputs=[video_input],
+                outputs=[image_output, status_output]
             )
-            extract_btn = gr.Button("提取最後一幀", variant="primary")
         
-        with gr.Column():
-            image_output = gr.Image(
-                label="最後一幀",
-                type="filepath",
-                height=400
+        # Tab 2: 合併影片
+        with gr.Tab("合併影片"):
+            gr.Markdown("上傳多個 MP4 影片，按照上傳順序合併為一個影片")
+            with gr.Row():
+                with gr.Column():
+                    videos_input = gr.File(
+                        label="上傳影片檔案（可多選）",
+                        file_count="multiple",
+                        file_types=["video"],
+                        height=300
+                    )
+                    merge_btn = gr.Button("合併影片", variant="primary")
+                
+                with gr.Column():
+                    merged_video_output = gr.Video(
+                        label="合併後的影片 (final.mp4)",
+                        height=400
+                    )
+                    merge_status_output = gr.Textbox(label="狀態", interactive=False)
+            
+            merge_btn.click(
+                fn=merge_videos_wrapper,
+                inputs=[videos_input],
+                outputs=[merged_video_output, merge_status_output]
             )
-            status_output = gr.Textbox(label="狀態", interactive=False)
-    
-    extract_btn.click(
-        fn=extract_last_frame,
-        inputs=[video_input],
-        outputs=[image_output, status_output]
-    )
 
 if __name__ == "__main__":
     demo.launch()

video_merger.py

@@ -0,0 +1,84 @@
+import subprocess
+import os
+import tempfile
+
+def merge_videos(video_files):
+    """
+    合併多個 MP4 影片檔案
+    
+    Args:
+        video_files: 影片檔案路徑列表
+    
+    Returns:
+        tuple: (輸出檔案路徑, 狀態訊息)
+    """
+    if not video_files or len(video_files) == 0:
+        return None, "請至少上傳一個影片檔案"
+    
+    if len(video_files) == 1:
+        return None, "請上傳至少兩個影片進行合併"
+    
+    try:
+        # 創建臨時檔案列表
+        list_file = tempfile.mktemp(suffix='.txt')
+        output_path = tempfile.mktemp(suffix='.mp4')
+        
+        # 寫入 ffmpeg concat 檔案列表
+        with open(list_file, 'w', encoding='utf-8') as f:
+            for video_file in video_files:
+                # 使用絕對路徑並轉義特殊字元
+                abs_path = os.path.abspath(video_file)
+                f.write(f"file '{abs_path}'\n")
+        
+        # 使用 ffmpeg concat demuxer 合併影片
+        merge_cmd = [
+            'ffmpeg',
+            '-f', 'concat',
+            '-safe', '0',
+            '-i', list_file,
+            '-c', 'copy',
+            output_path,
+            '-y'
+        ]
+        
+        result = subprocess.run(merge_cmd, capture_output=True, text=True)
+        
+        # 清理臨時列表檔案
+        if os.path.exists(list_file):
+            os.remove(list_file)
+        
+        if result.returncode != 0:
+            # 如果直接複製失敗，嘗試重新編碼
+            merge_cmd_reencode = [
+                'ffmpeg',
+                '-f', 'concat',
+                '-safe', '0',
+                '-i', list_file,
+                '-c:v', 'libx264',
+                '-c:a', 'aac',
+                '-preset', 'medium',
+                output_path,
+                '-y'
+            ]
+            
+            # 重新創建列表檔案
+            with open(list_file, 'w', encoding='utf-8') as f:
+                for video_file in video_files:
+                    abs_path = os.path.abspath(video_file)
+                    f.write(f"file '{abs_path}'\n")
+            
+            result = subprocess.run(merge_cmd_reencode, capture_output=True, text=True)
+            
+            if os.path.exists(list_file):
+                os.remove(list_file)
+            
+            if result.returncode != 0:
+                return None, f"合併失敗: {result.stderr}"
+        
+        if os.path.exists(output_path) and os.path.getsize(output_path) > 0:
+            return output_path, f"成功合併 {len(video_files)} 個影片！"
+        else:
+            return None, "合併失敗，請確認影片格式正確"
+            
+    except Exception as e:
+        return None, f"發生錯誤: {str(e)}"