Update app.py

app.py

@@ -2,7 +2,6 @@ import pysrt
 import gradio as gr
 import pandas as pd
 from transformers import MarianMTModel, MarianTokenizer
-from tqdm import tqdm
 
 # Fetch and parse language options from the provided URL
 url = "https://huggingface.co/Lenylvt/LanguageISO/resolve/main/iso.md"
@@ -34,30 +33,34 @@ def translate_text(text, source_language_code, target_language_code):
     
     return translated_text
 
-def translate_srt(input_file, source_language_code, target_language_code):
+def translate_srt(input_file, source_language_code, target_language_code, progress=gr.Progress()):
     # Load SRT file
-    subs = pysrt.open(input_file)
+    subs = pysrt.open(input_file.name)
 
     # Initialize an empty list to store translated subtitles
     translated_subs = []
 
+    # Set total steps for progress bar
+    total_steps = len(subs)
+
     # Translate each subtitle
-    for sub in tqdm(subs, desc="Translating", total=len(subs)):
+    for idx, sub in enumerate(subs):
         translated_text = translate_text(sub.text, source_language_code, target_language_code)
         translated_subs.append(translated_text)
+        progress((idx + 1) / total_steps, desc=f"Translating subtitle {idx+1}/{total_steps}")
 
     return "\n".join(translated_subs)
 
-source_language_dropdown = gr.Dropdown(choices=language_options, label="Source Language")
-target_language_dropdown = gr.Dropdown(choices=language_options, label="Target Language")
-file_input = gr.File(label="Upload SRT File")
+source_language_dropdown = gr.inputs.Dropdown(choices=language_options, label="Source Language")
+target_language_dropdown = gr.inputs.Dropdown(choices=language_options, label="Target Language")
+file_input = gr.inputs.File(label="Upload SRT File", type="text")
 
 iface = gr.Interface(
     fn=translate_srt,
     inputs=[file_input, source_language_dropdown, target_language_dropdown],
-    outputs=gr.Textbox(label="Translated SRT"),
+    outputs=gr.outputs.Textbox(label="Translated SRT"),
     title="SRT Translator",
     description="Translate subtitles from one language to another."
 )
 
-iface.launch()
+iface.launch()