Update app.py

app.py

@@ -657,11 +657,11 @@ class XylariaChat:
                     base64_image = base64.b64encode(image_bytes).decode("utf-8")
                     image_html = f'<img src="data:image/png;base64,{base64_image}" alt="Generated Image" style="max-width: 100%; max-height: 400px;">'
                     
-                    # Replace loading SVG with the generated image
+                    # Replace loading SVG with the generated image in the AI's chat bubble
                     chat_history[-1][1] = image_html
                     
                     self.conversation_history.append(ChatMessage(role="user", content=message).to_dict())
-                    self.conversation_history.append(ChatMessage(role="assistant", content=image_html).to_dict())
+                    self.conversation_history.append(ChatMessage(role="assistant", content=image_html).to_dict()) # Correctly append image to conversation history
 
                     yield "", chat_history, None, None
                 else: