Update app.py

app.py

@@ -30,8 +30,11 @@ SYSTEM_PROMPT = "You are MathBioAgent, an expert AI assistant specialized in mat
 def chat(message, history):
     messages = [{"role": "system", "content": SYSTEM_PROMPT}]
     for h in history:
-        messages.append({"role": "user", "content": h[0]})
-        messages.append({"role": "assistant", "content": h[1]})
+        if isinstance(h, dict):
+            messages.append(h)
+        else:
+            messages.append({"role": "user", "content": h[0]})
+            messages.append({"role": "assistant", "content": h[1]})
     messages.append({"role": "user", "content": message})
     
     prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
@@ -50,6 +53,7 @@ def chat(message, history):
 
 demo = gr.ChatInterface(
     fn=chat,
+    type="messages",
     title="MathBio AI — Mathematical Biology Assistant",
     description="A specialized LLM for mathematical biology, epidemic modeling, PDEs, and operator learning. Fine-tuned on 27K arxiv-derived examples.",
     examples=[