Update app.py

app.py

@@ -8,7 +8,22 @@ asr = pipeline("automatic-speech-recognition", model="openai/whisper-large")
 # MarianMT or M2M100 for translation (multi-language)
 translator = pipeline("translation", model="facebook/m2m100_418M")
 
-def generate_subtitles(video_path, target_language):
+# Supported languages with their codes
+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"
+}
+
+def generate_subtitles(video_path, language_name):
+    # Extract the target language code from the selected language name
+    target_language = languages[language_name]
+    
     # Extract audio from video
     video = mp.VideoFileClip(video_path)
     audio = video.audio
@@ -30,21 +45,9 @@ def generate_subtitles(video_path, target_language):
     return subtitles
 
 # Define Gradio interface
-def subtitle_video(video_file, target_language):
+def subtitle_video(video_file, language_name):
     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"
-}
+    return generate_subtitles(video_path, language_name)
 
 # Gradio app layout
 interface = gr.Interface(