54

app.py

@@ -24,7 +24,13 @@ class GradioArchitectureApp:
         self.assistant = None
         self.current_user_id = None
         self.conversation_history = []
-        self.state_manager = user_state_manager
+        
+        # Try to import state manager, but don't fail if it's not available
+        try:
+            from user_state_manager import user_state_manager
+            self.state_manager = user_state_manager
+        except ImportError:
+            self.state_manager = None
     
     def initialize_user_session(self, user_identifier: str = None, api_key: str = None):
         """Initialize user session with persistent state"""
@@ -60,7 +66,10 @@ class GradioArchitectureApp:
             self.assistant = ArchitectureAssistant(effective_api_key, self.current_user_id)
             
             # Try to load their latest conversation
-            history = self.state_manager.get_user_history(self.current_user_id)
+            if self.state_manager:
+                history = self.state_manager.get_user_history(self.current_user_id)
+            else:
+                history = []
             
             if history:
                 # Load most recent conversation
@@ -213,6 +222,9 @@ class GradioArchitectureApp:
         if not user_id.strip():
             return "Please enter a User ID to view history."
         
+        if not self.state_manager:
+            return "❌ User history not available - state manager not initialized."
+        
         try:
             history = self.state_manager.get_user_history(user_id.strip())
             
@@ -263,6 +275,9 @@ class GradioArchitectureApp:
     
     def get_all_users_summary(self):
         """Get summary of all users"""
+        if not self.state_manager:
+            return "❌ User management not available - state manager not initialized."
+            
         try:
             users = self.state_manager.get_all_users()