Update app.py

app.py

@@ -1,120 +1,35 @@
 import streamlit as st
-import re
 from googletrans import Translator
-import logging
 
-# Suppress Streamlit warnings
-logging.getLogger('streamlit').setLevel(logging.ERROR)
+def translate_subtitles(input_file):
+    nums = ['0','1', '2', '3', '4', '5', '6', '7', '8', '9']
+    translator = Translator()
+    output_lines = []
 
-# Initialize Translator
-translator = Translator()
+    with open(input_file, "r", encoding="utf-8") as f:
+        sub_lines = f.readlines()
 
-# Custom CSS for Azure-themed UI
-st.markdown(
-    """
-    <style>
-    .stApp {
-        background-color: #15202b;
-        color: white;
-        font-family: Arial, sans-serif;
-    }
-    .stButton > button {
-        background-color: #007fbf;
-        color: white;
-        border-radius: 20px;
-        border: none;
-    }
-    .stTextInput > div > div > input {
-        background-color: #2c3947;
-        color: white;
-        border: 1px solid #007fbf;
-        border-radius: 20px;
-    }
-    .footer {
-        text-align: center;
-        font-size: 14px;
-        margin-top: 20px;
-        color: #999;
-    }
-    </style>
-    """,
-    unsafe_allow_html=True
-)
+    for line_en in sub_lines:
+        if line_en[0] in nums:
+            output_lines.append(line_en)
+        elif line_en.strip() == '':
+            output_lines.append('\n')
+        else:
+            if line_en.strip():
+                line_fa = translator.translate(line_en, src='en', dest='fa')
+                output_lines.append(line_fa.text + '\n')
 
-st.title("SRT Subtitle Translator")
-st.write("Upload your SRT file and choose the target language to translate.")
+    return output_lines
 
-# File Upload
-uploaded_file = st.file_uploader("Choose an SRT file", type="srt")
+st.title("ترجمه زیرنویس با استفاده از Google Translate")
 
-if uploaded_file is not None:
-    content = uploaded_file.read().decode('utf-8')
-    lines = content.split('\n')
-
-    # Extract text without timestamps
-    def clean_and_extract(lines):
-        cleaned_lines = []
-        temp_lines = []
-        for line in lines:
-            if re.match(r'^\d+$', line) or re.match(r'^\d{2}:\d{2}:\d{2},\d{3} --> \d{2}:\d{2}:\d{2},\d{3}$', line):
-                continue
-            if line == '':
-                if temp_lines:
-                    cleaned_text = '\n'.join(temp_lines)
-                    cleaned_lines.append(cleaned_text)
-                temp_lines = []
-            else:
-                temp_lines.append(line)
-        if temp_lines:
-            cleaned_text = '\n'.join(temp_lines)
-            cleaned_lines.append(cleaned_text)
-        return cleaned_lines
+uploaded_file = st.file_uploader("فایل زیرنویس خود را بارگذاری کنید", type=["srt", "txt"])
 
-    texts = clean_and_extract(lines)
-
-    # Choose Target Language
-    target_language = st.text_input("Enter target language code (e.g., 'en', 'fr'):").lower()
+if uploaded_file is not None:
+    output_lines = translate_subtitles(uploaded_file)
+    st.write("ترجمه شده:")
+    for line in output_lines:
+        st.write(line, end='')
     
-    if st.button("Translate"):
-        translated_texts = []
-        for text in texts:
-            try:
-                translated = translator.translate(text, dest=target_language)
-                translated_texts.append(translated.text)
-            except Exception as e:
-                st.error(f"Translation error: {e}")
-
-        # Reformat SRT with translated text
-        new_lines = []
-        text_iter = iter(translated_texts)
-        for line in lines:
-            if re.match(r'^\d+$', line):
-                new_lines.append(line)
-            elif re.match(r'^\d{2}:\d{2}:\d{2},\d{3} --> \d{2}:\d{2}:\d{2},\d{3}$', line):
-                new_lines.append(line)
-                try:
-                    new_lines.append(next(text_iter))
-                except StopIteration:
-                    break
-            elif line == '':
-                new_lines.append(line)
-
-        new_content = '\n'.join(new_lines)
-
-        # Download Button
-        st.download_button(
-            label="Download Translated SRT File",
-            data=new_content,
-            file_name="translated_subtitles.srt",
-            mime="text/plain"
-        )
-
-# Footer
-st.markdown(
-    """
-    <div class="footer">
-        Designed by <a href="https://github.com/kayhgng" style="color: #007fbf; text-decoration: none;">alikay_h</a>
-    </div>
-    """,
-    unsafe_allow_html=True
-)
+    output_text = "".join(output_lines)
+    st.download_button(label="دانلود فایل ترجمه شده", data=output_text, file_name="زیرنویس_ترجمه‌شده.srt", mime='text/srt')