Update app.py

app.py

@@ -27,11 +27,8 @@ pipe = pipeline(
 
 def respond(message, history):
     messages = []
-    for user_msg, assistant_msg in history:
-        if user_msg:
-            messages.append({"role": "user", "content": user_msg})
-        if assistant_msg:
-            messages.append({"role": "assistant", "content": assistant_msg})
+    for item in history:
+        messages.append({"role": item["role"], "content": item["content"]})
 
     messages.append({"role": "user", "content": message})
 
@@ -50,6 +47,7 @@ def respond(message, history):
 
 demo = gr.ChatInterface(
     fn=respond,
+    type="messages",
     title="TaskMind Interface",
     description="Chat with the TaskMind LoRA model.",
     examples=[