Update app.py

app.py

@@ -3,42 +3,32 @@ import subprocess
 import os
 from moviepy.editor import VideoFileClip
 import whisper
-import tempfile
 
+# Load the smaller model for speed
 model = whisper.load_model("small")
 
-def process_video(video_file):
+def process_video(video_path):
     try:
-        # Handle Gradio file input
-        if isinstance(video_file, str) and video_file.startswith("http"):
-            return None, "❌ Direct URLs not supported in Space API. Please upload file."
-
-        # Save to temp
-        tmp_video = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
-        tmp_video.write(video_file.read())
-        tmp_video.close()
-
         # Extract audio
-        video = VideoFileClip(tmp_video.name)
-        audio_path = tmp_video.name.replace(".mp4", ".wav")
+        video = VideoFileClip(video_path)
+        audio_path = "audio.wav"
         video.audio.write_audiofile(audio_path, codec="pcm_s16le")
 
         # Transcribe
         result = model.transcribe(audio_path, language="en")
         transcript = result["text"]
 
-        # Write .srt file
-        srt_path = tmp_video.name.replace(".mp4", ".srt")
-        with open(srt_path, "w", encoding="utf-8") as f:
+        # Save transcript to file
+        with open("transcript.srt", "w", encoding="utf-8") as f:
             for i, seg in enumerate(result["segments"]):
                 start = seg["start"]
                 end = seg["end"]
                 text = seg["text"].strip()
                 f.write(f"{i+1}\n{format_time(start)} --> {format_time(end)}\n{text}\n\n")
 
-        # Burn subtitles
-        output_path = tmp_video.name.replace(".mp4", "_subtitled.mp4")
-        burn_subtitles(tmp_video.name, srt_path, output_path)
+        # Burn subtitles using ffmpeg
+        output_path = "output_with_subs.mp4"
+        burn_subtitles(video_path, "transcript.srt", output_path)
 
         return output_path, transcript
 
@@ -61,10 +51,14 @@ def burn_subtitles(video_path, srt_path, output_path):
     ]
     subprocess.run(cmd, check=True)
 
+# Gradio UI
 iface = gr.Interface(
     fn=process_video,
-    inputs=gr.Video(label="Upload Video"),
-    outputs=[gr.Video(label="Video with Subtitles"), gr.Textbox(label="Transcript")],
+    inputs=gr.Video(label="Upload your video"),
+    outputs=[
+        gr.Video(label="Video with Subtitles"),
+        gr.Textbox(label="Transcript", lines=8)
+    ],
     title="🎬 Whisper Subtitle Burner",
     description="Uploads a video, generates English subtitles using Whisper, and burns them into the video."
 )