Update app.py

app.py

@@ -3,138 +3,168 @@ import subprocess
 import static_ffmpeg
 import os
 import tempfile
+from datetime import datetime
 
 # Add static ffmpeg to PATH
 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
- 
- Args:
-     video_file: Path to video file (.mp4)
-     audio_file: Path to audio file (.wav)
-     subtitle_file: Path to subtitle file (.srt)
-     crf_quality: Video quality (0-51, lower is better, default: 23)
- 
- Returns:
-     Path to the output video file and status message
- """
- 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)"
-     
-     # Create temporary output file
-     output_dir = tempfile.gettempdir()
-     output_path = os.path.join(output_dir, f"stitched_video_{os.getpid()}.mp4")
-     
-     # Build ffmpeg command
-     # ffmpeg -i video.mp4 -i audio.wav -vf "subtitles=audio.srt" -map 0:v -map 1:a -c:v libx264 -crf 23 -c:a aac -y res.mp4
-     cmd = [
-         "ffmpeg",
-         "-i", video_file,
-         "-i", audio_file,
-         "-vf", f"subtitles={subtitle_file}",
-         "-map", "0:v",
-         "-map", "1:a",
-         "-c:v", "libx264",
-         "-crf", str(crf_quality),
-         "-c:a", "aac",
-         "-y",
-         output_path
-     ]
-     
-     # Execute ffmpeg
-     result = subprocess.run(
-         cmd, 
-         check=True, 
-         capture_output=True, 
-         text=True
-     )
-     
-     return output_path, "✅ Video stitched successfully!"
-     
- except subprocess.CalledProcessError as e:
-     error_msg = f"❌ FFmpeg error: {e.stderr}"
-     return None, error_msg
- except Exception as e:
-     return None, f"❌ Error: {str(e)}"
+"""
+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") as app:
- gr.Markdown(
-     """
-     # 🎬 Video Audio Subtitle Stitcher
-     Upload a video, audio, and subtitle file to stitch them together using FFmpeg.
-     
-     **The video length will be determined by the shortest stream (video or audio).**
-     """
- )
- 
- with gr.Row():
-     with gr.Column():
-         video_input = gr.File(
-             label="📹 Video File (.mp4)",
-             file_types=[".mp4", ".mov", ".avi", ".mkv"]
-         )
-         audio_input = gr.File(
-             label="🎵 Audio File (.wav)",
-             file_types=[".wav", ".mp3", ".aac", ".m4a"]
-         )
-         subtitle_input = gr.File(
-             label="📝 Subtitle File (.srt)",
-             file_types=[".srt"]
-         )
-         
-         crf_input = gr.Slider(
-             minimum=18,
-             maximum=28,
-             value=23,
-             step=1,
-             label="🎨 Video Quality (CRF: lower = better quality, larger file)",
-             info="Recommended: 18-23 for high quality, 23-28 for smaller files"
-         )
-         
-         stitch_btn = gr.Button("🎬 Stitch Video", variant="primary", size="lg")
-     
-     with gr.Column():
-         status_output = gr.Textbox(
-             label="Status",
-             placeholder="Status will appear here..."
-         )
-         video_output = gr.Video(
-             label="Output Video",
-             autoplay=False
-         )
- 
- # Info section
- gr.Markdown(
-     """
-     ### 📋 How it works:
-     - **Video**: Takes video stream from this file
-     - **Audio**: Replaces the video's audio with this file
-     - **Subtitle**: Overlays subtitles from SRT file on the video
-     
-     ### ⚙️ Settings:
-     - **CRF (Constant Rate Factor)**: Controls video quality
-       - 18-23: High quality (larger files)
-       - 23-28: Good quality (smaller files)
-       - Default: 23 (balanced)
-     
-     ### 🎯 Output:
-     - Codec: H.264 (libx264) for video, AAC for audio
-     - Length: Determined by shortest stream
-     """
- )
- 
- # 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]
- )
+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.
+    """
+)
+
+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]
+)
 
 # Launch the app
 if __name__ == "__main__":
- app.launch()
+app.launch()