Update app.py

app.py

@@ -9,162 +9,162 @@ from datetime import datetime
 static_ffmpeg.add_paths()
 
 def stitch_media(video_file, audio_file, subtitle_file, crf_quality=23):
-"""
-Stitch video, audio, and subtitle files together using ffmpeg
-All files are handled through Gradio's temporary file system
-"""
-try:
-    # Validate inputs
-    if not video_file or not audio_file or not subtitle_file:
-        return None, "❌ Please upload all three files (video, audio, subtitle)"
-    
-    # Gradio automatically provides temporary file paths
-    video_path = video_file.name if hasattr(video_file, 'name') else video_file
-    audio_path = audio_file.name if hasattr(audio_file, 'name') else audio_file
-    subtitle_path = subtitle_file.name if hasattr(subtitle_file, 'name') else subtitle_file
-    
-    # Create output file in temp directory
-    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-    output_path = os.path.join(tempfile.gettempdir(), f"stitched_{timestamp}.mp4")
-    
-    # FFmpeg command
-    cmd = [
-        "ffmpeg",
-        "-i", video_path,
-        "-i", audio_path,
-        "-vf", f"subtitles={subtitle_path}",
-        "-map", "0:v",
-        "-map", "1:a",
-        "-c:v", "libx264",
-        "-crf", str(crf_quality),
-        "-c:a", "aac",
-        "-y",
-        output_path
-    ]
-    
-    # Execute ffmpeg
-    process = subprocess.run(
-        cmd, 
-        check=True, 
-        capture_output=True, 
-        text=True
-    )
-    
-    # Check if output file was created
-    if os.path.exists(output_path):
-        file_size = os.path.getsize(output_path) / (1024 * 1024)  # Size in MB
-        return output_path, f"✅ Video stitched successfully! (Size: {file_size:.2f} MB)"
-    else:
-        return None, "❌ Output file was not created"
-    
-except subprocess.CalledProcessError as e:
-    error_msg = f"❌ FFmpeg error:\n{e.stderr[-500:]}"  # Show last 500 chars of error
-    return None, error_msg
-except Exception as e:
-    return None, f"❌ Error: {str(e)}"
-
-# Create Gradio interface
-with gr.Blocks(title="Video Audio Subtitle Stitcher", theme=gr.themes.Soft()) as app:
-gr.Markdown(
     """
-    # 🎬 Video Audio Subtitle Stitcher
-    
-    **Upload your files and stitch them together - No local installation needed!**
-    
-    This app works entirely in your browser. Just upload, click stitch, and download your result.
+    Stitch video, audio, and subtitle files together using ffmpeg
+    All files are handled through Gradio's temporary file system
     """
-)
-
-with gr.Row():
-    with gr.Column():
-        gr.Markdown("### 📤 Upload Files")
+    try:
+        # Validate inputs
+        if not video_file or not audio_file or not subtitle_file:
+            return None, "❌ Please upload all three files (video, audio, subtitle)"
         
-        video_input = gr.File(
-            label="📹 Video File",
-            file_types=[".mp4", ".mov", ".avi", ".mkv"],
-            type="filepath"
-        )
-        audio_input = gr.File(
-            label="🎵 Audio File",
-            file_types=[".wav", ".mp3", ".aac", ".m4a"],
-            type="filepath"
-        )
-        subtitle_input = gr.File(
-            label="📝 Subtitle File (.srt)",
-            file_types=[".srt"],
-            type="filepath"
-        )
+        # Gradio automatically provides temporary file paths
+        video_path = video_file.name if hasattr(video_file, 'name') else video_file
+        audio_path = audio_file.name if hasattr(audio_file, 'name') else audio_file
+        subtitle_path = subtitle_file.name if hasattr(subtitle_file, 'name') else subtitle_file
+        
+        # Create output file in temp directory
+        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+        output_path = os.path.join(tempfile.gettempdir(), f"stitched_{timestamp}.mp4")
         
-        gr.Markdown("### ⚙️ Settings")
+        # FFmpeg command
+        cmd = [
+            "ffmpeg",
+            "-i", video_path,
+            "-i", audio_path,
+            "-vf", f"subtitles={subtitle_path}",
+            "-map", "0:v",
+            "-map", "1:a",
+            "-c:v", "libx264",
+            "-crf", str(crf_quality),
+            "-c:a", "aac",
+            "-y",
+            output_path
+        ]
         
