...

app.py

@@ -1,53 +1,84 @@
 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
-import os
+from db import init_connection
+from user import register_user, login_user, get_user_info
+from history import save_history, get_user_history
+from groq_api import get_groq_response  # Simulated or to be implemented
 
 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"])
+# Session state
+if "authenticated" not in st.session_state:
+    st.session_state.authenticated = False
+if "user_id" not in st.session_state:
+    st.session_state.user_id = ""
+
+st.title("🩺 Health Assistant")
+
+# -------------------------
+# Authentication
+# -------------------------
+tab1, tab2 = st.tabs(["🔐 Login", "🆕 Create Account"])
+
+with tab1:
+    st.subheader("Login")
     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:
-        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")
+    if st.button("Login"):
+        if login_user(conn, user_id, password):
+            st.session_state.authenticated = True
+            st.session_state.user_id = user_id
+            st.success("Login successful!")
+        else:
+            st.error("Invalid credentials.")
+
+with tab2:
+    st.subheader("Create Account")
+    new_user_id = st.text_input("Choose a User ID")
+    name = st.text_input("Full Name")
+    dob = st.date_input("Date of Birth", max_value=datetime.today())
+    gender = st.selectbox("Gender", ["Male", "Female", "Other"])
+    new_password = st.text_input("Set a Password", type="password")
+    if st.button("Register"):
+        success, msg = register_user(conn, new_user_id, name, dob, gender, new_password)
+        if success:
+            st.success(msg)
+        else:
+            st.error(msg)
 
+# -------------------------
 # 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.")
+# -------------------------
+if st.session_state.authenticated:
+    user_id = st.session_state.user_id
+    user = get_user_info(conn, user_id)
+    age = (datetime.today().date() - user['dob']).days // 365 if user['dob'] else "N/A"
+    
+    st.markdown(f"**👤 Name:** {user['name']}  \n**🎂 Age:** {age}  \n**⚧ Gender:** {user['gender']}")
+
+    st.divider()
+    st.subheader("📝 Symptom Checker")
+    symptoms = st.text_area("Describe your symptoms")
+
+    if st.button("Get Possible Causes"):
+        if symptoms.strip():
+            query = f"Symptoms: {symptoms}"
+            response = get_groq_response(query)
+            st.markdown("#### 🤖 Assistant Response")
+            st.write(response)
+            save_history(conn, user_id, query, response, symptoms)
+        else:
+            st.warning("Please enter your symptoms.")
+
+    st.divider()
+    st.subheader("📜 Your History")
+
+    history = get_user_history(conn, user_id)
+    if history:
+        for item in history:
+            with st.expander(f"{item['created_at'].strftime('%Y-%m-%d %H:%M')} — {item['symptoms']}"):
+                st.markdown(f"**Query:** {item['query']}")
+                st.markdown(f"**Response:** {item['response']}")
     else:
-        for h in history:
-            with st.expander(f"{h['timestamp']} | Symptoms: {h['symptoms']}"):
-                st.write(h['result'])
+        st.info("No history found.")

db.py

@@ -3,20 +3,3 @@ 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]

diagnosis.py

@@ -0,0 +1,17 @@
+from db import get_connection
+
+def save_history(user_id, symptoms, diagnosis):
+    with get_connection() as conn, conn.cursor() as cur:
+        cur.execute("INSERT INTO mb_history (user_id, symptoms, diagnosis) VALUES (%s, %s, %s)",
+                    (user_id, symptoms, diagnosis))
+        conn.commit()
+
+def get_history(user_id):
+    with get_connection() as conn, conn.cursor() as cur:
+        cur.execute("""
+            SELECT timestamp, symptoms, diagnosis
+            FROM mb_history
+            WHERE user_id = %s
+            ORDER BY timestamp DESC
+        """, (user_id,))
+        return cur.fetchall()

history.py

@@ -0,0 +1,18 @@
+def save_history(conn, user_id, query, response, symptoms):
+    cur = conn.cursor()
+    cur.execute("""
+        INSERT INTO mb_history (user_id, query, response, symptoms)
+        VALUES (%s, %s, %s, %s)
+    """, (user_id, query, response, symptoms))
+    conn.commit()
+
+def get_user_history(conn, user_id):
+    cur = conn.cursor()
+    cur.execute("""
+        SELECT query, response, symptoms, created_at
+        FROM mb_history
+        WHERE user_id = %s
+        ORDER BY created_at DESC
+    """, (user_id,))
+    rows = cur.fetchall()
+    return [{"query": r[0], "response": r[1], "symptoms": r[2], "created_at": r[3]} for r in rows]

prompts.py

@@ -0,0 +1,2 @@
+def health_prompt(symptoms):
+    return f"What are the possible causes and precautions for the symptoms: {symptoms}?"

qroq_api.py

@@ -0,0 +1,12 @@
+import os
+from groq import Groq
+
+client = Groq(api_key=os.getenv("GROQ_API_KEY"))
+
+def get_diagnosis(prompt):
+    response = client.chat.completions.create(
+        model="mixtral-8x7b-32768",
+        messages=[{"role": "user", "content": prompt}],
+        temperature=0.7,
+    )
+    return response.choices[0].message.content.strip()

user.py

@@ -0,0 +1,32 @@
+import bcrypt
+
+def register_user(conn, user_id, name, dob, gender, password):
+    cur = conn.cursor()
+    try:
+        cur.execute("SELECT user_id FROM mb_users WHERE user_id = %s", (user_id,))
+        if cur.fetchone():
+            return False, "User ID already exists."
+
+        hashed_pw = bcrypt.hashpw(password.encode(), bcrypt.gensalt()).decode()
+        cur.execute("""
+            INSERT INTO mb_users (user_id, name, dob, gender, password)
+            VALUES (%s, %s, %s, %s, %s)
+        """, (user_id, name, dob, gender, hashed_pw))
+        conn.commit()
+        return True, "Account created successfully."
+    except Exception as e:
+        return False, str(e)
+
+def login_user(conn, user_id, password):
+    cur = conn.cursor()
+    cur.execute("SELECT password FROM mb_users WHERE user_id = %s", (user_id,))
+    result = cur.fetchone()
+    if result and bcrypt.checkpw(password.encode(), result[0].encode()):
+        return True
+    return False
+
+def get_user_info(conn, user_id):
+    cur = conn.cursor()
+    cur.execute("SELECT name, dob, gender FROM mb_users WHERE user_id = %s", (user_id,))
+    result = cur.fetchone()
+    return {"name": result[0], "dob": result[1], "gender": result[2]} if result else {}

user_auth.py

@@ -1,28 +0,0 @@
-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 {}