Switch to db.

app.py

@@ -1,77 +1,53 @@
 import streamlit as st
+from db import init_connection, get_history_for_user, save_history
+from user_auth import login_user, register_user, get_user_info
 from datetime import datetime
-from streamlit_cookies_manager import EncryptedCookieManager
-from uuid import uuid4
-import hashlib
-
-# Ensure secret is set
-cookies = EncryptedCookieManager(prefix="health_app_", password="your_cookie_secret_here")
-if not cookies.ready():
-    st.stop()
-
-# Generate or retrieve session ID
-user_id = cookies.get("user_id")
-if not user_id:
-    user_id = hashlib.sha256(str(uuid4()).encode()).hexdigest()
-    cookies["user_id"] = user_id
-
-# Retrieve saved details
-user_name = cookies.get("user_name")
-user_gender = cookies.get("user_gender")
-user_age = cookies.get("user_age")
-
-st.title("🩺 AI Health Assistant")
-
-# User Info Form
-with st.expander("👤 User Information", expanded=not all([user_name, user_gender, user_age])):
-    name = st.text_input("Name", value=user_name or "")
-    gender = st.selectbox("Gender", ["Male", "Female", "Other"], index=["Male", "Female", "Other"].index(user_gender) if user_gender else 0)
-
-    # Safe age default
-    default_age = int(user_age) if user_age and user_age.isdigit() and 1 <= int(user_age) <= 120 else 25
-    age = st.number_input("Age", min_value=1, max_value=120, value=default_age)
-
-    if st.button("Save Info"):
-        cookies["user_name"] = name
-        cookies["user_gender"] = gender
-        cookies["user_age"] = str(age)
-        st.success("User info saved!")
-
-# Session storage for history
-if "history" not in st.session_state:
-    st.session_state.history = []
-
-st.markdown("---")
-st.subheader("📝 Describe your symptoms:")
-
-user_input = st.text_area("Enter symptoms", height=100)
-
-if st.button("Diagnose"):
-    if not user_input.strip():
-        st.warning("Please enter some symptoms.")
+import os
+
+st.set_page_config(page_title="Health Assistant", layout="centered")
+conn = init_connection()
+
+# Sidebar - Login/Register
+with st.sidebar:
+    st.title("🔐 User Login")
+    action = st.radio("Choose action:", ["Login", "Register"])
+    user_id = st.text_input("User ID")
+    password = st.text_input("Password", type="password")
+    if action == "Register":
+        name = st.text_input("Full Name")
+        age = st.number_input("Age", min_value=1, max_value=120, step=1)
+        sex = st.selectbox("Sex", ["Male", "Female", "Other"])
+        if st.button("Register"):
+            success, msg = register_user(conn, user_id, password, name, age, sex)
+            st.info(msg)
     else:
-        now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-        # Simulated diagnosis result
-        diagnosis = f"Based on symptoms, possible cause: 'Common cold' or 'Flu'. Please consult a doctor for confirmation."
-
-        # Extract keywords (simple placeholder logic)
-        keywords = ", ".join(user_input.lower().replace('.', '').replace(',', '').split()[:5])
-
-        st.session_state.history.append({
-            "timestamp": now,
-            "symptoms": keywords,
-            "diagnosis": diagnosis
-        })
-
-        st.success("✅ Diagnosis complete!")
-        st.write(diagnosis)
-
-st.markdown("---")
-st.subheader("📚 Previous History")
-
-if not st.session_state.history:
-    st.info("No previous history available.")
-else:
-    for idx, entry in reversed(list(enumerate(st.session_state.history))):
-        with st.expander(f"{entry['timestamp']} | Symptoms: {entry['symptoms']}"):
-            st.write(entry["diagnosis"])
+        if st.button("Login"):
+            if login_user(conn, user_id, password):
+                st.session_state.user_id = user_id
+                st.success("Login successful")
+            else:
+                st.error("Invalid credentials")
+
+# Main App Logic
+if "user_id" in st.session_state:
+    user_info = get_user_info(conn, st.session_state.user_id)
+    st.markdown(f"### 👤 Welcome, {user_info['name']} ({user_info['sex']}, {user_info['age']} yrs)")
+
+    st.header("💬 Symptom Checker")
+    query = st.text_area("Describe your symptoms")
+    if st.button("Analyze Symptoms"):
+        if query.strip():
+            keywords = [word.strip(".,") for word in query.split() if len(word) > 4]
+            result = f"Possible causes for symptoms like: {', '.join(keywords[:5])}..."
+            timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+            save_history(conn, st.session_state.user_id, timestamp, ", ".join(keywords[:5]), result)
+            st.success(result)
+
+    st.markdown("### 📜 Previous History")
+    history = get_history_for_user(conn, st.session_state.user_id)
+    if not history:
+        st.info("No previous history found.")
+    else:
+        for h in history:
+            with st.expander(f"{h['timestamp']} | Symptoms: {h['symptoms']}"):
+                st.write(h['result'])

