Update app.py

app.py

@@ -20,14 +20,14 @@ subtitle_storage = {}
 
 # 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"
+    "Persian": "fa",
+    "French": "fr",
+    "Spanish": "es",
+    "German": "de",
+    "Chinese": "zh",
+    "Arabic": "ar",
+    "Hindi": "hi",
+    "Russian": "ru"
 }
 
 def transcribe_audio(chunk):
@@ -66,22 +66,23 @@ def add_subtitle(video):
     except Exception as e:
         return f"Error in adding subtitle: {e}"
 
-def translate_subtitle(video):
+def translate_subtitle(video, language):
     try:
         # Translate the stored subtitle
         original_subtitle = subtitle_storage.get("original")
         if not original_subtitle:
             return "No subtitle to translate!"
         
+        # Translate using the selected language
         translated_subtitle = translator(
             original_subtitle, 
             src_lang="en",  # Source language (assuming the subtitle is in English)
-            tgt_lang=languages["Persian (fa)"]  # Set to the target language, here Persian
+            tgt_lang=languages[language]  # Get the language code from the dropdown selection
         )[0]["translation_text"]
 
         subtitle_storage["translated"] = translated_subtitle  # Store the translated subtitle
         
-        return "Subtitle translated successfully!"
+        return f"Subtitle translated to {language} successfully!"
     
     except Exception as e:
         return f"Error in translating subtitle: {e}"
@@ -120,8 +121,8 @@ def download_video():
         # Function to generate subtitle text
         generator = lambda txt: TextClip(txt, font='Arial', fontsize=24, color='white')
 
-        # Simulated subtitle time intervals for simplicity
-        subs = [(i * 5, i * 5 + 5, translated_subtitle[i:i+50]) for i in range(0, len(translated_subtitle), 50)]
+        # Simulated subtitle time intervals for simplicity (assumes 5 seconds per subtitle)
+        subs = [(i * 5, translated_subtitle[i:i+50]) for i in range(0, len(translated_subtitle), 50)]
 
         # Create subtitle clips
         subtitles = SubtitlesClip(subs, generator)
@@ -159,10 +160,11 @@ with gr.Blocks() as demo:
     
     # Translate Subtitle
     with gr.Row():
+        language_dropdown = gr.Dropdown(choices=list(languages.keys()), label="Choose Target Language", value="Persian")
         translate_button = gr.Button("Translate Subtitle")
         translate_status = gr.Textbox(label="Translation Status")
     
-    translate_button.click(translate_subtitle, inputs=video_input, outputs=translate_status)
+    translate_button.click(translate_subtitle, inputs=[video_input, language_dropdown], outputs=translate_status)
     
     # Download as Word
     with gr.Row():