dropdown

app.py

@@ -11,6 +11,7 @@ import os
 #
 # login(token=token, add_to_git_credential=True)
 pipe = pipeline(model="dacavi/whisper-small-es")
+language_codes = ["en", "fr", "de", "it"]
 def translate_text(text, target_language_code):
     translate = boto3.client('translate')
 
@@ -23,30 +24,35 @@ def translate_text(text, target_language_code):
     translated_text = response['TranslatedText']
     return translated_text
 
-def translate_video(video_url):
-    # Download video and extract audio
-    with tempfile.NamedTemporaryFile(suffix=".wav", delete=True) as temp_audio:
-        # os.system(f"yt-dlp -o {temp_audio.name} -x --audio-format wav {video_url}")
-        os.system(f"yt-dlp -o audioSample.wav -x --audio-format wav {video_url}")
+def translate_video(video_url,language):
+    if video_url and language:
+        # Download video and extract audio
+        with tempfile.NamedTemporaryFile(suffix=".wav", delete=True) as temp_audio:
+            # os.system(f"yt-dlp -o {temp_audio.name} -x --audio-format wav {video_url}")
+            os.system(f"yt-dlp -o audioSample.wav -x --audio-format wav {video_url}")
 
-        print("Downloaded audio:", temp_audio.name)
+            print("Downloaded audio:", temp_audio.name)
 
 
-    # Transcribe audio
-        text = pipe("audioSample.wav")["text"]
-        translatedText = translate_text(text,"en")
+        # Transcribe audio
+            text = pipe("audioSample.wav")["text"]
+            translatedText = translate_text(text,language)
 
-    # Clean up temporary files
-        os.remove("audioSample.wav")
+        # Clean up temporary files
+            os.remove("audioSample.wav")
+    else:
+        translatedText = "Please fill in both parameters."
     return translatedText
 
 iface = gr.Interface(
     fn=translate_video,
-    inputs="text",
+    inputs=[gr.Textbox(label="Video URL"),
+            gr.Dropdown(choices=language_codes, label="Select Target Language")
+            ],
     outputs="text",
     live=True,
-    title="Spanish to English video translation",
-    description="Paste the URL of a Spanish video to translate the text to English spoken content.",
+    title="SPANISH VIDEO TRANSLATION",
+    description="Paste the URL of a Spanish video to translate the text to any of the available languages.",
 )
 
 iface.launch()