Update app.py

app.py

@@ -15,9 +15,27 @@ from pathlib import Path
 
 load_dotenv()
 
-# Check Gradio version
+# Check Gradio version and provide guidance
 print(f"Gradio version: {gr.__version__}")
 
+# Parse version to check compatibility
+try:
+    version_parts = gr.__version__.split('.')
+    major_version = int(version_parts[0])
+    minor_version = int(version_parts[1]) if len(version_parts) > 1 else 0
+    
+    if major_version < 4:
+        print("⚠️  WARNING: You're using an older version of Gradio.")
+        print("   Some features may be limited. Consider upgrading:")
+        print("   pip install --upgrade gradio>=4.0.0")
+    elif major_version >= 4:
+        print("✅ Gradio version is compatible with all features.")
+        
+except (ValueError, IndexError):
+    print("Could not parse Gradio version.")
+    
+print()  # Add spacing
+
 # There are tools set here dependent on environment variables
 from graph import graph, weak_model, search_enabled # noqa
 
@@ -394,6 +412,13 @@ footer {visibility: hidden}
     display: none !important;
 }
 
+.sidebar-replacement {
+    background-color: #f8f9fa;
+    border-left: 1px solid #dee2e6;
+    padding: 10px;
+    min-height: 400px;
+}
+
 .wrap.sidebar-parent {
     min-height: 2400px !important;
     height: 2400px !important;
@@ -429,8 +454,14 @@ if __name__ == "__main__":
     logger.info(f"BrowserState available: {has_browser_state}")
     
     if not has_browser_state:
+        print("📝 Note: Using session-only state (data won't persist after refresh)")
+        print("   For data persistence, upgrade to Gradio 4.0+")
         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")
+    else:
+        print("💾 Using persistent browser state (data persists after refresh)")
+    
+    print()  # Add spacing
     
     with gr.Blocks(title="DIYO - DIY Assistant", fill_height=True, css=CSS, elem_id="main-app") as demo:
         # State management - Use BrowserState if available, otherwise regular State
@@ -545,9 +576,11 @@ if __name__ == "__main__":
                         
                         # Check if cache_examples parameter is supported
                         try:
-                            if 'cache_examples' in gr.Examples.__init__.__code__.co_varnames:
+                            init_params = gr.Examples.__init__.__code__.co_varnames
+                            if 'cache_examples' in init_params:
                                 examples_kwargs["cache_examples"] = False
-                        except:
+                        except Exception:
+                            # Parameter not supported, skip it
                             pass
                             
                         gr.Examples(**examples_kwargs)
@@ -576,16 +609,20 @@ if __name__ == "__main__":
             
             # Check if newer textbox parameters are supported
             try:
-                if 'autofocus' in textbox_component.__init__.__code__.co_varnames:
+                init_params = textbox_component.__init__.__code__.co_varnames
+                
+                if 'autofocus' in init_params:
                     textbox_kwargs["autofocus"] = True
-                if 'submit_btn' in textbox_component.__init__.__code__.co_varnames:
+                if 'submit_btn' in init_params:
                     textbox_kwargs["submit_btn"] = True
-                if 'stop_btn' in textbox_component.__init__.__code__.co_varnames:
+                if 'stop_btn' in init_params:
                     textbox_kwargs["stop_btn"] = True
             except Exception as e:
                 logger.warning(f"Error checking textbox parameters: {e}")
+                # Keep minimal parameters as fallback
             
             textbox = textbox_component(**textbox_kwargs)
+            
             # Check if newer Chatbot parameters are supported
             chatbot_kwargs = {
                 "height": 400,
@@ -594,23 +631,32 @@ if __name__ == "__main__":
             
             # Add parameters that might not be available in older versions
             try:
-                # Try to create a test chatbot to see what parameters are supported
-                test_chatbot = gr.Chatbot()
+                # Check parameter availability without creating test instance
+                init_params = gr.Chatbot.__init__.__code__.co_varnames
                 
-                # Check if 'type' parameter is supported
-                if hasattr(test_chatbot, 'type') or 'type' in gr.Chatbot.__init__.__code__.co_varnames:
+                # Always try to set type="messages" to avoid the deprecation warning
+                if 'type' in init_params:
                     chatbot_kwargs["type"] = "messages"
+                    logger.info("Using 'messages' type for chatbot")
+                else:
+                    logger.warning("Chatbot 'type' parameter not supported, using default")
                 
                 # Check if 'show_copy_button' parameter is supported  
-                if 'show_copy_button' in gr.Chatbot.__init__.__code__.co_varnames:
+                if 'show_copy_button' in init_params:
                     chatbot_kwargs["show_copy_button"] = True
                     
                 # Check if 'scale' parameter is supported
-                if 'scale' in gr.Chatbot.__init__.__code__.co_varnames:
+                if 'scale' in init_params:
                     chatbot_kwargs["scale"] = 0
                     
             except Exception as e:
                 logger.warning(f"Error checking Chatbot parameters: {e}")
+                # Use minimal parameters as fallback, but try to set type to avoid warning
+                chatbot_kwargs = {"height": 400}
+                try:
+                    chatbot_kwargs["type"] = "messages"
+                except:
+                    pass
             
             chatbot = gr.Chatbot(**chatbot_kwargs)
             
@@ -631,8 +677,10 @@ if __name__ == "__main__":
             outputs=[current_prompt_state]
         )
         
-        # Sidebar with chat history
-        with gr.Sidebar() as sidebar:
+        # Chat History Sidebar (using simple approach for compatibility)
+        with gr.Column():
+            gr.Markdown("### Chat History")
+            
             @gr.render(inputs=[tab_edit_uuid_state, end_of_assistant_response_state, sidebar_names_state, current_uuid_state, chatbot, offloaded_tabs_data_storage])
             def render_chats(tab_uuid_edit, end_of_chat_response, sidebar_summaries, active_uuid, messages, tabs):
                 # Ensure sidebar_summaries is a dict
@@ -777,16 +825,19 @@ if __name__ == "__main__":
         
         # Check if newer ChatInterface parameters are supported
         try:
+            init_params = gr.ChatInterface.__init__.__code__.co_varnames
+            
             # Check if 'type' parameter is supported
-            if 'type' in gr.ChatInterface.__init__.__code__.co_varnames:
+            if 'type' in init_params:
                 chat_interface_kwargs["type"] = "messages"
                 
             # Check if 'multimodal' parameter is supported
-            if 'multimodal' in gr.ChatInterface.__init__.__code__.co_varnames:
+            if 'multimodal' in init_params:
                 chat_interface_kwargs["multimodal"] = multimodal
                 
         except Exception as e:
             logger.warning(f"Error checking ChatInterface parameters: {e}")
+            # Keep minimal parameters as fallback
         
         chat_interface = gr.ChatInterface(**chat_interface_kwargs)
 
@@ -831,13 +882,21 @@ if __name__ == "__main__":
         def setup_change_handler(fn, inputs, outputs, trigger_mode=None):
             """Helper function to set up change handlers with optional trigger_mode"""
             try:
-                if trigger_mode and 'trigger_mode' in chatbot.change.__code__.co_varnames:
+                # Get the change method's parameter names
+                change_params = chatbot.change.__code__.co_varnames
+                
+                if trigger_mode and 'trigger_mode' in change_params:
                     return chatbot.change(fn=fn, inputs=inputs, outputs=outputs, trigger_mode=trigger_mode)
                 else:
                     return chatbot.change(fn=fn, inputs=inputs, outputs=outputs)
             except Exception as e:
                 logger.warning(f"Error setting up change handler: {e}")
-                return chatbot.change(fn=fn, inputs=inputs, outputs=outputs)
+                # Fallback to basic change handler
+                try:
+                    return chatbot.change(fn=fn, inputs=inputs, outputs=outputs)
+                except Exception as fallback_error:
+                    logger.error(f"Failed to set up change handler: {fallback_error}")
+                    return None
         
         setup_change_handler(
             fn=populate_followup_questions,