Update app.py

app.py

@@ -4,8 +4,30 @@ import streamlit as st
 from models.db import init_db
 init_db()
 
-# 2) Auth
+# ————— First-Run Signup Flow —————
+from repositories.user_repo import UserRepo
+from config.settings import settings
+
+repo = UserRepo(settings.database_url)
+if not repo.get_all_users():
+    st.title("🚀 Welcome to Quantum Healthcare AI")
+    st.warning("No users exist yet. Please create the first admin account:")
+    new_user = st.text_input("Username")
+    new_name = st.text_input("Full name")
+    new_pw   = st.text_input("Password", type="password")
+    if st.button("Create Admin User"):
+        if new_user and new_name and new_pw:
+            repo.add_user(new_user, new_name, new_pw)
+            st.success(f"Admin `{new_user}` created! Please refresh to log in.")
+        else:
+            st.error("All fields are required.")
+    # Stop here until admin is created
+    st.stop()
+# ————————————————————————————————
+
+# 2) Authentication (after at least one user exists)
 from services.auth import authenticator, require_login
+username = require_login()
 
 # 3) Core AI + Clinical NLP + Quantum
 from agent.gemini_agent import chat_with_gemini
@@ -17,8 +39,7 @@ from repositories.chat_repo import ChatRepo
 from services.logger import logger
 from services.metrics import CHAT_COUNT, OPTIMIZE_COUNT
 
-# ——— UI ———
-username = require_login()
+# === UI Setup ===
 st.set_page_config(page_title="Quantum Health AI", layout="wide")
 st.image("assets/logo.png", width=64)
 st.title(f"Welcome, {username}!")
@@ -26,7 +47,7 @@ st.title(f"Welcome, {username}!")
 tab1, tab2 = st.tabs(["🩺 Consult", "📊 Reports"])
 
 with tab1:
-    query = st.text_area("Describe symptoms or enter a medical question:", height=100)
+    query = st.text_area("Describe symptoms or ask a medical question:", height=100)
 
     if st.button("Ask Gemini"):
         CHAT_COUNT.inc()
@@ -37,12 +58,10 @@ with tab1:
         ChatRepo().save(user=username, prompt=query, response=response)
 
         with st.expander("🔎 UMLS Concept Lookup"):
-            umls_res = lookup_umls(query)
-            st.write(umls_res or "No UMLS concepts found.")
+            st.write(lookup_umls(query) or "No UMLS concepts found.")
 
         with st.expander("🔬 BioPortal Concept Lookup"):
-            bio_res = lookup_bioportal(query)
-            st.write(bio_res or "No BioPortal matches found.")
+            st.write(lookup_bioportal(query) or "No BioPortal matches found.")
 
     if st.button("Quantum Optimize Care Plan"):
         OPTIMIZE_COUNT.inc()
@@ -62,4 +81,4 @@ with tab2:
             st.download_button("Download PDF", f, file_name=pdf_path)
 
 st.markdown("---")
-st.caption("Powered by Gemini LLM • UMLS/BioPortal • Quantum-inspired optimization")    
+st.caption("Powered by Gemini LLM • UMLS/BioPortal • Quantum-inspired optimization")