Update app.py

app.py

@@ -2,30 +2,31 @@ from huggingface_hub import hf_hub_download
 from llama_cpp import Llama
 import gradio as gr
 
-# دانلود مدل از Hugging Face Hub
+# بارگیری مدل
 model_path = hf_hub_download(
     repo_id="rinrikatoki/dorna-quantized",
     filename="dorna-q8_0.gguf"
 )
 
-# بارگذاری مدل
-llm = Llama(
-    model_path=model_path,
-    n_ctx=512,         # می‌تونی بیشتر هم بزاری اگه رم داشته باشی
-    n_threads=4,       # بسته به منابع
-    n_batch=8
-)
+llm = Llama(model_path=model_path, n_ctx=512, n_threads=4, n_batch=8)
 
-# تابع چت
+# تابع چت برای قالب پیام جدید Gradio
 def chat_fn(message, history):
-    output = llm(message, max_tokens=128)
+    prompt = message
+    output = llm(prompt, max_tokens=128)
     response = output["choices"][0]["text"].strip()
-    history.append((message, response))
+    
+    # ساخت قالب جدید
+    if history is None:
+        history = []
+    history.append({"role": "user", "content": message})
+    history.append({"role": "assistant", "content": response})
+    
     return history, history
 
 # رابط Gradio
 with gr.Blocks() as demo:
-    gr.Markdown("## 🤖 چت با مدل Dorna")
+    gr.Markdown("## 🤖 چت با مدل درنا")
     
     chatbot = gr.Chatbot(label="درنا", type="messages")
     msg = gr.Textbox(label="پیام شما...")