...

app.py

@@ -1,38 +1,98 @@
 import streamlit as st
+import uuid
+import os
+from datetime import datetime
 from groq_api import query_groq
 from history import save_interaction, load_history
 
-st.set_page_config(page_title="Health Assistant", layout="centered")
+# Generate session-based cookie ID
+if "user_id" not in st.session_state:
+    st.session_state.user_id = str(uuid.uuid4())
 
-st.title("🩺 AI Health Assistant")
+# Profile info (name, age, gender)
+with st.sidebar:
+    st.header("👤 Your Info")
+    st.session_state.user_name = st.text_input("Name", st.session_state.get("user_name", ""))
+    st.session_state.user_age = st.number_input("Age", 10, 120, value=30)
+    st.session_state.user_gender = st.selectbox("Gender", ["Male", "Female", "Other"])
 
+# Load history for this session
 if "history" not in st.session_state:
-    st.session_state.history = load_history()
+    st.session_state.history = load_history(st.session_state.user_id)
+
+st.title("🩺 AI Health Assistant")
 
 mode = st.radio("Select mode", ["Symptom Checker", "Health Q&A"])
 
+def extract_keywords(symptoms: str) -> str:
+    prompt = f"Extract the main symptoms from this text as comma-separated keywords: {symptoms}"
+    keywords = query_groq(prompt)
+    return keywords.strip().strip(".,").lower()
+
 if mode == "Symptom Checker":
     symptoms = st.text_area("Enter your symptoms", placeholder="e.g., headache, fever, sore throat")
     if st.button("Analyze Symptoms"):
-        prompt = f"A user reports the following symptoms: {symptoms}. What could be the possible causes and precautions?"
+        dt = datetime.now().strftime("%Y-%m-%d %H:%M")
+        keywords = extract_keywords(symptoms)
+        prompt = (
+            f"A user named {st.session_state.user_name} (Age: {st.session_state.user_age}, "
+            f"Gender: {st.session_state.user_gender}) reports: {symptoms}. "
+            "List possible causes and precautions."
+        )
         response = query_groq(prompt)
         st.success(response)
-        save_interaction("Symptom Checker", symptoms, response)
-        st.session_state.history.append(("Symptom Checker", symptoms, response))
+
+        save_interaction(st.session_state.user_id, {
+            "mode": "Symptom Checker",
+            "datetime": dt,
+            "query": symptoms,
+            "keywords": keywords,
+            "response": response
+        })
+        st.session_state.history.append({
+            "mode": "Symptom Checker",
+            "datetime": dt,
+            "query": symptoms,
+            "keywords": keywords,
+            "response": response
+        })
 
 elif mode == "Health Q&A":
     question = st.text_input("Ask your health question")
     if st.button("Get Answer"):
-        prompt = f"User's health-related question: {question}. Provide an informative and medically-aligned answer."
+        dt = datetime.now().strftime("%Y-%m-%d %H:%M")
+        prompt = (
+            f"User named {st.session_state.user_name} (Age: {st.session_state.user_age}, "
+            f"Gender: {st.session_state.user_gender}) asked: {question}"
+        )
         response = query_groq(prompt)
         st.success(response)
-        save_interaction("Health Q&A", question, response)
-        st.session_state.history.append(("Health Q&A", question, response))
 
-# Show history as clickable expandable headers
+        save_interaction(st.session_state.user_id, {
+            "mode": "Health Q&A",
+            "datetime": dt,
+            "query": question,
+            "keywords": "",
+            "response": response
+        })
+        st.session_state.history.append({
+            "mode": "Health Q&A",
+            "datetime": dt,
+            "query": question,
+            "keywords": "",
+            "response": response
+        })
+
+# Display history
 with st.expander("📜 Show Previous Interactions"):
-    for mode, query, response in st.session_state.history[::-1]:
-        heading = f"[{mode}] {query[:80]}{'...' if len(query) > 80 else ''}"
-        with st.expander(heading):
-            st.markdown("**Diagnosis / Answer:**")
-            st.write(response)
+    if st.session_state.history:
+        for entry in st.session_state.history[::-1]:
+            if entry["mode"] == "Symptom Checker":
+                title = f"{entry['datetime']} | Symptoms: {entry['keywords']}"
+            else:
+                title = f"{entry['datetime']} | Question: {entry['query'][:50]}"
+            with st.expander(title):
+                st.markdown("**Diagnosis / Answer:**")
+                st.write(entry["response"])
+    else:
+        st.info("No history found.")

history.py

@@ -1,20 +1,21 @@
 import json
 import os
 
-HISTORY_FILE = "user_history.json"
+def get_user_history_file(user_id: str) -> str:
+    return f"user_history_{user_id}.json"
 
-def save_interaction(mode: str, query: str, response: str):
-    data = {"mode": mode, "query": query, "response": response}
-    history = load_history()
-    history.append((mode, query, response))
-    with open(HISTORY_FILE, "w") as f:
+def save_interaction(user_id: str, entry: dict):
+    history = load_history(user_id)
+    history.append(entry)
+    with open(get_user_history_file(user_id), "w") as f:
         json.dump(history, f)
 
-def load_history():
-    if not os.path.exists(HISTORY_FILE):
+def load_history(user_id: str):
+    file = get_user_history_file(user_id)
+    if not os.path.exists(file):
         return []
     try:
-        with open(HISTORY_FILE, "r") as f:
+        with open(file, "r") as f:
             return json.load(f)
     except Exception:
         return []