Update app.py

app.py

@@ -1,36 +1,60 @@
-
 import streamlit as st
-import requests
-import pandas as pd
-
-st.set_page_config(page_title="ZeroSearch Medical Q&A", layout="wide")
-st.title("🩺 ZeroSearch Medical Q&A – Google‑Free Clinical Answers")
-
-API_URL = st.secrets.get("API_URL", "http://localhost:8000/ask")
-
-query = st.text_input(
-    "Enter a clinical question:",
-    placeholder="e.g. What are the first‑line treatments for hypertension in pregnancy?",
-)
-
-if st.button("Search") and query:
-    with st.spinner("Consulting internal medical knowledge…"):
-        resp = requests.post(API_URL, json={"question": query})
-    if resp.status_code == 200:
-        data = resp.json()
+from agent.gemini_agent import run_gemini_chat
+from agent.clinical_nlp import umls_concept_lookup, bioportal_concept_lookup
+from quantum.optimizer import optimize_treatment
+
+st.set_page_config(page_title="Quantum Healthcare AI", layout="wide")
+
+st.markdown("""
+    <div style='display:flex;align-items:center'>
+        <img src='https://cdn-icons-png.flaticon.com/512/891/891419.png' width='56' style='margin-right:15px;'/>
+        <h1 style='display:inline'>Quantum-AI Healthcare Assistant</h1>
+    </div>
+    """, unsafe_allow_html=True)
+
+st.write("**Ask anything about symptoms, conditions, diagnosis, or optimal care. Get instant LLM answers, clinical concept mapping, and even quantum-inspired optimization.**")
+
+query = st.text_area("Describe symptoms, a clinical problem, or a medical question:", height=70)
+
+cols = st.columns([1,1])
+with cols[0]:
+    get_ai = st.button("Get AI & Clinical Insight")
+with cols[1]:
+    run_quantum = st.button("Quantum Optimize Care Plan")
+
+if get_ai and query.strip():
+    with st.spinner("Gemini AI is reviewing your query..."):
+        ai_response = run_gemini_chat(query)
+    st.markdown(f"**Gemini Agent:**\n\n{ai_response}")
+
+    st.markdown("---")
+    with st.expander("🔎 UMLS Concept Mapping"):
+        concepts = umls_concept_lookup(query)
+        if concepts:
+            for c in concepts:
+                if 'error' in c:
+                    st.error(c['error'])
+                else:
+                    st.write(f"**{c['name']}**  |  ID: `{c['ui']}`  |  Source: *{c['rootSource']}*")
+        else:
+            st.info("No UMLS concepts found.")
+
+    with st.expander("🔬 BioPortal Ontology Lookup"):
+        concepts = bioportal_concept_lookup(query)
+        if concepts:
+            for c in concepts:
+                if 'error' in c:
+                    st.error(c['error'])
+                else:
+                    st.write(f"**{c['prefLabel']}**  |  [Ontology]({c['ontology']})  |  [Concept ID]({c['id']})")
+        else:
+            st.info("No BioPortal concepts found.")
 
-        st.subheader("Answer")
-        st.write(data["answer"])
+if run_quantum and query.strip():
+    with st.spinner("Quantum optimizer working..."):
+        plan = optimize_treatment(query)
+    st.markdown("### 🧬 Optimized Care Plan")
+    st.json(plan)
 
-        st.subheader("UMLS Concepts")
-        if data["umls"]:
-            st.dataframe(pd.DataFrame(data["umls"]))
-        else:
-            st.info("No UMLS concepts detected.")
-
-        st.subheader("Simulated Search Documents")
-        for i, doc in enumerate(data["docs"], 1):
-            with st.expander(f"Document {i}"):
-                st.write(doc)
-    else:
-        st.error(f"API error: {resp.status_code} – {resp.text}")
+st.markdown("---")
+st.caption("Powered by Gemini LLM, UMLS, BioPortal, and quantum-inspired algorithms. For research use only. No patient data is stored.")