Update app.py

app.py

@@ -1,29 +1,48 @@
 import gradio as gr
 from moviepy.editor import VideoFileClip
 import os
+from googletrans import Translator
 
 import whisper
 model = whisper.load_model("base")
 
-def generate_subtitles(video_path):
+TRANSLATION_LANGUAGES = {
+    "Urdu": "ur",
+    "French": "fr",
+    "Spanish": "es",
+    "German": "de",
+    "Chinese": "zh",
+    "Arabic": "ar",
+    "Hindi": "hi"
+}
+
+translator = Translator()
+
+def generate_translated_subtitles(video_path, target_language):
     # Extract audio from video
     video = VideoFileClip(video_path)
     audio_path = "temp_audio.wav"
     video.audio.write_audiofile(audio_path)
     
-    # Transcribe audio using Whisper
-    result = model.transcribe(audio_path)
+    # Transcribe (but do NOT translate) using Whisper
+    result = model.transcribe(audio_path, language="en")
     
     # Clean up temporary audio file
     os.remove(audio_path)
     
-    # Format the subtitles in .srt format
+    # Extract all subtitle texts for batch translation
+    texts = [segment['text'] for segment in result['segments']]
+    
+    # Perform batch translation (all at once)
+    translations = translator.translate(texts, dest=TRANSLATION_LANGUAGES[target_language])
+    
+    # Format subtitles in .srt format
     srt_content = ""
-    for index, segment in enumerate(result['segments']):
+    for index, (segment, translation) in enumerate(zip(result['segments'], translations)):
         start_time = segment['start']
         end_time = segment['end']
-        text = segment['text']
-        
+        translated_text = translation.text
+
         # Convert seconds to SRT time format (HH:MM:SS,mmm)
         def format_time(seconds):
             hours = int(seconds // 3600)
@@ -34,29 +53,32 @@ def generate_subtitles(video_path):
         
         srt_content += f"{index + 1}\n"
         srt_content += f"{format_time(start_time)} --> {format_time(end_time)}\n"
-        srt_content += f"{text}\n\n"
+        srt_content += f"{translated_text}\n\n"
     
     # Save the .srt file
-    srt_filename = "subtitles.srt"
+    srt_filename = f"subtitles_{TRANSLATION_LANGUAGES.get(target_language, 'en')}.srt"
     with open(srt_filename, "w", encoding="utf-8") as srt_file:
         srt_file.write(srt_content)
     
     return srt_content, srt_filename
 
 
-def video_to_subtitles(video):
-    subtitles, srt_filename = generate_subtitles(video)
-    return subtitles, srt_filename  # Just return the file path directly
+def video_to_translated_subtitles(video, target_language):
+    subtitles, srt_filename = generate_translated_subtitles(video, target_language)
+    return subtitles, srt_filename  # Return translated subtitles and file path
 
 iface = gr.Interface(
-    fn=video_to_subtitles,
-    inputs=gr.Video(label="Upload Video"),
+    fn=video_to_translated_subtitles,
+    inputs=[
+        gr.Video(label="Upload English Video"),
+        gr.Dropdown(choices=list(TRANSLATION_LANGUAGES.keys()), label="Translate to", value="Urdu")
+    ],
     outputs=[
-        gr.Textbox(label="Subtitles", lines=10),
-        gr.DownloadButton(label="Download .srt File")  # No file_name needed
+        gr.Textbox(label="Translated Subtitles", lines=10),
+        gr.File(label="Download Translated .srt File")  # Correct file download
     ],
-    title="Video to Subtitles",
-    description="Upload a video to generate subtitles using Whisper AI and download the .srt file."
+    title="Video to Translated Subtitles",
+    description="Upload an English video, select a language, and get translated subtitles."
 )
 
 iface.launch(share=True)