Update app.py

app.py

@@ -80,7 +80,6 @@ class XylariaChat:
             print(f"Error resetting API client: {e}")
 
         self.save_chat()  # Save the empty chat history
-        #return None  # To clear the chatbot interface
 
     def get_response(self, user_input):
         # Prepare messages with conversation context and persistent memory
@@ -264,7 +263,8 @@ class XylariaChat:
                             height=500,
                             show_copy_button=True,
                             avatar_images=("user.png", "xylaria.png"),  # Replace with your image paths
-                            bubble_full_width=False
+                            bubble_full_width=False,
+                            type="messages"
                         )
 
                         # Clear history and memory buttons
@@ -303,7 +303,7 @@ class XylariaChat:
                 queue=False
             ).then(
                 fn=lambda: toggle_page("start"),
-                inputs=gr.State("start"),
+                inputs=None,
                 outputs=[chat_page, start_page]
             )
 
@@ -311,7 +311,7 @@ class XylariaChat:
             clear_memory.click(
                 fn=self.reset_conversation,
                 inputs=None,
-                outputs=[chatbot],
+                outputs=None,
                 queue=False
             ).then(
                 fn=lambda: toggle_page("start"),
@@ -319,8 +319,8 @@ class XylariaChat:
                 outputs=[chat_page, start_page]
             )
 
-            # Load chat history on interface load
-            demo.load(self.reset_conversation, [], None)
+            # Load on startup
+            demo.load(fn=self.reset_conversation, inputs=None, outputs=None)
 
             return demo