Update app.py

app.py

@@ -6,19 +6,17 @@ model_name = "HooshvareLab/gpt2-fa"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
 
+chat_history = []
 # تابع پاسخ‌دهی
 def student_bot(question):
-    prompt = f"""پرسش: زمان انتخاب واحد چه زمانی است؟
+    chat_history.append(f"پرسش: {question}")
+    
+    prompt = """پرسش: زمان انتخاب واحد چه زمانی است؟
 پاسخ: معمولا در پایان هر ترم تحصیلی، یعنی شهریور و بهمن انجام می‌شود.
 
 پرسش: زمان حذف و اضافه کی هست؟
 پاسخ: حدود یک هفته بعد از شروع ترم تحصیلی است.
-
-پرسش: چه زمانی می‌توان انصراف از درس داد؟
-پاسخ: معمولاً تا هفته هشتم می‌توان درس را حذف کرد.
-
-پرسش: {question}
-پاسخ:"""
+""" + "\n".join(chat_history[-3:]) + "\nپاسخ:"
 
     inputs = tokenizer(prompt, return_tensors="pt")
     outputs = model.generate(
@@ -27,11 +25,10 @@ def student_bot(question):
         pad_token_id=tokenizer.eos_token_id
     )
     answer = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    
-    # حذف prompt اولیه از متن خروجی
     return answer.split("پاسخ:")[-1].strip()
 
 
+
 # رابط Gradio
 gr.Interface(
     fn=student_bot,