Update app.py

app.py

@@ -1,16 +1,13 @@
 import os
 import gradio as gr
 from transformers import pipeline
-from argostranslate import package, translate
+from googletrans import Translator
 
-# تولید مدل انگلیسی برای موضوع پایان‌نامه
-generator = pipeline("text-generation", model="tiiuae/falcon-rw-1b")
+# تولید موضوع به انگلیسی
+text_generator = pipeline("text-generation", model="tiiuae/falcon-rw-1b")
 
-# گردان ترجمه انگلیسی به فارسی (argos-translate)
-installed_languages = translate.get_installed_languages()
-en_lang = list(filter(lambda x: x.code == "en", installed_languages))[0]
-fa_lang = list(filter(lambda x: x.code == "fa", installed_languages))[0]
-translator = en_lang.get_translation(fa_lang)
+# ترجمه به فارسی با googletrans
+translator = Translator()
 
 def generate_topics(field, major, keywords, audience, level):
     prompt = f"""
@@ -22,15 +19,10 @@ Target audience: {audience}
 Level: {level}
 """
 
-    # generate English topics
-    english_output = generator(prompt, max_new_tokens=256)[0]['generated_text']
+    english_output = text_generator(prompt, max_new_tokens=256)[0]['generated_text']
+    translated_output = translator.translate(english_output, src='en', dest='fa').text
 
-    # translate to Persian
-    translated_output = translator.translate(english_output)
-
-    # add final message
     final_output = translated_output.strip() + "\n\n📢 برای مشاوره و راهنمایی تخصصی با گروه مشاوره کاسپین تماس بگیرید:\n02188252497"
-
     return final_output
 
 iface = gr.Interface(