Update app.py

app.py

@@ -5,30 +5,28 @@ from transformers import pipeline
 # گرفتن توکن از متغیر محیطی
 token = os.environ.get("HF_TOKEN")
 
-# استفاده از مدل سازگار با Transformers (نسخه 2B)
+# استفاده از مدل سازگار با Transformers
 pipe = pipeline(
     "text-generation",
-    model="google/gemma-2b-it",  # ✅ نسخه صحیح و قابل اجرا
+    model="google/gemma-2b-it",  # ✅ مدل سبک و سازگار
     token=token
 )
 
 def generate_topics(field, major, keywords, audience, level):
     prompt = f"""
-    با توجه به اطلاعات زیر، چند موضوع مناسب برای پایان‌نامه پیشنهاد بده:
-
-    رشته: {field}
-    گرایش: {major}
-    کلیدواژه‌ها: {keywords}
-    جامعه هدف: {audience}
-    مقطع: {level}
-
-    لطفاً ۳ موضوع پیشنهادی ارائه کن.
-    در پایان، بنویس:
-    برای مشاوره و راهنمایی تخصصی با گروه مشاوره کاسپین تماس بگیرید:
-    02188252497
-    """
-    result = pipe(prompt, max_new_tokens=250)[0]['generated_text']
-    return result
+۳ موضوع پایان‌نامه در رشته {field} با گرایش {major} پیشنهاد بده که به کلیدواژه‌های "{keywords}" مربوط باشه و جامعه هدف آن "{audience}" باشد. مقطع: {level}.
+موضوعات را فارسی بنویس.
+"""
+    response = pipe(prompt, max_new_tokens=250)[0]['generated_text']
+
+    # حذف متن prompt از ابتدای پاسخ (در صورتی که مدل آن را تکرار کند)
+    if response.startswith(prompt.strip()):
+        response = response[len(prompt.strip()):].strip()
+
+    # افزودن متن پایانی ثابت
+    response += "\n\nبرای مشاوره و راهنمایی تخصصی با گروه مشاوره کاسپین تماس بگیرید:\n02188252497"
+
+    return response
 
 iface = gr.Interface(
     fn=generate_topics,