Update app.py

app.py

@@ -4,25 +4,32 @@ import tempfile
 from moviepy.editor import VideoFileClip, TextClip, CompositeVideoClip
 import os
 
-def generate_subtitles(video, language="hi"):
+def generate_auto_subtitles(video):
     if video is None:
         return "Please upload a video first!"
 
-    # Temporary file for video
+    # Temporary save video
     temp_video = tempfile.NamedTemporaryFile(delete=False, suffix=".mp4")
     temp_video.write(video.read())
     temp_video.close()
 
-    # Load whisper model
+    # Load Whisper model
     model = whisper.load_model("small")
-    result = model.transcribe(temp_video.name, language=language)
 
-    # Process video
+    # Auto language detection + transcription
+    result = model.transcribe(temp_video.name)
+    detected_lang = result.get("language", "unknown").upper()
+
+    # Prepare video
     video_clip = VideoFileClip(temp_video.name)
     subtitle_clips = []
 
+    # Generate subtitle overlays
     for seg in result["segments"]:
-        txt = seg["text"]
+        txt = seg["text"].strip()
+        if not txt:
+            continue
+
         subtitle = TextClip(
             txt,
             fontsize=60,
@@ -35,22 +42,23 @@ def generate_subtitles(video, language="hi"):
         subtitle = subtitle.set_start(seg["start"]).set_duration(seg["end"] - seg["start"]).set_pos(("center", "bottom"))
         subtitle_clips.append(subtitle)
 
+    # Combine subtitles with video
     final = CompositeVideoClip([video_clip, *subtitle_clips])
-    output_path = os.path.join(tempfile.gettempdir(), "output_with_subs.mp4")
-    final.write_videofile(output_path, codec="libx264", audio_codec="aac")
-
-    return output_path
+    output_path = os.path.join(tempfile.gettempdir(), "vozo_auto_subs.mp4")
+    final.write_videofile(output_path, codec="libx264", audio_codec="aac", verbose=False, logger=None)
 
+    return output_path, f"Detected Language: {detected_lang}"
 
+# Gradio UI
 interface = gr.Interface(
-    fn=generate_subtitles,
-    inputs=[
-        gr.Video(label="Upload your video"),
-        gr.Dropdown(choices=["hi", "en"], value="hi", label="Language")
+    fn=generate_auto_subtitles,
+    inputs=gr.Video(label="🎥 Upload your video (Hindi / English / Hinglish)"),
+    outputs=[
+        gr.Video(label="🎬 Video with Auto Subtitles"),
+        gr.Textbox(label="🌐 Detected Language")
     ],
-    outputs=gr.Video(label="Output with Subtitles"),
-    title="🎬 VOZO-style Subtitle Generator (Free)",
-    description="Upload a short Hindi or English video and get VOZO-style subtitles with green background automatically!"
+    title="🎞️ Auto Subtitle Generator (VOZO-style)",
+    description="Upload any short Hindi, English, Hinglish, or Indian English video — this AI auto-detects language and adds VOZO-style subtitles with a green background."
 )
 
 interface.launch()