Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 from huggingface_hub import hf_hub_download
 from llama_cpp import Llama
 from llama_cpp_agent import LlamaCppAgent, MessagesFormatterType
-from llama_cpp_agent.providers import LlamaCppPythonProvider  # ✅ این خط اضافه شد
+from llama_cpp_agent.providers import LlamaCppPythonProvider
 from llama_cpp_agent.chat_history import BasicChatHistory
 from llama_cpp_agent.chat_history.messages import Roles
 
@@ -21,7 +21,7 @@ llm = Llama(
     verbose=False
 )
 
-provider = LlamaCppPythonProvider(llm)  # ✅ حالا این خط کار میکنه
+provider = LlamaCppPythonProvider(llm)
 
 agent = LlamaCppAgent(
     provider,
@@ -31,18 +31,13 @@ agent = LlamaCppAgent(
 )
 
 # 💬 تابع پاسخ‌دهنده
-def respond(message, chat_history):
-    messages = BasicChatHistory()
+def respond(message, chat_history=[], system_message="You are Arsh...", max_tokens=2048, temperature=0.6, top_p=0.95, top_k=40, repeat_penalty=1.1):
+    if chat_history is None:
+        chat_history = []
 
-    for user_msg, assistant_msg in chat_history:
-        messages.add_message({
-            'role': Roles.user,
-            'content': user_msg
-        })
-        messages.add_message({
-            'role': Roles.assistant,
-            'content': assistant_msg
-        })
+    messages = BasicChatHistory()
+    for msg in chat_history:
+        messages.add_message(msg)
 
     stream = agent.get_chat_response(
         message,
@@ -70,6 +65,7 @@ with gr.Blocks() as demo:
             gr.Slider(minimum=0.0, maximum=2.0, value=1.1, step=0.1, label="Repetition Penalty"),
         ],
         chatbot=gr.Chatbot(
+            type="messages",  # ✅ اینجا نوعشو مشخص کردیم
             label="Chat with Arsh-LLM",
             bubble_full_width=False,
             show_copy_button=True