Add debug panel to sidebar and implement error handling for LLM responses

app.py

@@ -8,12 +8,15 @@ sys.path.append(str(Path(__file__).parent))
 from utils.config import config
 from core.llm import send_to_ollama, send_to_hf
 from core.session import session_manager
+from core.memory import check_redis_health
 
 st.set_page_config(page_title="AI Life Coach", page_icon="🧠", layout="wide")
 
 # Initialize session state
 if "messages" not in st.session_state:
     st.session_state.messages = []
+if "last_error" not in st.session_state:
+    st.session_state.last_error = ""
 
 # Sidebar
 with st.sidebar:
@@ -47,6 +50,17 @@ with st.sidebar:
     if st.button("Clear History"):
         st.session_state.messages = []
         st.success("History cleared!")
+    
+    # Debug info
+    with st.sidebar.expander("🔧 Debug Info"):
+        st.write(f"**OLLAMA_HOST**: `{st.session_state.ngrok_url}`")
+        st.write(f"**Selected Model**: `{st.session_state.selected_model}`")
+        st.write(f"**Fallback Mode**: {'✅ On' if config.use_fallback else '❌ Off'}")
+        st.write(f"**Redis Status**: {'✅ Healthy' if check_redis_health() else '⚠️ Unavailable'}")
+        st.write(f"**Env Detected As**: {'☁️ HF Space' if config.is_hf_space else '🏠 Local'}")
+        st.write(f"**HF Token Set**: {'✅ Yes' if config.hf_token else '❌ No'}")
+        if st.session_state.last_error:
+            st.warning(f"Last Error: {st.session_state.last_error}")
 
 # Main chat interface
 st.title("🧠 AI Life Coach")
@@ -78,6 +92,9 @@ if send_button and user_input.strip():
     # Add user message to history
     st.session_state.messages.append({"role": "user", "content": user_input})
     
+    # Reset error state
+    st.session_state.last_error = ""
+    
     # Get conversation history
     user_session = session_manager.get_session("default_user")
     conversation = user_session.get("conversation", [])
@@ -89,26 +106,33 @@ if send_button and user_input.strip():
         with st.spinner("AI Coach is thinking..."):
             ai_response = None
             backend_used = ""
+            error_msg = ""
             
             # Try Ollama first if not falling back
             if not config.use_fallback:
-                ai_response = send_to_ollama(
-                    user_input,
-                    conversation_history,
-                    st.session_state.ngrok_url,
-                    st.session_state.selected_model
-                )
-                backend_used = "Ollama"
+                try:
+                    ai_response = send_to_ollama(
+                        user_input,
+                        conversation_history,
+                        st.session_state.ngrok_url,
+                        st.session_state.selected_model
+                    )
+                    backend_used = "Ollama"
+                except Exception as e:
+                    error_msg = f"Ollama error: {str(e)}"
             
             # Fallback to Hugging Face
             if not ai_response and config.hf_token:
-                ai_response = send_to_hf(user_input, conversation_history)
-                backend_used = "Hugging Face"
+                try:
+                    ai_response = send_to_hf(user_input, conversation_history)
+                    backend_used = "Hugging Face"
+                except Exception as e:
+                    error_msg = f"Hugging Face error: {str(e)}"
             
             if ai_response:
                 st.markdown(f"{ai_response}")
                 
-                # Update conversation history (stub – actual save will come later)
+                # Update conversation history
                 conversation.append({"role": "user", "content": user_input})
                 conversation.append({"role": "assistant", "content": ai_response})
                 
@@ -120,6 +144,7 @@ if send_button and user_input.strip():
                 st.session_state.messages.append({"role": "assistant", "content": ai_response})
             else:
                 st.error("Failed to get response from both providers.")
+                st.session_state.last_error = error_msg or "No response from either provider"
     
     # Clear input by forcing rerun
     st.experimental_rerun()