Update app.py

app.py

@@ -2,6 +2,16 @@ import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
 
+faq_dict = {
+    "زمان انتخاب واحد": "معمولاً پایان شهریور و بهمن است.",
+    "زمان حذف و اضافه": "حدود یک هفته پس از شروع ترم تحصیلی است.",
+    "معدل لازم برای 24 واحد": "حداقل معدل 17 نیاز است.",
+    "حذف اضطراری": "تا هفته هشتم ترم مجاز است.",
+    "شرایط مهمان شدن": "با موافقت دانشگاه مبدا و مقصد انجام می‌شود.",
+}
+
+
+
 # بارگذاری پرسش و پاسخ‌های FAQ
 faq_dict = {}
 with open("faq.txt", encoding="utf-8") as f:
@@ -18,13 +28,20 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
 
 chat_history = []
+
 # تابع پاسخ‌دهی
 def student_bot(question):
+    # بررسی وجود پاسخ در دیکشنری FAQ
+    for key in faq_dict:
+        if key in question:
+            return faq_dict[key]
+
+    # اگر یافت نشد، مدل زبانی فعال شود
     prompt = f"""پرسش: زمان انتخاب واحد چه زمانی است؟
-پاسخ: معمولا در پایان هر ترم تحصیلی، یعنی شهریور و بهمن انجام می‌شود.
+پاسخ: معمولاً پایان شهریور و بهمن است.
 
 پرسش: زمان حذف و اضافه کی هست؟
-پاسخ: حدود یک هفته بعد از شروع ترم تحصیلی است.
+پاسخ: حدود یک هفته پس از شروع ترم تحصیلی است.
 
 پرسش: {question}
 پاسخ:"""
@@ -47,6 +64,7 @@ def student_bot(question):
 
 
 
+
 # رابط Gradio
 gr.Interface(
     fn=student_bot,