Update app.py

app.py

@@ -52,7 +52,7 @@ def translate_srt(input_file, source_language_code, target_language_code):
     return translated_file
 
 st.title("SRT Translator")
-st.write("Translate subtitles from one language to another.")
+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
 url = "https://huggingface.co/Lenylvt/LanguageISO/resolve/main/iso.md"
@@ -61,10 +61,10 @@ language_options = fetch_languages(url)
 source_language_code, target_language_code = None, None
 
 if language_options:
-    source_language_code = st.selectbox("Select Source Language", options=language_options, format_func=lambda x: x[1])[0]
-    target_language_code = st.selectbox("Select Target Language", options=language_options, format_func=lambda x: x[1])[0]
+    source_language_code = st.selectbox("1️⃣ Select Source Language", options=language_options, format_func=lambda x: x[1])[0]
+    target_language_code = st.selectbox("2️⃣ Select Target Language", options=language_options, format_func=lambda x: x[1])[0]
 
-file_input = st.file_uploader("Upload SRT File", type=["srt"])
+file_input = st.file_uploader("📁 Upload SRT File", type=["srt"], accept_multiple_files=False)
 
 if file_input is not None and source_language_code and target_language_code:
     with tempfile.NamedTemporaryFile(delete=False, suffix=".srt") as temp_file:
@@ -73,15 +73,15 @@ if file_input is not None and source_language_code and 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",
+        label="⬇️ Download Translated SRT",
         data=translated_srt_bytes,
         file_name="translated_subtitles.srt",
         mime="text/plain",
-    )
+    )