Update app.py

app.py

@@ -1,57 +1,53 @@
-import gradio as gr
 from gradio_client import Client
+import gradio as gr
 
-def process_video(video_url, target_language, translate_video):
-    try:
-        client = Client("https://artificialguybr-video-translation-transcription.hf.space/")
-        result = client.predict(
-            video_url,
-            target_language,
-            translate_video,
-            api_name="/predict"
-        )
-        return result
-    except Exception as e:
-        print(f"Error al llamar a la API: {e}")
-        return "Ocurrió un error al procesar el vídeo."
 
-# Componentes de Gradio
-video_url = gr.Textbox(label="URL del Vídeo", placeholder="Inserte la URL del vídeo aquí")
-dropdown = gr.Dropdown(
-    choices=[
-        "Afrikáans", "Albanés", "Amárico", "Árabe", "Armenio", "Azerbaiyano", "Vasco", 
-        "Bielorruso", "Bengalí", "Bosnio", "Búlgaro", "Catalán", "Cebuano", "Chichewa", 
-        "Chino (Simplificado)", "Chino (Tradicional)", "Corso", "Croata", "Checo", 
-        "Danés", "Holandés", "Inglés", "Esperanto", "Estonio", "Filipino", "Finlandés", 
-        "Francés", "Frisón", "Gallego", "Georgiano", "Alemán", "Griego", "Gujarati", 
-        "Criollo haitiano", "Hausa", "Hawaiano", "Hebreo", "Hindi", "Hmong", "Húngaro", 
-        "Islandés", "Igbo", "Indonesio", "Irlandés", "Italiano", "Japonés", "Javanés", 
-        "Canarés", "Kazajo", "Jemer", "Coreano", "Kurdo (Kurmanji)", "Kirguís", "Lao", 
-        "Latín", "Letón", "Lituano", "Luxemburgués", "Macedonio", "Malagasy", 
-        "Malayo", "Malayalam", "Maltés", "Maorí", "Maratí", "Mongol", "Myanmar (Birmano)", 
-        "Nepalí", "Noruego", "Odia", "Pastún", "Persa", "Polaco", "Portugués", "Panyabí", 
-        "Rumano", "Ruso", "Samoano", "Gaélico escocés", "Serbio", "Sesotho", "Shona", 
-        "Sindhi", "Cingalés", "Eslovaco", "Esloveno", "Somalí", "Español", "Sundanés", 
-        "Suajili", "Sueco", "Tayiko", "Tamil", "Telugu", "Tailandés", "Turco", "Ucraniano", 
-        "Urdu", "Uigur", "Uzbeko", "Vietnamita", "Galés", "Xhosa", "Yidis", "Yoruba", "Zulú"
-    ],
-    label="Idioma de Destino para Traducción"
-)
-checkbox = gr.Checkbox(label="Traducir Vídeo", value=True)
+def interact_with_api(video_url, target_language, translate_video):
+    client = Client("https://artificialguybr-video-translation-transcription.hf.space/")
+    result = client.predict(
+        video_url,
+        target_language,
+        translate_video,
+        api_name="/predict"
+    )
+    return result
+
+# Lista de opções de idiomas
+language_options = [
+    "Afrikaans", "Albanian", "Amharic", "Arabic", "Armenian", "Azerbaijani", "Basque", "Belarusian", 
+    "Bengali", "Bosnian", "Bulgarian", "Catalan", "Cebuano", "Chichewa", "Chinese (Simplified)", 
+    "Chinese (Traditional)", "Corsican", "Croatian", "Czech", "Danish", "Dutch", "English", "Esperanto", 
+    "Estonian", "Filipino", "Finnish", "French", "Frisian", "Galician", "Georgian", "German", "Greek", 
+    "Gujarati", "Haitian Creole", "Hausa", "Hawaiian", "Hebrew", "Hindi", "Hmong", "Hungarian", "Icelandic", 
+    "Igbo", "Indonesian", "Irish", "Italian", "Japanese", "Javanese", "Kannada", "Kazakh", "Khmer", 
+    "Korean", "Kurdish (Kurmanji)", "Kyrgyz", "Lao", "Latin", "Latvian", "Lithuanian", "Luxembourgish", 
+    "Macedonian", "Malagasy", "Malay", "Malayalam", "Maltese", "Maori", "Marathi", "Mongolian", 
+    "Myanmar (Burmese)", "Nepali", "Norwegian", "Odia", "Pashto", "Persian", "Polish", "Portuguese", 
+    "Punjabi", "Romanian", "Russian", "Samoan", "Scots Gaelic", "Serbian", "Sesotho", "Shona", "Sindhi", 
+    "Sinhala", "Slovak", "Slovenian", "Somali", "Spanish", "Sundanese", "Swahili", "Swedish", "Tajik", 
+    "Tamil", "Telugu", "Thai", "Turkish", "Ukrainian", "Urdu", "Uyghur", "Uzbek", "Vietnamese", "Welsh", 
+    "Xhosa", "Yiddish", "Yoruba", "Zulu"
+]
 
-# Interfaz de Gradio
 iface = gr.Interface(
-    fn=process_video,
-    inputs=[video_url, dropdown, checkbox],
-    outputs=gr.Video(),
-    live=False,
-    title="Traducción y Transcripción de Vídeo con IA",
-    description="Inserta la URL del vídeo y selecciona el idioma de destino para la traducción. Esta herramienta utiliza la API de traducción y transcripción de vídeo.",
-    gr.Markdown("""
-    **Nota:**
-    - El límite de vídeo es de 15 minutos. Realizará la transcripción y traducción de subtítulos.
-    - La herramienta utiliza modelos de código abierto para todos los modelos. Es una versión alfa.
-    """),
+    fn=interact_with_api,
+    inputs=[
+        gr.Video(label="Upload Video or Enter URL"),
+        gr.Dropdown(choices=language_options, label="Target Language for Translation", value="English"),
+        gr.Checkbox(label="Translate Video", value=True, info="Check to translate the video to the selected language. Uncheck for transcription only."),
+    ],
+    outputs="video",
+    title="VIDEO TRANSCRIPTION AND TRANSLATION",
+    description="""This tool was developed by [@artificialguybr](https://twitter.com/artificialguybr) using entirely open-source tools. Special thanks to Hugging Face for the GPU support. Test the [Video Dubbing](https://huggingface.co/spaces/artificialguybr/video-dubbing) space!""",
+    allow_flagging=False
 )
 
-iface.launch()
+with gr.Blocks() as demo:
+    iface.render()
+    gr.Markdown("""
+    **Note:**
+    - Video limit is 15 minutes. It will do the transcription and translate of subtitles.
+    - The tool uses open-source models for all models. It's an alpha version.
+    """)
+demo.queue(concurrency_count=1, max_size=15)
+demo.launch()