Update app.py

app.py

@@ -15,6 +15,9 @@ from pathlib import Path
 
 load_dotenv()
 
+# Check Gradio version
+print(f"Gradio version: {gr.__version__}")
+
 # There are tools set here dependent on environment variables
 from graph import graph, weak_model, search_enabled # noqa
 
@@ -26,9 +29,14 @@ USER_INPUT_MAX_LENGTH = 10000  # Characters
 # If you store sensitive data, you should store your secret in .env
 BROWSER_STORAGE_SECRET = "itsnosecret"
 
-with open('logging-config.json', 'r') as fh:
-    config = json.load(fh)
-logging.config.dictConfig(config)
+try:
+    with open('logging-config.json', 'r') as fh:
+        config = json.load(fh)
+    logging.config.dictConfig(config)
+except FileNotFoundError:
+    # Fallback logging configuration
+    logging.basicConfig(level=logging.INFO)
+    
 logger = logging.getLogger(__name__)
 
 def load_initial_greeting(filepath="greeting_prompt.txt") -> str:
@@ -416,44 +424,59 @@ function triggerChatButtonClick() {
 if __name__ == "__main__":
     logger.info("Starting the DIYO interface")
     
+    # Check if BrowserState is available
+    has_browser_state = hasattr(gr, 'BrowserState')
+    logger.info(f"BrowserState available: {has_browser_state}")
+    
+    if not has_browser_state:
+        logger.warning("BrowserState not available in this Gradio version. Using regular State instead.")
+        logger.warning("To use BrowserState, upgrade Gradio: pip install gradio>=4.0.0")
+    
     with gr.Blocks(title="DIYO - DIY Assistant", fill_height=True, css=CSS, elem_id="main-app") as demo:
-        # State management
+        # State management - Use BrowserState if available, otherwise regular State
         is_new_user_for_greeting = gr.State(True)
         
-        current_prompt_state = gr.BrowserState(
-            value="You are a helpful DIY assistant.",
-            storage_key="current_prompt_state",
-            secret=BROWSER_STORAGE_SECRET,
-        )
-        current_uuid_state = gr.BrowserState(
-            value=uuid4,
-            storage_key="current_uuid_state",
-            secret=BROWSER_STORAGE_SECRET,
-        )
-        current_langgraph_state = gr.BrowserState(
-            value=dict,
-            storage_key="current_langgraph_state",
-            secret=BROWSER_STORAGE_SECRET,
-        )
-        end_of_assistant_response_state = gr.State(False)
+        if has_browser_state:
+            current_prompt_state = gr.BrowserState(
+                value="You are a helpful DIY assistant.",
+                storage_key="current_prompt_state",
+                secret=BROWSER_STORAGE_SECRET,
+            )
+            current_uuid_state = gr.BrowserState(
+                value=uuid4,
+                storage_key="current_uuid_state",
+                secret=BROWSER_STORAGE_SECRET,
+            )
+            current_langgraph_state = gr.BrowserState(
+                value=dict,
+                storage_key="current_langgraph_state",
+                secret=BROWSER_STORAGE_SECRET,
+            )
+            sidebar_names_state = gr.BrowserState(
+                value=dict,
+                storage_key="sidebar_names_state",
+                secret=BROWSER_STORAGE_SECRET,
+            )
+            offloaded_tabs_data_storage = gr.BrowserState(
+                value=dict,
+                storage_key="offloaded_tabs_data_storage",
+                secret=BROWSER_STORAGE_SECRET,
+            )
+            chatbot_message_storage = gr.BrowserState(
+                value=list,
+                storage_key="chatbot_message_storage",
+                secret=BROWSER_STORAGE_SECRET,
+            )
+        else:
+            # Fallback to regular State
+            current_prompt_state = gr.State("You are a helpful DIY assistant.")
+            current_uuid_state = gr.State(uuid4())
+            current_langgraph_state = gr.State({})
+            sidebar_names_state = gr.State({})
+            offloaded_tabs_data_storage = gr.State({})
+            chatbot_message_storage = gr.State([])
         
-        # [uuid] -> summary of chat
-        sidebar_names_state = gr.BrowserState(
-            value=dict,
-            storage_key="sidebar_names_state",
-            secret=BROWSER_STORAGE_SECRET,
-        )
-        # [uuid] -> {"graph": gradio_graph, "messages": messages}
-        offloaded_tabs_data_storage = gr.BrowserState(
-            value=dict,
-            storage_key="offloaded_tabs_data_storage",
-            secret=BROWSER_STORAGE_SECRET,
-        )
-        chatbot_message_storage = gr.BrowserState(
-            value=list,
-            storage_key="chatbot_message_storage",
-            secret=BROWSER_STORAGE_SECRET,
-        )
+        end_of_assistant_response_state = gr.State(False)
         
         # Header
         with gr.Row(elem_classes="header-margin"):
@@ -785,43 +808,51 @@ if __name__ == "__main__":
             trigger_mode="always_last"
         )
 
