Fix HF monitor import error and improve error handling

app.py

@@ -16,6 +16,7 @@ from core.errors import translate_error
 from core.personality import personality
 from src.analytics.user_logger import user_logger
 from src.analytics.session_analytics import session_analytics
+from src.llm.factory import llm_factory
 import logging
 
 # Set up logging
@@ -24,49 +25,25 @@ logger = logging.getLogger(__name__)
 
 st.set_page_config(page_title="CosmicCat AI Assistant", page_icon="🐱", layout="wide")
 
-# Initialize session state properly with error handling
-def initialize_session_state():
-    """Initialize all session state variables safely"""
-    session_vars = {
-        "messages": [],
-        "is_processing": False,
-        "ngrok_url_temp": st.session_state.get("ngrok_url", "https://7bcc180dffd1.ngrok-free.app"),
-        "cosmic_mode": True,
-        "show_welcome": True,
-        "last_processed_message": "",
-        "selected_model_value": "auto",
-        "session_id": f"sess_{int(time.time())}_{abs(hash(str(time.time()))) % 10000}"
-    }
-    
-    for var_name, default_value in session_vars.items():
-        if var_name not in st.session_state:
-            st.session_state[var_name] = default_value
-            logger.info(f"Initialized session variable: {var_name}")
+# Initialize session state properly
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+if "is_processing" not in st.session_state:
+    st.session_state.is_processing = False
+if "ngrok_url_temp" not in st.session_state:
+    st.session_state.ngrok_url_temp = st.session_state.get("ngrok_url", "https://7bcc180dffd1.ngrok-free.app")
+if "cosmic_mode" not in st.session_state:
+    st.session_state.cosmic_mode = True
+if "show_welcome" not in st.session_state:
+    st.session_state.show_welcome = True
+if "last_processed_message" not in st.session_state:
+    st.session_state.last_processed_message = ""
+if "session_id" not in st.session_state:
+    st.session_state.session_id = f"sess_{int(time.time())}_{abs(hash(str(time.time()))) % 10000}"
+if "selected_model_value" not in st.session_state:
+    st.session_state.selected_model_value = "auto"
 
-# Initialize session state
-try:
-    initialize_session_state()
-except Exception as e:
-    logger.error(f"Session state initialization failed: {e}")
-    # Fallback initialization
-    if "messages" not in st.session_state:
-        st.session_state.messages = []
-    if "is_processing" not in st.session_state:
-        st.session_state.is_processing = False
-    if "ngrok_url_temp" not in st.session_state:
-        st.session_state.ngrok_url_temp = st.session_state.get("ngrok_url", "https://7bcc180dffd1.ngrok-free.app")
-    if "cosmic_mode" not in st.session_state:
-        st.session_state.cosmic_mode = True
-    if "show_welcome" not in st.session_state:
-        st.session_state.show_welcome = True
-    if "last_processed_message" not in st.session_state:
-        st.session_state.last_processed_message = ""
-    if "selected_model_value" not in st.session_state:
-        st.session_state.selected_model_value = "auto"
-    if "session_id" not in st.session_state:
-        st.session_state.session_id = f"sess_{int(time.time())}_{abs(hash(str(time.time()))) % 10000}"
-
-# Start session tracking with error handling
+# Start session tracking
 try:
     session_analytics.start_session_tracking("default_user", st.session_state.session_id)
 except Exception as e:
@@ -99,10 +76,20 @@ with st.sidebar:
     # Show which provider will actually be used
     actual_provider = "Unknown"
     if st.session_state.selected_model_value == "auto":
-        if config.hf_token and hf_monitor.get_endpoint_status()["available"]:
-            actual_provider = "🤗 HF Endpoint"
-        elif config.ollama_host:
-            actual_provider = "🦙 Ollama"
+        # Import HF monitor here to avoid undefined error
+        try:
+            from src.services.hf_endpoint_monitor import hf_monitor
+            if config.hf_token:
+                status = hf_monitor.get_endpoint_status()
+                if status["available"]:
+                    actual_provider = "🤗 HF Endpoint"
+                elif config.ollama_host:
+                    actual_provider = "🦙 Ollama"
+            elif config.ollama_host:
+                actual_provider = "🦙 Ollama"
+        except:
+            if config.ollama_host:
+                actual_provider = "🦙 Ollama"
     else:
         actual_provider = "🤗 HF Endpoint" if st.session_state.selected_model_value == "huggingface" else "🦙 Ollama"
     
@@ -197,7 +184,7 @@ with st.sidebar:
         
         # HF Endpoint Status (Enhanced with initialization info)
         try:
-            from src.services.hf_monitor import hf_monitor
+            from src.services.hf_endpoint_monitor import hf_monitor
             status_message = hf_monitor.get_human_readable_status()
             
             # Display appropriate status icon
@@ -304,6 +291,7 @@ user_input = st.chat_input("Type your message here...", key="chat_input")
 
 # Process message when received
 if user_input and user_input.strip():
+    # Handle user message display first
     if not st.session_state.get('is_processing', False):
         chat_handler.process_user_message(user_input, st.session_state.selected_model_value)
     else: