bug fix.

app.py

@@ -48,19 +48,25 @@ agent = Agent(
     markdown=True,
 )
 
-# ✅ Chat function
 def ask_agent(messages, history):
     print("====== DEBUG: ask_agent START ======")
     print("📥 Incoming messages:", messages)
     print("📚 Chat history:", history)
 
+    # ✅ Handle Hugging Face's tendency to send just a string
     if isinstance(messages, str):
-        print("🧪 Detected string instead of message list. Wrapping as message.")
         messages = [{"role": "user", "content": messages}]
-
-    if not isinstance(messages, list) or not messages:
-        print("⚠️ Invalid or empty messages list.")
-        return {"role": "assistant", "content": "Hi! How can I help you with treasury insights today?"}
+    elif isinstance(messages, list):
+        # Defensive fix: if list of strings, wrap each
+        if all(isinstance(m, str) for m in messages):
+            messages = [{"role": "user", "content": m} for m in messages]
+        # If it's a list but not valid messages, throw a descriptive error
+        elif not all(isinstance(m, dict) and "role" in m and "content" in m for m in messages):
+            print("❌ Malformed message list")
+            return {"role": "assistant", "content": "Sorry, I couldn't understand the message format."}
+    else:
+        print("❌ Completely invalid input type")
+        return {"role": "assistant", "content": "Sorry, I didn't understand that input."}
 
     try:
         latest_user_message = messages[-1]["content"]
@@ -72,13 +78,10 @@ def ask_agent(messages, history):
     try:
         response = agent.run(latest_user_message)
         print(f"✅ Agent response:\n{response}")
-        return messages + [{"role": "assistant", "content": response.content}]  # ✅ RETURN FULL LIST
+        return {"role": "assistant", "content": response.content}
     except Exception as e:
         print(f"❌ Agent failed: {e}")
-        return messages + [{"role": "assistant", "content": "Something went wrong on my end. Try again?"}]
-
-    print("====== DEBUG: ask_agent END ======")
-
+        return {"role": "assistant", "content": "Something went wrong on my end. Try again?"}
 
 def yes(message, history):
     return "yes"