Add conversation history display and finalize Redis integration

api/chat.py

@@ -1,8 +1,8 @@
+import json
 from fastapi import APIRouter, HTTPException
 from fastapi.responses import StreamingResponse
 from core.llm import LLMClient
 from core.memory import save_user_state, load_user_state
-import json
 
 router = APIRouter()
 

app.py

@@ -1,6 +1,8 @@
 import streamlit as st
 from utils.config import config
 import requests
+import json
+from core.memory import load_user_state
 
 # Set page config
 st.set_page_config(page_title="AI Life Coach", page_icon="🧘", layout="centered")
@@ -19,6 +21,13 @@ def get_ollama_status():
     except Exception:
         return {"running": False, "model_loaded": None}
 
+# 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"])
+    return []
+
 ollama_status = get_ollama_status()
 
 # Display Ollama status
@@ -34,6 +43,13 @@ st.markdown("Talk to your personal development assistant.")
 if not ollama_status["running"]:
     st.warning("⚠️ Ollama is not running. Please start Ollama to use the AI Life Coach.")
 else:
+    # Display conversation history
+    conversation = get_conversation_history(user)
+    for msg in conversation:
+        role = msg["role"].capitalize()
+        content = msg["content"]
+        st.markdown(f"**{role}:** {content}")
+    
     # Chat input
     user_input = st.text_input("Your message...", key="input")
     if st.button("Send"):
@@ -51,7 +67,8 @@ else:
                         json={"user_id": user, "message": user_input}
                     )
                     if response.status_code == 200:
-                        ai_response = response.text
+                        response_data = response.json()
+                        ai_response = response_data.get("response", "")
                         st.markdown(f"**AI Coach:** {ai_response}")
                     else:
                         st.error("Failed to get response from AI Coach.")

core/memory.py

@@ -0,0 +1,29 @@
+import json
+import redis
+from utils.config import config
+
+# 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
+)
+
+def save_user_state(user_id: str, state: dict):
+    """Save user state to Redis"""
+    try:
+        redis_client.hset(f"user:{user_id}", mapping=state)
+        return True
+    except Exception as e:
+        print(f"Error saving user state: {e}")
+        return False
+
+def load_user_state(user_id: str):
+    """Load user state from Redis"""
+    try:
+        return redis_client.hgetall(f"user:{user_id}")
+    except Exception as e:
+        print(f"Error loading user state: {e}")
+        return {}