db.py

@@ -0,0 +1,22 @@
+import psycopg2
+import os
+
+def init_connection():
+    return psycopg2.connect(os.environ["DB_CONNECT"])
+
+def save_history(conn, user_id, timestamp, symptoms, result):
+    with conn.cursor() as cur:
+        cur.execute("""
+            INSERT INTO mb_history (user_id, timestamp, symptoms, result)
+            VALUES (%s, %s, %s, %s)
+        """, (user_id, timestamp, symptoms, result))
+        conn.commit()
+
+def get_history_for_user(conn, user_id):
+    with conn.cursor() as cur:
+        cur.execute("""
+            SELECT timestamp, symptoms, result FROM mb_history
+            WHERE user_id = %s ORDER BY timestamp DESC
+        """, (user_id,))
+        rows = cur.fetchall()
+        return [{"timestamp": r[0], "symptoms": r[1], "result": r[2]} for r in rows]

groq_api.py

@@ -1,21 +0,0 @@
-import os
-import requests
-
-GROQ_API_KEY = os.getenv("GROQ_API_KEY")
-MODEL = "llama3-70b-8192"  # or "mixtral-8x7b-32768"
-
-def query_groq(prompt: str) -> str:
-    url = "https://api.groq.com/openai/v1/chat/completions"
-    headers = {
-        "Authorization": f"Bearer {GROQ_API_KEY}",
-        "Content-Type": "application/json"
-    }
-    payload = {
-        "model": MODEL,
-        "messages": [{"role": "user", "content": prompt}],
-        "temperature": 0.6
-    }
-
-    res = requests.post(url, headers=headers, json=payload)
-    res.raise_for_status()
-    return res.json()["choices"][0]["message"]["content"]

history.py

@@ -1,21 +0,0 @@
-import json
-import os
-
-def get_user_history_file(user_id: str) -> str:
-    return f"user_history_{user_id}.json"
-
-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(user_id: str):
-    file = get_user_history_file(user_id)
-    if not os.path.exists(file):
-        return []
-    try:
-        with open(file, "r") as f:
-            return json.load(f)
-    except Exception:
-        return []

prompts.py

@@ -1,12 +0,0 @@
-SYMPTOM_PROMPT_TEMPLATE = """
-You are an experienced medical assistant. A patient reports the following symptoms: {symptoms}.
-Please list possible causes and suggest general precautions.
-Avoid giving any direct diagnosis. Keep your answer informative, clear, and responsible.
-"""
-
-QUESTION_PROMPT_TEMPLATE = """
-You are an AI medical assistant trained in clinical and health knowledge.
-Answer the following health-related question in a responsible, factual, and non-diagnostic manner.
-
-Question: {question}
-"""

requirements.txt

@@ -1,4 +1,6 @@
 streamlit
-requests
-streamlit-cookies-manager
+psycopg2-binary
+sqlalchemy
+bcrypt
+python-dotenv
 groq

user_auth.py

@@ -0,0 +1,28 @@
+import hashlib
+
+def hash_password(password):
+    return hashlib.sha256(password.encode()).hexdigest()
+
+def login_user(conn, user_id, password):
+    with conn.cursor() as cur:
+        cur.execute("SELECT password FROM mb_users WHERE user_id = %s", (user_id,))
+        row = cur.fetchone()
+        return row and row[0] == hash_password(password)
+
+def register_user(conn, user_id, password, name, age, sex):
+    with conn.cursor() as cur:
+        cur.execute("SELECT user_id FROM mb_users WHERE user_id = %s", (user_id,))
+        if cur.fetchone():
+            return False, "User ID already exists."
+        cur.execute("""
+            INSERT INTO mb_users (user_id, password, name, age, sex)
+            VALUES (%s, %s, %s, %s, %s)
+        """, (user_id, hash_password(password), name, age, sex))
+        conn.commit()
+        return True, "Registered successfully."
+
+def get_user_info(conn, user_id):
+    with conn.cursor() as cur:
+        cur.execute("SELECT name, age, sex FROM mb_users WHERE user_id = %s", (user_id,))
+        row = cur.fetchone()
+        return {"name": row[0], "age": row[1], "sex": row[2]} if row else {}