Update app.py

app.py

@@ -37,12 +37,18 @@ def translate_text(text, source_language_code, target_language_code):
 
 def translate_srt(input_file, source_language_code, target_language_code):
     subs = pysrt.open(input_file)
+    total_subs = len(subs)
     translated_subs = []
+    progress_bar = st.progress(0)  # Initialize the progress bar
+
     for idx, sub in enumerate(subs):
         translated_text = translate_text(sub.text, source_language_code, target_language_code)
         translated_sub = pysrt.SubRipItem(index=idx+1, start=sub.start, end=sub.end, text=translated_text)
         translated_subs.append(translated_sub)
+        progress_bar.progress((idx + 1) / total_subs)  # Update progress bar
+
     translated_file = pysrt.SubRipFile(translated_subs)
+    progress_bar.empty()  # Optionally clear the progress bar after completion
     return translated_file
 
 st.title("SRT Translator")
@@ -61,15 +67,12 @@ if language_options:
 file_input = st.file_uploader("Upload SRT File", type=["srt"])
 
 if file_input is not None and source_language_code and target_language_code:
-    # Save the uploaded file to a temporary file
     with tempfile.NamedTemporaryFile(delete=False, suffix=".srt") as temp_file:
         temp_file.write(file_input.getvalue())
         temp_file_path = temp_file.name
 
-    # Now, temp_file_path contains the path to the temporary file which you can pass to translate_srt
     translated_srt = translate_srt(temp_file_path, source_language_code, target_language_code)
     
-    # Proceed with the translation and file download preparation as before
     buffer = BytesIO()
     translated_srt.save(buffer, encoding='utf-8')
     buffer.seek(0)
@@ -81,4 +84,4 @@ if file_input is not None and source_language_code and target_language_code:
         data=translated_srt_bytes,
         file_name="translated_subtitles.srt",
         mime="text/plain",
-    )
+    )