Create app.py

app.py

@@ -0,0 +1,143 @@
+import gradio as gr
+import moviepy.editor as mp
+from faster_whisper import WhisperModel
+from googletrans import Translator
+import srt
+import datetime
+import tempfile
+import os
+
+# ------------------------------
+# Load Whisper (small model)
+# ------------------------------
+model = WhisperModel("small", device="cpu", compute_type="float32")
+translator = Translator()
+
+
+# ------------------------------
+# Generate SRT subtitles
+# ------------------------------
+def generate_subtitles(video_file, target_lang):
+    # Extract audio
+    video = mp.VideoFileClip(video_file)
+    temp_audio = "temp_audio.wav"
+    video.audio.write_audiofile(temp_audio, verbose=False, logger=None)
+
+    # Transcribe with Whisper
+    segments, info = model.transcribe(temp_audio)
+
+    # Build SRT
+    subtitles = []
+    for i, seg in enumerate(segments):
+        start = datetime.timedelta(seconds=seg.start)
+        end = datetime.timedelta(seconds=seg.end)
+        original_text = seg.text.strip()
+
+        # Translate text
+        if target_lang != "original":
+            translated_text = translator.translate(original_text, dest=target_lang).text
+        else:
+            translated_text = original_text
+
+        subtitles.append(
+            srt.Subtitle(
+                index=i + 1,
+                start=start,
+                end=end,
+                content=translated_text
+            )
+        )
+
+    # Create SRT file
+    srt_data = srt.compose(subtitles)
+    srt_path = "output_subtitles.srt"
+    with open(srt_path, "w", encoding="utf-8") as srt_file:
+        srt_file.write(srt_data)
+
+    return srt_path
+
+
+# ------------------------------
+# Burn subtitles onto video
+# ------------------------------
+def burn_subtitles(video_file, srt_file):
+    output_path = "video_with_subtitles.mp4"
+
+    video = mp.VideoFileClip(video_file)
+
+    # Read SRT file
+    with open(srt_file, "r", encoding="utf-8") as f:
+        srt_content = f.read()
+
+    subs = list(srt.parse(srt_content))
+
+    subtitle_clips = []
+    for sub in subs:
+        txt = mp.TextClip(
+            sub.content,
+            fontsize=30,
+            color='white',
+            stroke_color='black',
+            stroke_width=2
+        )
+        txt = txt.set_start(sub.start.total_seconds()) \
+                 .set_end(sub.end.total_seconds()) \
+                 .set_position(('center', 'bottom'))
+
+        subtitle_clips.append(txt)
+
+    final = mp.CompositeVideoClip([video] + subtitle_clips)
+    final.write_videofile(output_path, codec="libx264", audio_codec="aac")
+
+    return output_path
+
+
+# ------------------------------
+# Main function (Gradio pipeline)
+# ------------------------------
+def process(video, language, burn):
+    if video is None:
+        return None, None
+
+    # Create subtitles
+    srt_path = generate_subtitles(video, language)
+
+    # Burn subtitles if selected
+    if burn:
+        video_out = burn_subtitles(video, srt_path)
+        return srt_path, video_out
+
+    return srt_path, None
+
+
+# ------------------------------
+# Gradio Interface UI
+# ------------------------------
+lang_options = {
+    "original": "Original (No Translation)",
+    "en": "English",
+    "ur": "Urdu",
+    "hi": "Hindi",
+    "ps": "Pashto",
+    "ar": "Arabic",
+}
+
+ui = gr.Interface(
+    fn=process,
+    inputs=[
+        gr.Video(label="Upload Video"),
+        gr.Dropdown(choices=list(lang_options.keys()), label="Translate To", value="original"),
+        gr.Checkbox(label="Burn Subtitles on Video", value=False)
+    ],
+    outputs=[
+        gr.File(label="Download SRT File"),
+        gr.Video(label="Video with Subtitles")
+    ],
+    title="AI Video Subtitle Generator",
+    description="Upload any video → Auto-transcribe + translate → Generate subtitles → Optional subtitle-burned video."
+)
+
+# ------------------------------
+# Launch App
+# ------------------------------
+ui.launch()