Update app.py

app.py

@@ -39,7 +39,8 @@ 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
+    progress_text = "Operation in progress. For information, the progress bar start when the translation begin."
+    progress_bar = st.progress(0, text=progress_text)  # Initialize the progress bar
 
     for idx, sub in enumerate(subs):
         translated_text = translate_text(sub.text, source_language_code, target_language_code)
@@ -51,7 +52,7 @@ def translate_srt(input_file, source_language_code, target_language_code):
     progress_bar.empty()  # Optionally clear the progress bar after completion
     return translated_file
 
-st.title("SRT Translator")
+st.title("SRT Translation")
 st.write("We use model from [Language Technology Research Group at the University of Helsinki](https://huggingface.co/Helsinki-NLP). For API use please visit [this space](https://huggingface.co/spaces/Lenylvt/SRT_Translation-API)")
 
 # Fetch language options
@@ -71,13 +72,13 @@ if file_input is not None and source_language_code and target_language_code:
         temp_file.write(file_input.getvalue())
         temp_file_path = temp_file.name
 
-    translated_srt_bytes = translate_srt(temp_file_path, source_language_code, target_language_code)
+    translated_srt = translate_srt(temp_file_path, source_language_code, target_language_code)
     
-    #buffer = BytesIO()
-    #translated_srt.save(buffer, encoding='utf-8')
-    #buffer.seek(0)
-    #
-    #translated_srt_bytes = buffer.getvalue()
+    buffer = BytesIO()
+    translated_srt.save(buffer, encoding='utf-8')
+    buffer.seek(0)
+    
+    translated_srt_bytes = buffer.getvalue()
     
     st.download_button(
         label="⬇️ Download Translated SRT",