Update app.py

app.py

@@ -88,6 +88,22 @@ Capabilities:
             return f"Error generating response: {str(e)}"
 
     def create_interface(self):
+        # Local storage JavaScript functions
+        local_storage_js = """
+        function saveToLocalStorage(chatHistory) {
+            localStorage.setItem('xylaria_chat_history', JSON.stringify(chatHistory));
+        }
+
+        function loadFromLocalStorage() {
+            const savedHistory = localStorage.getItem('xylaria_chat_history');
+            return savedHistory ? JSON.parse(savedHistory) : [];
+        }
+
+        function clearLocalStorage() {
+            localStorage.removeItem('xylaria_chat_history');
+        }
+        """
+
         def streaming_response(message, chat_history):
             # Clear input textbox
             response_stream = self.get_response(message)
@@ -143,13 +159,14 @@ Capabilities:
         }
         """
 
-        with gr.Blocks(theme='soft', css=custom_css) as demo:
+        with gr.Blocks(theme='soft', css=custom_css, js=local_storage_js) as demo:
             # Chat interface with improved styling
             with gr.Column():
                 chatbot = gr.Chatbot(
                     label="Xylaria 1.4 Senoa",
                     height=500,
-                    show_copy_button=True
+                    show_copy_button=True,
+                    value=lambda: gr.load("json", "loadFromLocalStorage()")
                 )
                 
                 # Input row with improved layout
@@ -166,30 +183,42 @@ Capabilities:
                 clear = gr.Button("Clear Conversation")
                 clear_memory = gr.Button("Clear Memory")
             
-            # Submit functionality
+            # Submit functionality with local storage save
             btn.click(
                 fn=streaming_response, 
                 inputs=[txt, chatbot], 
                 outputs=[txt, chatbot]
+            ).then(
+                fn=None,  # JavaScript callback
+                _js='(chatHistory) => saveToLocalStorage(chatHistory)'
             )
             txt.submit(
                 fn=streaming_response, 
                 inputs=[txt, chatbot], 
                 outputs=[txt, chatbot]
+            ).then(
+                fn=None,  # JavaScript callback
+                _js='(chatHistory) => saveToLocalStorage(chatHistory)'
             )
             
-            # Clear conversation history
+            # Clear conversation history with local storage clear
             clear.click(
                 fn=lambda: [], 
                 inputs=None, 
                 outputs=[chatbot]
+            ).then(
+                fn=None,  # JavaScript callback
+                _js='() => clearLocalStorage()'
             )
             
-            # Clear persistent memory and reset conversation
+            # Clear persistent memory and reset conversation with local storage clear
             clear_memory.click(
                 fn=lambda: [], 
                 inputs=None, 
                 outputs=[chatbot]
+            ).then(
+                fn=None,  # JavaScript callback
+                _js='() => clearLocalStorage()'
             )
         
         return demo