Update app.py

app.py

@@ -11,7 +11,7 @@ class XylariaChat:
 
         # Initialize the inference client
         self.client = InferenceClient(
-            model="Qwen/Qwen-72B-Chat",  # Changed model name
+            model="Qwen/Qwen-72B-Chat",  # Changed model name 
             token=self.hf_token
         )
 
@@ -103,18 +103,22 @@ Capabilities:
         return prompt
 
     def create_interface(self):
-        # Local storage JavaScript functions
-        local_storage_js = """
-        function saveToLocalStorage(chatHistory) {
-            localStorage.setItem('xylaria_chat_history', JSON.stringify(chatHistory));
-        }
-
-        function loadFromLocalStorage() {
+        # Local storage JavaScript functions (these are strings, not functions)
+        load_from_local_storage_js = """
+        function() {
             const savedHistory = localStorage.getItem('xylaria_chat_history');
             return savedHistory ? JSON.parse(savedHistory) : [];
         }
+        """
 
-        function clearLocalStorage() {
+        save_to_local_storage_js = """
+        function(chatHistory) {
+            localStorage.setItem('xylaria_chat_history', JSON.stringify(chatHistory));
+        }
+        """
+
+        clear_local_storage_js = """
+        function() {
             localStorage.removeItem('xylaria_chat_history');
         }
         """
@@ -196,8 +200,15 @@ Capabilities:
                 clear = gr.Button("Clear Conversation")
                 clear_memory = gr.Button("Clear Memory")
 
-            # Load chat history from local storage on page load using `gr.JSON`
-            chatbot.value = gr.JSON(value=loadFromLocalStorage())
+            # Load chat history from local storage on page load
+            demo.load(
+                fn=None,
+                _js=load_from_local_storage_js  # Pass the JS function as a string
+            ).then(
+                fn=lambda x: x,
+                inputs=chatbot,
+                outputs=chatbot
+            )
 
             # Submit functionality with local storage save
             btn.click(
@@ -208,7 +219,7 @@ Capabilities:
                 fn=None,
                 inputs=[chatbot],  # Pass chatbot history to JavaScript
                 outputs=None,
-                _js='(chatHistory) => saveToLocalStorage(chatHistory)'
+                _js=save_to_local_storage_js
             )
             txt.submit(
                 fn=streaming_response,
@@ -218,7 +229,7 @@ Capabilities:
                 fn=None,
                 inputs=[chatbot],  # Pass chatbot history to JavaScript
                 outputs=None,
-                _js='(chatHistory) => saveToLocalStorage(chatHistory)'
+                _js=save_to_local_storage_js
             )
 
             # Clear conversation history with local storage clear
@@ -228,7 +239,7 @@ Capabilities:
                 outputs=[chatbot]
             ).then(
                 fn=None,
-                _js='() => clearLocalStorage()'
+                _js=clear_local_storage_js
             )
 
             # Clear persistent memory and reset conversation with local storage clear
@@ -238,7 +249,7 @@ Capabilities:
                 outputs=[chatbot]
             ).then(
                 fn=None,
-                _js='() => clearLocalStorage()'
+                _js=clear_local_storage_js
             )
 
         return demo