-        # Load event handlers
-        @demo.load(
-            inputs=[is_new_user_for_greeting, chatbot_message_storage],
-            outputs=[chatbot_message_storage, is_new_user_for_greeting]
-        )
-        def handle_initial_greeting_load(current_is_new_user_flag: bool, existing_chat_history: list):
-            """Handle initial greeting when the app loads"""
-            if current_is_new_user_flag:
-                greeting_message_text = load_initial_greeting()
-                greeting_entry = {"role": "assistant", "content": greeting_message_text}
-                
-                if not isinstance(existing_chat_history, list):
-                    existing_chat_history = []
+        # Load event handlers - only add these if we have BrowserState
+        if has_browser_state:
+            @demo.load(
+                inputs=[is_new_user_for_greeting, chatbot_message_storage],
+                outputs=[chatbot_message_storage, is_new_user_for_greeting]
+            )
+            def handle_initial_greeting_load(current_is_new_user_flag: bool, existing_chat_history: list):
+                """Handle initial greeting when the app loads"""
+                if current_is_new_user_flag:
+                    greeting_message_text = load_initial_greeting()
+                    greeting_entry = {"role": "assistant", "content": greeting_message_text}
                     
-                updated_chat_history = [greeting_entry] + existing_chat_history
-                updated_is_new_user_flag = False
-                logger.info("Greeting added for new user.")
-                return updated_chat_history, updated_is_new_user_flag
-            else:
-                logger.info("Not a new user or already greeted.")
-                if not isinstance(existing_chat_history, list):
-                    existing_chat_history = []
-                return existing_chat_history, False
-
-        @demo.load(inputs=[chatbot_message_storage], outputs=[chatbot])
-        def load_messages(messages):
-            """Load stored messages into chatbot"""
-            if isinstance(messages, list):
-                return messages
-            return []
-
-        @demo.load(inputs=[current_prompt_state], outputs=[prompt_textbox])
-        def load_prompt(current_prompt):
-            """Load stored prompt"""
-            if current_prompt:
-                return current_prompt
-            return "You are a helpful DIY assistant."
+                    if not isinstance(existing_chat_history, list):
+                        existing_chat_history = []
+                        
+                    updated_chat_history = [greeting_entry] + existing_chat_history
+                    updated_is_new_user_flag = False
+                    logger.info("Greeting added for new user.")
+                    return updated_chat_history, updated_is_new_user_flag
+                else:
+                    logger.info("Not a new user or already greeted.")
+                    if not isinstance(existing_chat_history, list):
+                        existing_chat_history = []
+                    return existing_chat_history, False
+
+            @demo.load(inputs=[chatbot_message_storage], outputs=[chatbot])
+            def load_messages(messages):
+                """Load stored messages into chatbot"""
+                if isinstance(messages, list):
+                    return messages
+                return []
+
+            @demo.load(inputs=[current_prompt_state], outputs=[prompt_textbox])
+            def load_prompt(current_prompt):
+                """Load stored prompt"""
+                if current_prompt:
+                    return current_prompt
+                return "You are a helpful DIY assistant."
+        else:
+            # For regular State, add a simple greeting on load
+            @demo.load(outputs=[chatbot])
+            def load_initial_greeting():
+                """Load initial greeting for users without BrowserState"""
+                greeting_text = load_initial_greeting()
+                return [{"role": "assistant", "content": greeting_text}]
 
     # Launch the application
     demo.launch(debug=True, share=True)