Update app.py

app.py

@@ -18,7 +18,12 @@ os.environ["HUGGINGFACEHUB_API_TOKEN"] = st.secrets['huggingface_token']
 def trs_fa_to_en(text):
   txt_en=translators.translate_text(text,to_language='en',from_language='auto')
   return txt_en
-
+#########################################################################################
+def trs_en_to_fa(text):
+  txt_fa=translators.translate_text(text,to_language='fa',from_language='en')
+  return txt_fa
+##############################################################################################
+    
 def get_pdf_text(pdf_docs : list) -> str:
     text = ""
     for pdf in pdf_docs:
@@ -72,8 +77,11 @@ def handle_userinput(user_question:str):
             st.write("سوال کاربر: " + message.content)
         else:
             st.write("پاسخ ربات: " + message.content)
+            text=message.content
+            txt_fa=trs_en_to_fa(text=text)
+            st.write("پاسخ ربات: " + txt_fa)
             
-
+#############################################################################################################
 
 def main():
     st.set_page_config(
@@ -124,4 +132,4 @@ def main():
                 st.write("compelete build model")
 
 if __name__ == "__main__":
-    main() 
+    main()