Update app.py

app.py

@@ -18,9 +18,13 @@ def generate_subtitles(video_path, target_language):
     with open("temp_audio.wav", "rb") as audio_file:
         transcription = asr(audio_file)["text"]
 
-    # Translate transcription to the target language
+    # Translate transcription to the target language using M2M100
+    # The forced_bos_token_id is set based on the target language
     translation_pipeline = pipeline('translation', model='facebook/m2m100_418M')
-    translated_subtitles = translation_pipeline(transcription, forced_bos_token_id=translation_pipeline.tokenizer.get_lang_id(target_language))[0]["translation_text"]
+    translated_subtitles = translation_pipeline(
+        transcription, 
+        forced_bos_token_id=translation_pipeline.tokenizer.get_lang_id(target_language)
+    )[0]["translation_text"]
 
     # Return subtitles (text for now)
     subtitles = f"Original: {transcription}\nTranslated: {translated_subtitles}"
@@ -36,7 +40,20 @@ interface = gr.Interface(
     fn=subtitle_video,
     inputs=[
         gr.Video(label="Upload Video"),
-        gr.Textbox(label="Target Language Code (e.g., 'fr' for French, 'es' for Spanish)"),
+        gr.Dropdown(  # Dropdown for language selection
+            label="Choose Target Language",
+            choices=[
+                "fa",  # Persian
+                "fr",  # French
+                "es",  # Spanish
+                "de",  # German
+                "zh",  # Chinese
+                "ar",  # Arabic
+                "hi",  # Hindi
+                "ru"   # Russian
+            ],
+            value="fa",  # Default to Persian
+        ),
     ],
     outputs="text",
     title="Automatic Video Subtitler & Translator"