Update app.py

app.py

@@ -20,26 +20,26 @@ def respond(
     top_p,
 ):
     # ساخت پیام‌ها برای مدل
-    messages = [{"role": "system", "content": system_message}]
-
-    # اضافه کردن تاریخچه چت
+    prompt = system_message + "\n\n"  # اضافه کردن پیام سیستم
     for val in history:
         if val[0]:
-            messages.append({"role": "user", "content": val[0]})
+            prompt += f"کاربر: {val[0]}\n"
         if val[1]:
-            messages.append({"role": "assistant", "content": val[1]})
-
-    messages.append({"role": "user", "content": message})
+            prompt += f"دستیار: {val[1]}\n"
+    prompt += f"کاربر: {message}\nدستیار:"
 
     # تولید پاسخ با استفاده از مدل
     response = pipe(
-        messages,
+        prompt,
         max_new_tokens=max_tokens,
         temperature=temperature,
         top_p=top_p,
         do_sample=True,
     )[0]["generated_text"]
 
+    # حذف prompt از پاسخ
+    response = response[len(prompt):].strip()
+
     yield response
 
 # ایجاد رابط کاربری با Gradio