Update app.py

app.py

@@ -4,31 +4,35 @@ import moviepy.editor as mp
 import tempfile
 import os
 
-# Load Whisper (base or small model for performance)
-model = whisper.load_model("small")
+# Load Whisper model (tiny = fast, small = accurate)
+model = whisper.load_model("tiny")
 
-def generate_subtitles(video):
+def generate_subtitles(video_path):
     try:
-        # Save temp video file
-        with tempfile.NamedTemporaryFile(delete=False, suffix=".mp4") as tmp:
-            tmp.write(video.read())
-            tmp_path = tmp.name
+        # Create temporary directory
+        temp_dir = tempfile.mkdtemp()
+        input_path = os.path.join(temp_dir, "input.mp4")
+        os.rename(video_path, input_path)
 
         # Extract audio
-        clip = mp.VideoFileClip(tmp_path)
-        audio_path = tmp_path.replace(".mp4", ".wav")
+        clip = mp.VideoFileClip(input_path)
+        audio_path = os.path.join(temp_dir, "audio.wav")
         clip.audio.write_audiofile(audio_path, logger=None)
 
-        # Transcribe using Whisper
-        result = model.transcribe(audio_path, task="transcribe")
+        # Transcribe audio using Whisper
+        result = model.transcribe(audio_path)
         text = result["text"]
         lang = result["language"]
 
-        # Add green background
+        # Make green background video
         w, h = clip.size
         bg = mp.ColorClip(size=(w, h), color=(0, 255, 0), duration=clip.duration)
+
+        # Combine background + video
         final = mp.CompositeVideoClip([bg, clip.set_position("center")])
-        output_path = tmp_path.replace(".mp4", "_subtitled.mp4")
+
+        # Save final video
+        output_path = os.path.join(temp_dir, "output.mp4")
         final.write_videofile(output_path, codec="libx264", audio_codec="aac", logger=None)
 
         return output_path, lang, text
@@ -36,22 +40,29 @@ def generate_subtitles(video):
     except Exception as e:
         return None, "Error", str(e)
 
-# Gradio UI
-with gr.Blocks(title="Auto Subtitle Generator (VOZO-style)") as app:
-    gr.Markdown("## 🎬 Auto Subtitle Generator (VOZO-style)")
-    gr.Markdown("Upload a short video in Hindi, English, Hinglish, or Indian English — the AI auto-detects the language and adds VOZO-style subtitles with a green screen background.")
+
+# ----------- UI (VOZO Blink Style) -----------
+with gr.Blocks(title="VOZO-style Auto Subtitle Generator") as demo:
+    gr.Markdown(
+        """
+        # 🎬 **Auto Subtitle Generator (VOZO Blink Style)**
+        Upload any short Hindi, English, Hinglish, or Indian English video.<br>
+        This AI auto-detects the language and generates subtitles with a green background overlay.<br>
+        _(Inspired by Blink Captions / VOZO)_
+        """
+    )
 
     with gr.Row():
-        video_input = gr.Video(label="🎥 Upload your video")
-        video_output = gr.Video(label="📺 Subtitled Output")
+        input_video = gr.Video(label="🎥 Upload your video", interactive=True)
+        output_video = gr.Video(label="📺 Subtitled Output")
 
     with gr.Row():
-        detected_lang = gr.Textbox(label="Detected Language")
-        subtitles_text = gr.Textbox(label="Generated Subtitles")
+        detected_lang = gr.Textbox(label="🌐 Detected Language")
+        subtitle_text = gr.Textbox(label="📝 Generated Subtitles")
 
-    run_btn = gr.Button("🚀 Generate")
+    run_btn = gr.Button("🚀 Generate Subtitles", variant="primary")
 
-    run_btn.click(fn=generate_subtitles, inputs=[video_input],
-                  outputs=[video_output, detected_lang, subtitles_text])
+    run_btn.click(fn=generate_subtitles, inputs=[input_video],
+                  outputs=[output_video, detected_lang, subtitle_text])
 
-app.launch()
+demo.launch()