...

app.py

@@ -1,152 +1,92 @@
 import streamlit as st
+from streamlit_cookies_manager import EncryptedCookieManager
 import uuid
+import json
 from datetime import datetime
-from groq_api import query_groq
-from history import save_interaction, load_history
-from streamlit_cookies_manager import EncryptedCookieManager
+from groq import Groq  # Ensure groq package is installed and configured
+import os
 
-# --- COOKIE MANAGER ---
+# Initialize cookies
 cookies = EncryptedCookieManager(
-    prefix="health_app_",
-    password="replace_this_with_a_random_very_long_key"
+    prefix="health_app", password=os.getenv("COOKIE_SECRET", "my_secret_password")
 )
 if not cookies.ready():
     st.stop()
 
-# --- INITIALIZATION ---
-# User ID
-if "user_id" not in st.session_state:
-    user_id = cookies.get("user_id")
-    if not user_id:
-        user_id = str(uuid.uuid4())
-        cookies["user_id"] = user_id
-    st.session_state.user_id = user_id
+# Session ID from cookie
+if not cookies.get("user_id"):
+    cookies["user_id"] = str(uuid.uuid4())
+    cookies["user_name"] = ""
+    cookies["user_age"] = "0"
+    cookies["user_sex"] = ""
 
-# User Info (name, age, gender)
-defaults = {"user_name": "", "user_age": 30, "user_gender": "Male"}
-for key, default in defaults.items():
-    if key not in st.session_state:
-        st.session_state[key] = cookies.get(key) or default
+user_id = cookies.get("user_id")
+user_name = cookies.get("user_name")
+user_age = cookies.get("user_age")
+user_sex = cookies.get("user_sex")
 
-# Load history
+# Initialize session history in session state
 if "history" not in st.session_state:
-    st.session_state.history = load_history(st.session_state.user_id)
-
-# --- PAGE LAYOUT ---
-st.set_page_config(page_title="Health Assistant", layout="centered")
-st.title("🩺 AI Health Assistant")
-
-# Sidebar: user info
-with st.sidebar:
-    st.header("👤 Your Info")
-    name = st.text_input("Name", st.session_state.user_name)
-    age = st.number_input("Age", 10, 120, value=st.session_state.user_age)
-    gender = st.selectbox("Gender", ["Male", "Female", "Other"], index=["Male", "Female", "Other"].index(st.session_state.user_gender))
-
-    # Save to cookies and session state
-    if name != st.session_state.user_name:
-        st.session_state.user_name = name
-        cookies["user_name"] = name
-    if age != st.session_state.user_age:
-        st.session_state.user_age = age
-        cookies["user_age"] = age
-    if gender != st.session_state.user_gender:
-        st.session_state.user_gender = gender
-        cookies["user_gender"] = gender
-
-cookies.save()
-
-# --- MAIN FUNCTIONALITY ---
-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"):
-        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)
-
-        entry = {
-            "mode": "Symptom Checker",
-            "datetime": dt,
-            "query": symptoms,
-            "keywords": keywords,
-            "response": response
-        }
-        save_interaction(st.session_state.user_id, entry)
-        st.session_state.history.append(entry)
-
-elif mode == "Health Q&A":
-    question = st.text_input("Ask your health question")
-    if st.button("Get Answer"):
-        dt = datetime.now().strftime("%Y-%m-%d %H:%M")
+    st.session_state.history = []
+
+# Page title
+st.title("Health Assistant")
+
+# Collect user info if not already stored
+with st.expander("👤 Personal Information", expanded=(not user_name or not user_age or not user_sex)):
+    name = st.text_input("Name", value=user_name)
+    age = st.number_input("Age", min_value=1, max_value=120, value=int(user_age) if user_age else 25)
+    sex = st.selectbox("Sex", ["Male", "Female", "Other"], index=["Male", "Female", "Other"].index(user_sex or "Male"))
+
+    if st.button("Save Information"):
+        cookies["user_name"] = str(name)
+        cookies["user_age"] = str(age)
+        cookies["user_sex"] = str(sex)
+        st.success("Information saved. You can now enter your symptoms.")
+
+# User input
+st.subheader("📝 Describe Your Symptoms")
+symptoms = st.text_area("Enter your symptoms (e.g., fever, sore throat, headache):")
+
+if st.button("Diagnose"):
+    if not symptoms:
+        st.warning("Please enter your symptoms.")
+    else:
         prompt = (
-            f"User named {st.session_state.user_name} (Age: {st.session_state.user_age}, "
-            f"Gender: {st.session_state.user_gender}) asked: {question}"
+            f"You are a medical assistant. A user named {name}, age {age}, sex {sex} reports the following symptoms:\n"
+            f"{symptoms}\n"
+            f"Please provide:\n1. Possible causes\n2. Recommended precautions."
         )
-        response = query_groq(prompt)
-        st.success(response)
-
-        entry = {
-            "mode": "Health Q&A",
-            "datetime": dt,
-            "query": question,
-            "keywords": "",
-            "response": response
-        }
-        save_interaction(st.session_state.user_id, entry)
-        st.session_state.history.append(entry)
 
-# --- HISTORY + DOWNLOAD ---
-def format_history_txt(history):
-    lines = []
-    for entry in history:
-        lines.append(f"Mode: {entry['mode']}")
-        lines.append(f"DateTime: {entry['datetime']}")
-        if entry['mode'] == "Symptom Checker":
-            lines.append(f"Symptoms: {entry['query']}")
-            lines.append(f"Keywords: {entry['keywords']}")
-        else:
-            lines.append(f"Question: {entry['query']}")
-        lines.append("Response:")
-        lines.append(entry["response"])
-        lines.append("-" * 50)
-    return "\n".join(lines)
-
-with st.expander("📜 Show Previous Interactions"):
-    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"])
-
-        st.download_button(
-            label="📄 Download My History",
-            data=format_history_txt(st.session_state.history),
-            file_name="my_health_history.txt",
-            mime="text/plain"
+        # Call Groq API (assumes GROQ_API_KEY and model is correctly configured)
+        client = Groq(api_key=os.getenv("GROQ_API_KEY"))
+        response = client.chat.completions.create(
+            model="llama3-70b-8192",
+            messages=[
+                {"role": "system", "content": "You are a helpful medical assistant."},
+                {"role": "user", "content": prompt}
+            ]
         )
-    else:
-        st.info("No history found.")
 
-# --- DISCLAIMER ---
-st.markdown("---")
-st.markdown(
-    "ℹ️ **Disclaimer:** This tool is an AI-powered assistant and does not replace professional medical advice. "
-    "Always consult a qualified healthcare provider for serious concerns or emergencies."
-)
+        diagnosis = response.choices[0].message.content.strip()
+        st.markdown("### 🧾 Diagnosis")
+        st.write(diagnosis)
+
+        # Extract keywords (basic method)
+        keywords = ", ".join(sym.strip() for sym in symptoms.split(",")[:4])
+
+        # Save to session history
+        st.session_state.history.append({
+            "timestamp": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
+            "symptoms": keywords,
+            "diagnosis": diagnosis
+        })
+
+# Display history
+if st.session_state.history:
+    st.markdown("### 📜 Previous History")
+    for i, item in reversed(list(enumerate(st.session_state.history))):
+        with st.expander(f"{item['timestamp']} | Symptoms: {item['symptoms']}"):
+            st.write(item["diagnosis"])
+else:
+    st.info("No previous history found.")