Update app.py

app.py

@@ -28,18 +28,25 @@ SYSTEM_PROMPT = "You are MathBioAgent, an expert AI assistant specialized in mat
 
 @spaces.GPU(duration=60)
 def chat(message, history):
+    def extract_text(content):
+        if isinstance(content, str):
+            return content
+        if isinstance(content, list):
+            return " ".join(c.get("text", "") if isinstance(c, dict) else str(c) for c in content)
+        return str(content)
+
     messages = [{"role": "system", "content": SYSTEM_PROMPT}]
     for h in history:
         if isinstance(h, dict):
-            messages.append(h)
+            messages.append({"role": h["role"], "content": extract_text(h.get("content", ""))})
         elif isinstance(h, (list, tuple)) and len(h) == 2:
-            messages.append({"role": "user", "content": h[0]})
-            messages.append({"role": "assistant", "content": h[1]})
-    messages.append({"role": "user", "content": message})
-    
+            messages.append({"role": "user", "content": extract_text(h[0])})
+            messages.append({"role": "assistant", "content": extract_text(h[1])})
+    messages.append({"role": "user", "content": extract_text(message)})
+
     prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
     inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
-    
+
     with torch.no_grad():
         outputs = model.generate(
             **inputs,