Fix session state initialization and improve Redis error handling

app.py

@@ -1,19 +1,24 @@
-# Force redeploy trigger - version 1.5
+# Force redeploy trigger - version 1.6
 import streamlit as st
 from utils.config import config
 import requests
 import json
 import os
-from core.memory import load_user_state
-from core.llm import LLMClient
+from core.memory import load_user_state, check_redis_health
 
 # Set page config
 st.set_page_config(page_title="AI Life Coach", page_icon="🧘", layout="centered")
 
 # Initialize session state for ngrok URL
-if "ngrok_url" not in st.session_state:
+if 'ngrok_url' not in st.session_state:
     st.session_state.ngrok_url = config.ollama_host
 
+if 'model_status' not in st.session_state:
+    st.session_state.model_status = "checking"
+
+if 'available_models' not in st.session_state:
+    st.session_state.available_models = []
+
 # Sidebar for user selection
 st.sidebar.title("🧘 AI Life Coach")
 user = st.sidebar.selectbox("Select User", ["Rob", "Sarah"])
@@ -24,6 +29,7 @@ st.sidebar.subheader("Ollama Connection")
 ngrok_input = st.sidebar.text_input("Ngrok URL", value=st.session_state.ngrok_url)
 if st.sidebar.button("Update Ngrok URL"):
     st.session_state.ngrok_url = ngrok_input
+    st.session_state.model_status = "checking"
     st.sidebar.success("Ngrok URL updated!")
     st.experimental_rerun()
 
@@ -39,13 +45,6 @@ NGROK_HEADERS = {
     "User-Agent": "AI-Life-Coach-App"
 }
 
-# Session state for model status
-if "model_status" not in st.session_state:
-    st.session_state.model_status = "checking"
-
-if "available_models" not in st.session_state:
-    st.session_state.available_models = []
-
 # Fetch Ollama status
 def get_ollama_status(ngrok_url):
     try:
@@ -110,9 +109,12 @@ def poll_model_status(ngrok_url):
 
 # After user selects name, load conversation history
 def get_conversation_history(user_id):
-    user_state = load_user_state(user_id)
-    if user_state and "conversation" in user_state:
-        return json.loads(user_state["conversation"])
+    try:
+        user_state = load_user_state(user_id)
+        if user_state and "conversation" in user_state:
+            return json.loads(user_state["conversation"])
+    except Exception as e:
+        st.warning(f"Could not load conversation history: {e}")
     return []
 
 # Check Ollama status with the current ngrok URL
@@ -145,6 +147,13 @@ elif st.session_state.model_status == "different_models":
 else:  # unreachable
     model_status_container.error("❌ Ollama unreachable")
 
+# Redis status indicator
+redis_status_container = st.sidebar.empty()
+if check_redis_health():
+    redis_status_container.success("✅ Redis Connected")
+else:
+    redis_status_container.warning("⚠️ Redis Not Available")
+
 # Main chat interface
 st.title("🧘 AI Life Coach")
 st.markdown("Talk to your personal development assistant.")
@@ -160,6 +169,7 @@ with st.expander("🔍 Connection Status"):
     st.write("- Configured Ollama Host:", config.ollama_host)
     st.write("- Current Ngrok URL:", st.session_state.ngrok_url)
     st.write("- Using Fallback:", use_fallback)
+    st.write("- Redis Health:", check_redis_health())
 
 # Function to send message to Ollama
 def send_to_ollama(user_input, conversation_history, ngrok_url):
@@ -191,6 +201,9 @@ def send_to_ollama(user_input, conversation_history, ngrok_url):
 # Function to send message to Hugging Face (fallback)
 def send_to_hf(user_input, conversation_history):
     try:
+        # Import here to avoid issues if not needed
+        from core.llm import LLMClient
+        
         # Initialize LLM client for Hugging Face
         llm_client = LLMClient(provider="huggingface")
         

core/memory.py

@@ -1,18 +1,57 @@
 import json
-import redis
+import time
 from utils.config import config
 
+# Try to import redis, but handle if not available
+try:
+    import redis
+    REDIS_AVAILABLE = True
+except ImportError:
+    REDIS_AVAILABLE = False
+    redis = None
+
+def get_redis_client():
+    """Create and return Redis client with retry logic"""
+    if not REDIS_AVAILABLE:
+        return None
+        
+    last_exception = None
+    
+    for attempt in range(config.redis_retries + 1):
+        try:
+            redis_client = redis.Redis(
+                host=config.redis_host,
+                port=config.redis_port,
+                username=config.redis_username,
+                password=config.redis_password,
+                decode_responses=True,
+                socket_connect_timeout=5,
+                socket_timeout=5
+            )
+            # Test the connection
+            redis_client.ping()
+            return redis_client
+        except Exception as e:
+            last_exception = e
+            if attempt < config.redis_retries:
+                time.sleep(config.redis_retry_delay * (2 ** attempt))  # Exponential backoff
+            continue
+    
+    return None
+
 # Initialize Redis connection
-redis_client = redis.Redis(
-    host=config.redis_host,
-    port=config.redis_port,
-    username=config.redis_username,
-    password=config.redis_password,
-    decode_responses=True
-)
+redis_client = None
+try:
+    redis_client = get_redis_client()
+except Exception as e:
+    print(f"Warning: Could not connect to Redis: {e}")
 
 def save_user_state(user_id: str, state: dict):
-    """Save user state to Redis"""
+    """Save user state to Redis with retry logic"""
+    if not redis_client:
+        print("Redis not available, skipping save_user_state")
+        return False
+        
     try:
         redis_client.hset(f"user:{user_id}", mapping=state)
         return True
@@ -21,9 +60,23 @@ def save_user_state(user_id: str, state: dict):
         return False
 
 def load_user_state(user_id: str):
-    """Load user state from Redis"""
+    """Load user state from Redis with retry logic"""
+    if not redis_client:
+        print("Redis not available, returning empty state")
+        return {}
+        
     try:
         return redis_client.hgetall(f"user:{user_id}")
     except Exception as e:
         print(f"Error loading user state: {e}")
         return {}
+
+def check_redis_health():
+    """Check if Redis is healthy"""
+    if not redis_client:
+        return False
+    try:
+        redis_client.ping()
+        return True
+    except:
+        return False