-        crf_input = gr.Slider(
-            minimum=18,
-            maximum=28,
-            value=23,
-            step=1,
-            label="Video Quality (CRF)",
-            info="Lower = Better quality but larger file"
+        # Execute ffmpeg
+        process = subprocess.run(
+            cmd, 
+            check=True, 
+            capture_output=True, 
+            text=True
         )
         
-        stitch_btn = gr.Button("🎬 Stitch Video", variant="primary", size="lg")
+        # Check if output file was created
+        if os.path.exists(output_path):
+            file_size = os.path.getsize(output_path) / (1024 * 1024)  # Size in MB
+            return output_path, f"✅ Video stitched successfully! (Size: {file_size:.2f} MB)"
+        else:
+            return None, "❌ Output file was not created"
         
-        gr.Markdown(
-            """
-            **💡 Tips:**
-            - CRF 18-20: Excellent quality
-            - CRF 23: Balanced (recommended)
-            - CRF 26-28: Smaller file size
-            """
-        )
-    
-    with gr.Column():
-        gr.Markdown("### 📥 Output")
+    except subprocess.CalledProcessError as e:
+        error_msg = f"❌ FFmpeg error:\n{e.stderr[-500:]}"  # Show last 500 chars of error
+        return None, error_msg
+    except Exception as e:
+        return None, f"❌ Error: {str(e)}"
+
+# Create Gradio interface
+with gr.Blocks(title="Video Audio Subtitle Stitcher", theme=gr.themes.Soft()) as app:
+    gr.Markdown(
+        """
+        # 🎬 Video Audio Subtitle Stitcher
         
-        status_output = gr.Textbox(
-            label="Status",
-            placeholder="Upload files and click 'Stitch Video' to begin...",
-            lines=3
-        )
-        video_output = gr.Video(
-            label="Result",
-            autoplay=False
-        )
+        **Upload your files and stitch them together - No local installation needed!**
         
-        gr.Markdown(
-            """
-            **Download:** Right-click the video preview and "Save video as..."
-            or use the download button in the video player.
-            """
-        )
+        This app works entirely in your browser. Just upload, click stitch, and download your result.
+        """
+    )
 
-gr.Markdown(
-    """
-    ---
-    ### 📖 How it works:
-    1. **Upload** your video, audio, and subtitle files
-    2. **Adjust** quality settings if needed
-    3. **Click** "Stitch Video" button
-    4. **Wait** for processing (may take a few moments)
-    5. **Preview** and download your result!
-    
-    ### 🎯 Technical Details:
-    - Video codec: H.264 (libx264)
-    - Audio codec: AAC
-    - Output length: Matches shortest stream (video or audio)
-    - Subtitles: Burned into video
-    
-    ### ☁️ Cloud Deployment Ready:
-    This app can be deployed to Hugging Face Spaces, Google Colab, or any cloud platform!
-    """
-)
+    with gr.Row():
+        with gr.Column():
+            gr.Markdown("### 📤 Upload Files")
+            
+            video_input = gr.File(
+                label="📹 Video File",
+                file_types=[".mp4", ".mov", ".avi", ".mkv"],
+                type="filepath"
+            )
+            audio_input = gr.File(
+                label="🎵 Audio File",
+                file_types=[".wav", ".mp3", ".aac", ".m4a"],
+                type="filepath"
+            )
+            subtitle_input = gr.File(
+                label="📝 Subtitle File (.srt)",
+                file_types=[".srt"],
+                type="filepath"
+            )
+            
+            gr.Markdown("### ⚙️ Settings")
+            
+            crf_input = gr.Slider(
+                minimum=18,
+                maximum=28,
+                value=23,
+                step=1,
+                label="Video Quality (CRF)",
+                info="Lower = Better quality but larger file"
+            )
+            
+            stitch_btn = gr.Button("🎬 Stitch Video", variant="primary", size="lg")
+            
+            gr.Markdown(
+                """
+                **💡 Tips:**
+                - CRF 18-20: Excellent quality
+                - CRF 23: Balanced (recommended)
+                - CRF 26-28: Smaller file size
+                """
+            )
+        
+        with gr.Column():
+            gr.Markdown("### 📥 Output")
+            
+            status_output = gr.Textbox(
+                label="Status",
+                placeholder="Upload files and click 'Stitch Video' to begin...",
+                lines=3
+            )
+            video_output = gr.Video(
+                label="Result",
+                autoplay=False
+            )
+            
+            gr.Markdown(
+                """
+                **Download:** Right-click the video preview and "Save video as..."
+                or use the download button in the video player.
+                """
+            )
+
+    gr.Markdown(
+        """
+        ---
+        ### 📖 How it works:
+        1. **Upload** your video, audio, and subtitle files
+        2. **Adjust** quality settings if needed
+        3. **Click** "Stitch Video" button
+        4. **Wait** for processing (may take a few moments)
+        5. **Preview** and download your result!
+        
+        ### 🎯 Technical Details:
+        - Video codec: H.264 (libx264)
+        - Audio codec: AAC
+        - Output length: Matches shortest stream (video or audio)
+        - Subtitles: Burned into video
+        
+        ### ☁️ Cloud Deployment Ready:
+        This app can be deployed to Hugging Face Spaces, Google Colab, or any cloud platform!
+        """
+    )
 
-# Connect the button to the function
-stitch_btn.click(
-    fn=stitch_media,
-    inputs=[video_input, audio_input, subtitle_input, crf_input],
-    outputs=[video_output, status_output]
-)
+    # Connect the button to the function
+    stitch_btn.click(
+        fn=stitch_media,
+        inputs=[video_input, audio_input, subtitle_input, crf_input],
+        outputs=[video_output, status_output]
+    )
 
 # Launch the app
 if __name__ == "__main__":
-app.launch()
+    app.launch()