Update app.py

app.py

@@ -6,12 +6,14 @@ from deep_translator import GoogleTranslator
 st.set_page_config(layout="wide")
 
 @st.cache_resource
-def text_summary(text, maxlength=None):
+def text_summary(text, translate_fa, maxlength=None):
     translatedtext = GoogleTranslator(source='auto', target='en').translate(text)
     #create summary instance
     summary = Summary()
     text = (text)
     result = summary(text)
+    if translate_fa:
+        result = GoogleTranslator(source='auto', target='en').translate(result)
     return result
 
 def extract_text_from_pdf(file_path):
@@ -23,6 +25,7 @@ def extract_text_from_pdf(file_path):
     return text
 
 choice = st.sidebar.selectbox("Select your choice", ["Summarize Text", "Summarize Document"])
+translate_fa = st.toggle('Translate to Persian')
 
 if choice == "Summarize Text":
     st.subheader("Summarize Text using txtai")
@@ -35,7 +38,7 @@ if choice == "Summarize Text":
                 st.info(input_text)
             with col2:
                 st.markdown("**Summary Result**")
-                result = text_summary(input_text)
+                result = text_summary(input_text, translate_fa)
                 st.success(result)
 
 elif choice == "Summarize Document":
@@ -54,6 +57,6 @@ elif choice == "Summarize Document":
             with col2:
                 st.markdown("**Summary Result**")
                 text = extract_text_from_pdf("doc_file.pdf")
-                doc_summary = text_summary(text)
+                doc_summary = text_summary(text, translate_fa)
                 st.success(doc_summary)