Update app.py

app.py

@@ -40,16 +40,19 @@ def translate_srt(input_file, source_language_code, target_language_code, progre
     # Initialize an empty list to store translated subtitles
     translated_subs = []
 
-    # Set total steps for progress bar
-    total_steps = len(subs)
-
     # Translate each subtitle
     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}")
+        # Construct the translated subtitle with timestamp and line number
+        translated_sub = pysrt.SubRipItem(index=idx+1, start=sub.start, end=sub.end, text=translated_text)
+        translated_subs.append(translated_sub)
+        progress((idx + 1) / len(subs), desc=f"Translating subtitle {idx+1}/{len(subs)}")
 
-    return "\n".join(translated_subs)
+    # Save translated subtitles to a new SRT file
+    translated_file = pysrt.SubRipFile(translated_subs)
+    translated_srt_path = input_file.name.replace(".srt", f"_{target_language_code}.srt")
+    translated_file.save(translated_srt_path)
+    return translated_srt_path
 
 source_language_dropdown = gr.Dropdown(choices=language_options, label="Source Language")
 target_language_dropdown = gr.Dropdown(choices=language_options, label="Target Language")
@@ -58,7 +61,7 @@ file_input = gr.File(label="Upload SRT File")
 iface = gr.Interface(
     fn=translate_srt,
     inputs=[file_input, source_language_dropdown, target_language_dropdown],
-    outputs=gr.Textbox(label="Translated SRT"),
+    outputs=gr.File(label="Translated SRT"),
     title="SRT Translator",
     description="Translate subtitles from one language to another."
 )