Update app.py

app.py

@@ -13,7 +13,7 @@ class ChatMessage:
     role: str
     content: str
 
-    def to_dict__(self):
+    def to_dict(self):
         """Converts ChatMessage to a dictionary for JSON serialization."""
         return {"role": self.role, "content": self.content}
 
@@ -162,10 +162,7 @@ class XylariaChat:
 
             # Convert existing conversation history to ChatMessage objects and then to dictionaries
             for msg in self.conversation_history:
-                messages.append(ChatMessage(
-                    role=msg['role'],
-                    content=msg['content']
-                ).to_dict())
+                messages.append(msg)
 
             # Process image if uploaded
             if image:
@@ -270,12 +267,8 @@ class XylariaChat:
                 return
 
             # Update conversation history
-            self.conversation_history.append(
-                {"role": "user", "content": message}
-            )
-            self.conversation_history.append(
-                {"role": "assistant", "content": full_response}
-            )
+            self.conversation_history.append(ChatMessage(role="user", content=message).to_dict())
+            self.conversation_history.append(ChatMessage(role="assistant", content=full_response).to_dict())
 
             # Limit conversation history
             if len(self.conversation_history) > 10: