Update app.py

app.py

@@ -1,10 +1,10 @@
 import gradio as gr
 from sentence_transformers import SentenceTransformer, util
 
-# بارگذاری مدل SBERT فارسی
-embedder = SentenceTransformer("HooshvareLab/sbert-fa-zwnj-base")
+# بارگذاری مدل تشابه معنایی
+embedder = SentenceTransformer("myrkur/sentence-transformer-parsbert-fa-2.0")
 
-# پایگاه داده‌ی سوالات متداول (FAQ)
+# پایگاه داده‌ی سوالات متداول
 faq_dict = {
     "زمان انتخاب واحد": "معمولاً پایان شهریور و بهمن است.",
     "زمان حذف و اضافه": "حدود یک هفته پس از شروع ترم تحصیلی است.",
@@ -13,29 +13,26 @@ faq_dict = {
     "شرایط مهمان شدن": "با موافقت دانشگاه مبدا و مقصد انجام می‌شود.",
 }
 
-# آماده‌سازی Embedding سوالات FAQ
 faq_questions = list(faq_dict.keys())
 faq_embeddings = embedder.encode(faq_questions, convert_to_tensor=True)
 
-# تابع پاسخ‌دهی با تشابه معنایی
 def student_bot(question):
     question_embedding = embedder.encode(question, convert_to_tensor=True)
     cos_scores = util.pytorch_cos_sim(question_embedding, faq_embeddings)[0]
     best_score = cos_scores.max().item()
     best_idx = cos_scores.argmax().item()
 
-    if best_score >= 0.7:  # آستانه شباهت
-        best_question = faq_questions[best_idx]
-        return faq_dict[best_question]
+    if best_score >= 0.65:  # آستانه شباهت نسبتاً معتدل
+        best_q = faq_questions[best_idx]
+        return faq_dict[best_q]
     else:
         return "❗️متاسفم، پاسخ این سؤال فعلاً در سیستم ثبت نشده است."
 
-# رابط کاربری Gradio
-iface = gr.Interface(fn=student_bot,
-                     inputs=gr.Textbox(label="سؤال خود را وارد کنید"),
-                     outputs=gr.Textbox(label="پاسخ"),
-                     title="ایجنت راهنمای دانشجویان",
-                     description="پاسخ به سوالات متداول دانشگاهی با تشخیص معنایی",
-                     theme="default")
-
+iface = gr.Interface(
+    fn=student_bot,
+    inputs=gr.Textbox(label="سؤال خود را وارد کنید"),
+    outputs=gr.Textbox(label="پاسخ"),
+    title="ایجنت راهنمای دانشجویان",
+    description="پاسخ به سوالات متداول دانشگاه با تشخیص معنایی فارسی"
+)
 iface.launch()