Update app.py

app.py

@@ -19,7 +19,6 @@ def generate_subtitles(video_path, target_language):
         transcription = asr(audio_file)["text"]
 
     # 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, 
@@ -35,6 +34,18 @@ def subtitle_video(video_file, target_language):
     video_path = video_file.name
     return generate_subtitles(video_path, target_language)
 
+# List of supported languages with their codes for M2M100
+languages = {
+    "Persian (fa)": "fa",
+    "French (fr)": "fr",
+    "Spanish (es)": "es",
+    "German (de)": "de",
+    "Chinese (zh)": "zh",
+    "Arabic (ar)": "ar",
+    "Hindi (hi)": "hi",
+    "Russian (ru)": "ru"
+}
+
 # Gradio app layout
 interface = gr.Interface(
     fn=subtitle_video,
@@ -42,18 +53,9 @@ interface = gr.Interface(
         gr.Video(label="Upload Video"),
         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
-        ),
+            choices=list(languages.keys()),  # Display language names in the dropdown
+            value="Persian (fa)"  # Default language
+        )
     ],
     outputs="text",
     title="Automatic Video Subtitler & Translator"