Update app.py

app.py

@@ -1,54 +1,37 @@
 import os
-import re
 import gradio as gr
 from transformers import pipeline
+from argostranslate import package, translate
 
-# گرفتن توکن از متغیر محیطی
-token = os.environ.get("HF_TOKEN")
+# تولید مدل انگلیسی برای موضوع پایان‌نامه
+generator = pipeline("text-generation", model="tiiuae/falcon-rw-1b")
 
-# استفاده از مدل نسبتاً سبک و قابل اجرا در Hugging Face Spaces
-pipe = pipeline(
-    "text-generation",
-    model="google/gemma-2b-it",
-    token=token
-)
-
-# تابع تشخیص و ترجمه کلمات انگلیسی رایج در حوزه روانشناسی
-EN_FA_DICT = {
-    "behavior": "رفتار",
-    "self-esteem": "عزت نفس",
-    "confidence": "اعتماد به نفس",
-    "learning": "یادگیری",
-    "attachment": "دلبستگی",
-    "communication": "ارتباط",
-    "stress": "استرس",
-    "parent": "والد",
-    "trauma": "آسیب روانی"
-}
-
-def replace_english_words(text):
-    for eng, fa in EN_FA_DICT.items():
-        text = re.sub(rf"\b{eng}\b", fa, text, flags=re.IGNORECASE)
-    return text
+# گردان ترجمه انگلیسی به فارسی (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)
 
 def generate_topics(field, major, keywords, audience, level):
     prompt = f"""
-۳ موضوع پایان‌نامه در رشته {field} با گرایش {major} پیشنهاد بده که به کلیدواژه‌های "{keywords}" مربوط باشه و جامعه هدف آن "{audience}" باشد. مقطع: {level}.
-موضوعات را فارسی بنویس.
+Suggest 3 academic thesis topics based on the following information:
+Field: {field}
+Specialization: {major}
+Keywords: {keywords}
+Target audience: {audience}
+Level: {level}
 """
-    output = pipe(prompt, max_new_tokens=250)[0]['generated_text']
 
-    # حذف prompt در صورتی که مدل آن را تکرار کرده باشد
-    if output.startswith(prompt.strip()):
-        output = output[len(prompt.strip()):].strip()
+    # generate English topics
+    english_output = generator(prompt, max_new_tokens=256)[0]['generated_text']
 
-    # حذف فاصله‌های اضافی و جایگزینی کلمات انگلیسی
-    output = replace_english_words(output.strip())
+    # translate to Persian
+    translated_output = translator.translate(english_output)
 
-    # افزودن پیام پایانی تبلیغاتی
-    output += "\n\nبرای مشاوره و راهنمایی تخصصی با گروه مشاوره کاسپین تماس بگیرید:\n02188252497"
+    # add final message
+    final_output = translated_output.strip() + "\n\n📢 برای مشاوره و راهنمایی تخصصی با گروه مشاوره کاسپین تماس بگیرید:\n02188252497"
 
-    return output
+    return final_output
 
 iface = gr.Interface(
     fn=generate_topics,
@@ -60,7 +43,7 @@ iface = gr.Interface(
         gr.Dropdown(choices=["کارشناسی ارشد", "دکتری"], label="مقطع")
     ],
     outputs="text",
-    title="پیشنهادگر هوشمند موضوع پایان‌نامه کاسپین 🎓"
+    title="🎓 پیشنهادگر موضوع پایان‌نامه کاسپین"
 )
 
 iface.launch()