Update app.py

app.py

@@ -3,13 +3,13 @@ import pandas as pd
 import pysrt
 from transformers import MarianMTModel, MarianTokenizer
 import tempfile
+import os
 from io import BytesIO
 import requests
 
 def fetch_languages(url):
     response = requests.get(url)
     if response.status_code == 200:
-        csv_content = response.content.decode('utf-8')
         df = pd.read_csv(BytesIO(response.content), delimiter="|", skiprows=2, header=None).dropna(axis=1, how='all')
         df.columns = ['ISO 639-1', 'ISO 639-2', 'Language Name', 'Native Name']
         df['ISO 639-1'] = df['ISO 639-1'].str.strip()
@@ -39,17 +39,17 @@ def translate_srt(input_file, source_language_code, target_language_code):
     subs = pysrt.open(input_file)
     total_subs = len(subs)
     translated_subs = []
-    progress_text = "Operation in progress. For information, the progress bar start when the translation begin."
-    progress_bar = st.progress(0)  # Initialize the progress bar
-
+    progress_text = "Operation in progress. Please wait."
+    st.progress(0, text=progress_text)  # Initialize the progress bar placeholder
+    
     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
+        progress_percent = int(((idx + 1) / total_subs) * 100)
+        st.progress(progress_percent)  # Update progress bar
 
     translated_file = pysrt.SubRipFile(items=translated_subs)
-    progress_bar.empty()  # Optionally clear the progress bar after completion
     return translated_file
 
 st.title("SRT Translation")
@@ -70,19 +70,20 @@ file_input = st.file_uploader("📁 Upload SRT File", type=["srt"], accept_multi
 if file_input is not None and source_language_code and target_language_code:
     with tempfile.NamedTemporaryFile(delete=False, suffix=".srt") as temp_file:
         temp_file.write(file_input.getvalue())
-        temp_file_path = temp_file.name
-
-    translated_srt = translate_srt(temp_file_path, source_language_code, target_language_code)
-    
-    # Convert translated SRT to a string, then encode to bytes for download
-    buffer = BytesIO()
-    translated_srt.save(buffer, encoding='utf-8')
-    buffer.seek(0)
-    translated_srt_bytes = buffer.getvalue()
+        temp_file.flush()
+        translated_srt = translate_srt(temp_file.name, source_language_code, target_language_code)
+        os.unlink(temp_file.name)  # Delete the temp file
     
+    # Save the translated subtitles to a temporary file and then read it into BytesIO
+    with tempfile.NamedTemporaryFile(delete=False, suffix=".srt") as temp_file:
+        translated_srt.save(temp_file.name, encoding='utf-8')
+        temp_file.seek(0)
+        translated_srt_bytes = open(temp_file.name, 'rb').read()
+        os.unlink(temp_file.name)  # Delete the temp file after reading
+
     st.download_button(
         label="⬇️ Download Translated SRT",
         data=translated_srt_bytes,
         file_name="translated_subtitles.srt",
         mime="text/plain",
-    )
+    )