Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,18 +1,4 @@
 
-"""
-Shinui Streamlit App - Futuristic Dark Theme (Option B)
-- Gemeni model set to gemini-2.5-pro
-- Landing page, About (pre-auth), Auth, Dashboard
-- Neon/dark UI + Lottie animations
-- Compatible with Hugging Face Spaces (Streamlit)
-
-Files included in this single code file:
-- The main Streamlit app code below
-- requirements.txt contents appended at bottom (commented)
-
-Save this as `shinui_app.py` when deploying to Spaces.
-"""
-
 import streamlit as st
 import os
 import sqlite3
@@ -25,19 +11,36 @@ import requests
 from streamlit_lottie import st_lottie
 
 # -----------------------------------------------------------------------------
-# AUTO-CONFIG (Hugging Face / Spaces friendly)
+# 0. CONFIGURATION & SETUP
 # -----------------------------------------------------------------------------
+st.set_page_config(
+    page_title="SHINUI | Medical Intelligence", 
+    page_icon="🧬", 
+    layout="wide",
+    initial_sidebar_state="expanded"
+)
+
+# Fix Upload Error (Auto-Config)
 config_dir = ".streamlit"
 if not os.path.exists(config_dir):
     os.makedirs(config_dir)
 with open(os.path.join(config_dir, "config.toml"), "w") as f:
     f.write("[server]\nenableXsrfProtection=false\nenableCORS=false\nmaxUploadSize=200\n")
 
+# API Setup
+GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
+if not GOOGLE_API_KEY:
+    st.error("⚠️ System Error: Google API Key is missing in Secrets.")
+    st.stop()
+
+genai.configure(api_key=GOOGLE_API_KEY)
+model = genai.GenerativeModel('gemini-1.5-pro')
+
 # -----------------------------------------------------------------------------
-# DATABASE SETUP
+# 1. DATABASE & AUTH SYSTEM
 # -----------------------------------------------------------------------------
 def init_db():
-    conn = sqlite3.connect('users.db')
+    conn = sqlite3.connect('shinui_users.db')
     c = conn.cursor()
     c.execute('''CREATE TABLE IF NOT EXISTS users
                  (email TEXT PRIMARY KEY, password TEXT, joined_date TEXT)''')
@@ -46,13 +49,11 @@ def init_db():
     conn.commit()
     conn.close()
 
-
 def hash_pass(password):
     return hashlib.sha256(str.encode(password)).hexdigest()
 
-
 def register_user(email, password):
-    conn = sqlite3.connect('users.db')
+    conn = sqlite3.connect('shinui_users.db')
     c = conn.cursor()
     try:
         c.execute("INSERT INTO users VALUES (?, ?, ?)", (email, hash_pass(password), datetime.now().strftime("%Y-%m-%d")))
@@ -63,437 +64,404 @@ def register_user(email, password):
     finally:
         conn.close()
 
-
 def login_user(email, password):
-    conn = sqlite3.connect('users.db')
+    conn = sqlite3.connect('shinui_users.db')
     c = conn.cursor()
     c.execute("SELECT * FROM users WHERE email=? AND password=?", (email, hash_pass(password)))
     data = c.fetchall()
     conn.close()
     return data
 
-
 def add_history(email, type, summary):
-    conn = sqlite3.connect('users.db')
+    conn = sqlite3.connect('shinui_users.db')
     c = conn.cursor()
+    # Store only the first 80 chars of summary for the list
+    clean_sum = summary.replace("|||", " ").replace("*", "")[:80] + "..."
     c.execute("INSERT INTO history (email, type, summary, date) VALUES (?, ?, ?, ?)", 
-              (email, type, summary, datetime.now().strftime("%Y-%m-%d %H:%M")))
+              (email, type, clean_sum, datetime.now().strftime("%Y-%m-%d %H:%M")))
     conn.commit()
     conn.close()
 
-
 def get_history(email):
-    conn = sqlite3.connect('users.db')
+    conn = sqlite3.connect('shinui_users.db')
     c = conn.cursor()
     c.execute("SELECT type, summary, date FROM history WHERE email=? ORDER BY id DESC LIMIT 10", (email,))
     data = c.fetchall()
     conn.close()
     return data
 
-# Init DB
 init_db()
 
 # -----------------------------------------------------------------------------
-# AI CONFIGURATION
+# 2. UTILITIES (Lottie & Navigation)
 # -----------------------------------------------------------------------------
-st.set_page_config(page_title="Shinui | Medical MVP", page_icon="✨", layout="wide")
-
-GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
-if not GOOGLE_API_KEY:
-    st.error("⚠️ System Error: Google API Key Missing. Set the environment variable GOOGLE_API_KEY in Spaces.")
-    st.stop()
-
-# Configure genai
-genai.configure(api_key=GOOGLE_API_KEY)
-# Use the model the user asked for
-model = genai.GenerativeModel('gemini-2.5-pro')
-
-# -----------------------------------------------------------------------------
-# STATE MANAGEMENT
-# -----------------------------------------------------------------------------
-if 'page' not in st.session_state: st.session_state.page = 'landing'
-if 'logged_in' not in st.session_state: st.session_state.logged_in = False
-if 'user_email' not in st.session_state: st.session_state.user_email = ""
-if 'current_result' not in st.session_state: st.session_state.current_result = None
-
-# -----------------------------------------------------------------------------
-# ASSET HELPERS (Lottie)
-# -----------------------------------------------------------------------------
-def load_lottie_url(url: str):
-    try:
-        r = requests.get(url)
-        if r.status_code == 200:
-            return r.json()
-        return None
-    except Exception:
+def load_lottieurl(url):
+    r = requests.get(url)
+    if r.status_code != 200:
         return None
+    return r.json()
 
-# Example Lottie URLs (replace if you have better assets)
-LOTTIE_HERO = "https://assets2.lottiefiles.com/packages/lf20_jtbfg2nb.json"
-LOTTIE_MEDICAL = "https://assets4.lottiefiles.com/packages/lf20_x62chJ.json"
-LOTTIE_LOADING = "https://assets6.lottiefiles.com/packages/lf20_usmfx6bp.json"
+def nav_to(page):
+    st.session_state.page = page
+    st.rerun()
 
-lottie_hero = load_lottie_url(LOTTIE_HERO)
-lottie_med = load_lottie_url(LOTTIE_MEDICAL)
-lottie_load = load_lottie_url(LOTTIE_LOADING)
+def sign_out():
+    st.session_state.logged_in = False
+    st.session_state.user_email = ""
+    st.session_state.current_result = None
+    nav_to('landing')
 
 # -----------------------------------------------------------------------------
-# UI STYLING (Futuristic Dark + Neon)
+# 3. THE "META-STYLE" CSS
 # -----------------------------------------------------------------------------
 st.markdown("""
 <style>
     @import url('https://fonts.googleapis.com/css2?family=Inter:wght@300;400;600;800&display=swap');
-    :root{
-      --bg: #0b1220;
-      --card: #0f1724;
-      --muted: #94a3b8;
-      --accent: #00f5ff;
-      --accent-2: #7c3aed;
-      --glass: rgba(255,255,255,0.03);
-    }
+    
+    /* GLOBAL THEME */
     .stApp {
-        background: linear-gradient(180deg, var(--bg) 0%, #071024 100%);
-        color: #e6eef8;
+        background-color: #09090b; /* Deep Zinc Black */
         font-family: 'Inter', sans-serif;
+        color: #e4e4e7;
+    }
+    
+    /* HEADERS */
+    h1, h2, h3 {
+        font-weight: 800;
+        letter-spacing: -0.5px;
+        color: #ffffff;
+    }
+    
+    /* CARDS */
+    .meta-card {
+        background-color: #18181b;
+        border: 1px solid #27272a;
+        border-radius: 12px;
+        padding: 24px;
+        transition: transform 0.2s ease, box-shadow 0.2s ease;
+    }
+    .meta-card:hover {
+        border-color: #3b82f6;
+        box-shadow: 0 4px 20px rgba(59, 130, 246, 0.1);
+    }
+    
+    /* RESULT BOXES (The Clean Look) */
+    .res-header {
+        font-size: 0.85rem;
+        text-transform: uppercase;
+        letter-spacing: 1px;
+        font-weight: 600;
+        margin-bottom: 10px;
+        display: flex;
+        align-items: center;
+        gap: 8px;
+    }
+    .box-obs { border-left: 3px solid #3b82f6; background: #111113; padding: 20px; border-radius: 0 8px 8px 0; margin-bottom: 15px; }
+    .box-risk { border-left: 3px solid #f59e0b; background: #111113; padding: 20px; border-radius: 0 8px 8px 0; margin-bottom: 15px; }
+    .box-act { border-left: 3px solid #10b981; background: #111113; padding: 20px; border-radius: 0 8px 8px 0; margin-bottom: 15px; }
+    
+    /* BUTTONS */
+    div.stButton > button {
+        background: linear-gradient(to right, #2563eb, #3b82f6);
+        color: white;
+        border: none;
+        border-radius: 8px;
+        padding: 10px 24px;
+        font-weight: 600;
+        width: 100%;
+        transition: all 0.3s;
+    }
+    div.stButton > button:hover {
+        opacity: 0.9;
+        transform: scale(1.01);
+    }
+    
+    /* INPUTS */
+    .stTextInput input, .stTextArea textarea {
+        background-color: #18181b !important;
+        border: 1px solid #27272a !important;
+        color: white !important;
+        border-radius: 8px;
     }
-    /* Cards */
-    .shinui-card{ background: linear-gradient(180deg, rgba(255,255,255,0.02), rgba(255,255,255,0.01)); border:1px solid rgba(255,255,255,0.04); border-radius:12px; padding:18px; }
-    .hero-cta{ background: linear-gradient(90deg, var(--accent), var(--accent-2)); padding:12px 22px; border-radius:10px; color:#001219; font-weight:700; }
-    /* Buttons */
-    div.stButton > button{ background:transparent; border:1px solid rgba(255,255,255,0.06); padding:10px 16px; border-radius:10px; color:var(--accent); font-weight:700; }
-    div.stButton > button:hover{ transform:translateY(-2px); box-shadow:0 10px 30px rgba(124,58,237,0.12); }
-    /* Inputs */
-    .stTextInput input, .stTextArea textarea{ background: rgba(255,255,255,0.02) !important; color: #e6eef8 !important; border:1px solid rgba(255,255,255,0.04) !important; }
-    /* Sidebar */
-    [data-testid="stSidebar"]{ background: linear-gradient(180deg, rgba(124,58,237,0.06), rgba(0,245,255,0.02)); border-right:1px solid rgba(255,255,255,0.02); }
-    /* Result badges */
-    .risk-badge{ background: linear-gradient(90deg, rgba(255,69,96,0.12), rgba(255,69,96,0.06)); padding:8px 12px; border-radius:8px; display:inline-block; }
-    .safe-badge{ background: linear-gradient(90deg, rgba(34,197,94,0.12), rgba(34,197,94,0.06)); padding:8px 12px; border-radius:8px; display:inline-block; }
-    /* Hide Streamlit header */
-    #MainMenu, footer, header {visibility: hidden;}
+    
+    /* NAVBAR REMOVAL */
+    #MainMenu {visibility: hidden;}
+    footer {visibility: hidden;}
+    header {visibility: hidden;}
 </style>
 """, unsafe_allow_html=True)
 
 # -----------------------------------------------------------------------------
-# LOGIC: Gemini call wrapper
+# 4. SESSION STATE
 # -----------------------------------------------------------------------------
+if 'page' not in st.session_state: st.session_state.page = 'landing'
+if 'logged_in' not in st.session_state: st.session_state.logged_in = False
+if 'user_email' not in st.session_state: st.session_state.user_email = ""
+if 'current_result' not in st.session_state: st.session_state.current_result = None
 
-def get_gemini_insight(input_type, content):
+# -----------------------------------------------------------------------------
+# 5. AI LOGIC (STRICT FORMATTING)
+# -----------------------------------------------------------------------------
+def get_gemini_analysis(input_type, content):
+    # We force a specific separator "|||" to split the UI into 3 cards later
     prompt = """
-You are Shinui, a professional medical AI assistant.
-Analyze the input carefully and format your response using these EXACT headers:
-
-### 1. Clinical Observation
-(Describe what you see or read clearly)
-
-### 2. Potential Risks
-(List potential conditions or risks associated)
-
-### 3. Recommended Actions
-(Suggest next steps, e.g., see a specialist, apply ice, etc.)
-"""
+    You are SHINUI, a high-level medical AI. Analyze the input.
+    
+    Strictly format your response into 3 parts separated by the string "|||".
+    
+    Part 1: Clinical Observation (What is visually or textually present?)
+    Part 2: Potential Risks (Pathologies or warnings)
+    Part 3: Recommended Actions (Next steps for the patient)
+    
+    Do NOT use Markdown headers (#). Just raw text for each section.
+    Example output:
+    Patient shows signs of dermatitis... ||| Risk of infection if untreated... ||| Clean area and consult dermatologist.
+    """
+    
     try:
         if input_type == "Image":
-            # If content is a PIL Image, convert to bytes for upload if needed by API
-            response = model.generate_content([prompt, "Image data: (embedded)"])
-            return response.text
+            response = model.generate_content([prompt, content])
         else:
             response = model.generate_content(f"{prompt}\n\nInput Data: {content}")
-            return response.text
+        return response.text
     except Exception as e:
-        return f"Error: {str(e)}"
+        return f"Error Processing Request.|||System could not interpret input.|||Try again with clearer data. Error: {str(e)}"
 
 # -----------------------------------------------------------------------------
-# UI PAGES
+# 6. PAGES
 # -----------------------------------------------------------------------------
 
-def nav_to(page):
-    st.session_state.page = page
-    st.experimental_rerun()
-
-
-def sign_out():
-    st.session_state.logged_in = False
-    st.session_state.user_email = ""
-    st.session_state.current_result = None
-    nav_to('landing')
-
-# --- PUBLIC LANDING PAGE ---
-
+# --- LANDING PAGE ---
 def show_landing():
-    st.markdown("""
-    <div style='display:flex; align-items:center; gap:34px;'>
-        <div style='flex:1'>
-            <h1 style='font-size:42px; margin:0; color: white'>Shinui — Medical AI Assistant</h1>
-            <p style='color: #9fb3d8; font-size:18px; margin-top:8px;'>Fast, explainable medical analysis for images and notes — MVP-ready and hackathon polished.</p>
-            <div style='margin-top:18px; display:flex; gap:12px;'>
-                <button class='hero-cta' onclick="window.parent.postMessage({type: 'nav', page: 'about'}, '*')">Learn More</button>
-            </div>
-        </div>
-        <div style='width:420px;'>
-    """, unsafe_allow_html=True)
-    if lottie_hero:
-        st_lottie(lottie_hero, height=300)
-    else:
-        st.image(Image.new('RGB', (420,300), color=(10,20,40)))
-
-    st.markdown("""
-        </div>
-    </div>
-    <hr style='border-color: rgba(255,255,255,0.03); margin-top:18px;'/>
-    """, unsafe_allow_html=True)
-
-    # Capabilities
-    c1, c2, c3 = st.columns(3)
+    # Navbar
+    c1, c2 = st.columns([1, 6])
     with c1:
+        st.markdown("<h3 style='margin:0; color:#3b82f6;'>SHINUI</h3>", unsafe_allow_html=True)
+    
+    st.markdown("<br><br>", unsafe_allow_html=True)
+    
+    col1, col2 = st.columns([1.2, 1])
+    
+    with col1:
         st.markdown("""
-        <div class='shinui-card'>
-            <h3>Visual Analysis</h3>
-            <p class='muted'>Upload scans or photos; get structured observations.</p>
-        </div>
-        """, unsafe_allow_html=True)
-    with c2:
-        st.markdown("""
-        <div class='shinui-card'>
-            <h3>Symptoms & Notes</h3>
-            <p class='muted'>Paste clinical notes or symptoms for instant triage.</p>
-        </div>
+        <h1 style='font-size: 3.5rem; line-height: 1.1; margin-bottom: 20px;'>
+            Medical AI,<br>
+            <span style='color:#3b82f6;'>Reimagined.</span>
+        </h1>
+        <p style='font-size: 1.1rem; color: #a1a1aa; margin-bottom: 40px; line-height: 1.6;'>
+            SHINUI uses state-of-the-art multimodal AI to translate complex medical data 
+            into clear, actionable insights. From X-rays to clinical notes, 
+            get answers in seconds.
+        </p>
         """, unsafe_allow_html=True)
-    with c3:
+        
+        b1, b2, b3 = st.columns([1, 1, 1])
+        with b1:
+            if st.button("Login"): nav_to('login')
+        with b2:
+            if st.button("Register"): nav_to('register')
+        with b3:
+            if st.button("About"): nav_to('about')
+
+        # Stats
+        st.markdown("<br><br>", unsafe_allow_html=True)
         st.markdown("""
-        <div class='shinui-card'>
-            <h3>Explainability</h3>
-            <p class='muted'>Clear steps and risk categories with action items.</p>
+        <div style='display:flex; gap:30px;'>
+            <div><h3 style='margin:0;'>99%</h3><small style='color:#71717a'>Uptime</small></div>
+            <div><h3 style='margin:0;'>1.5 Flash</h3><small style='color:#71717a'>Model Engine</small></div>
+            <div><h3 style='margin:0;'>24/7</h3><small style='color:#71717a'>Availability</small></div>
         </div>
         """, unsafe_allow_html=True)
 
-    st.markdown("""
-    <div style='margin-top:18px; display:flex; gap:10px;'>
-        <div style='width:160px;'>
-            <button onclick="window.parent.postMessage({type: 'nav', page: 'auth'}, '*')">Get Started</button>
-        </div>
-        <div style='width:180px;'>
-            <button onclick="window.parent.postMessage({type: 'nav', page: 'about'}, '*')">About Shinui</button>
-        </div>
-    </div>
-    """, unsafe_allow_html=True)
-
-# --- ABOUT (PUBLIC) ---
-
-def show_public_about():
-    st.markdown("""
-    <div class='shinui-card'>
-        <h2 style='color:#00f5ff'>About Shinui</h2>
-        <p style='color:#b9d6ff'>Shinui is an MVP medical assistant that transforms images and notes into easy-to-action medical observations.</p>
-        <hr style='border-color: rgba(255,255,255,0.03)'>
-        <h3>How it works</h3>
-        <ol>
-            <li>Upload an image or paste notes.</li>
-            <li>Shinui analyzes using a Gemini model and returns structured output.</li>
-            <li>Results include observations, potential risks, and recommended actions.</li>
-        </ol>
-        <h3>Safety</h3>
-        <p>Shinui is an assistant — not a diagnosis tool. Always consult a licensed professional for critical decisions.</p>
-        <div style='margin-top:12px;'>
-            <button onclick="window.parent.postMessage({type: 'nav', page: 'auth'}, '*')">Proceed to Sign In</button>
-        </div>
-    </div>
-    """, unsafe_allow_html=True)
-
-# --- AUTH PAGE ---
-
-def show_auth():
-    st.markdown("<br>", unsafe_allow_html=True)
-    c1, c2, c3 = st.columns([1, 2, 1])
-    with c2:
+    with col2:
+        # Medical Animation (DNA or Brain)
+        lottie_url = "https://lottie.host/9f6a8843-5582-417f-b273-0938323a7492/123456.json" 
+        # Fallback graphic if lottie fails (or offline)
         st.markdown("""
-        <div class='shinui-card' style='text-align:center;'>
-            <h2 style='margin-bottom:4px;'>Welcome to Shinui</h2>
-            <p style='color:var(--muted); margin-top:0;'>Sign in or create an account to start your analysis.</p>
+        <div class='meta-card' style='height: 300px; display:flex; align-items:center; justify-content:center; text-align:center;'>
+            <div>
+                <div style='font-size: 4rem;'>🧬</div>
+                <h3 style='margin-top:10px;'>Multimodal Engine</h3>
+                <p style='color:#52525b;'>Vision • Text • Reasoning</p>
+            </div>
         </div>
         """, unsafe_allow_html=True)
 
-        tab1, tab2 = st.tabs(["Sign In", "Create Account"])
-        with tab1:
-            email = st.text_input("Email Address", key="l_email")
-            password = st.text_input("Password", type="password", key="l_pass")
-            if st.button("Login"):
-                if login_user(email, password):
-                    st.session_state.logged_in = True
-                    st.session_state.user_email = email
-                    st.success("Login Successful")
-                    time.sleep(0.4)
-                    nav_to('dashboard')
-                else:
-                    st.error("Invalid email or password")
-        with tab2:
-            new_email = st.text_input("Email Address", key="r_email")
-            new_pass = st.text_input("Password", type="password", key="r_pass")
-            confirm_pass = st.text_input("Confirm Password", type="password", key="r_pass2")
-            if st.button("Register"):
-                if new_pass != confirm_pass:
-                    st.error("Passwords do not match")
-                elif register_user(new_email, new_pass):
-                    st.success("Account created! Please log in.")
-                else:
-                    st.error("Email already exists")
-
-# --- DASHBOARD (AUTHENTICATED) ---
-
+# --- AUTH PAGES ---
+def show_auth_login():
+    c1, c2, c3 = st.columns([1, 1, 1])
+    with c2:
+        st.markdown("<br><br>", unsafe_allow_html=True)
+        st.markdown("<div class='meta-card'>", unsafe_allow_html=True)
+        st.markdown("<h2 style='text-align:center'>Welcome Back</h2>", unsafe_allow_html=True)
+        email = st.text_input("Email", key="l_email")
+        password = st.text_input("Password", type="password", key="l_pass")
+        
+        if st.button("Sign In"):
+            if login_user(email, password):
+                st.session_state.logged_in = True
+                st.session_state.user_email = email
+                nav_to('dashboard')
+            else:
+                st.error("Invalid email or password.")
+        
+        st.markdown("<hr style='border-color:#27272a'>", unsafe_allow_html=True)
+        if st.button("Create Account", type="secondary"): nav_to('register')
+        if st.button("← Home", type="secondary"): nav_to('landing')
+        st.markdown("</div>", unsafe_allow_html=True)
+
+def show_auth_register():
+    c1, c2, c3 = st.columns([1, 1, 1])
+    with c2:
+        st.markdown("<br><br>", unsafe_allow_html=True)
+        st.markdown("<div class='meta-card'>", unsafe_allow_html=True)
+        st.markdown("<h2 style='text-align:center'>Create Account</h2>", unsafe_allow_html=True)
+        email = st.text_input("Email", key="r_email")
+        p1 = st.text_input("Password", type="password", key="r_p1")
+        p2 = st.text_input("Confirm Password", type="password", key="r_p2")
+        
+        if st.button("Sign Up"):
+            if p1 != p2:
+                st.error("Passwords do not match")
+            elif register_user(email, p1):
+                st.success("Account created! Please Login.")
+                time.sleep(1)
+                nav_to('login')
+            else:
+                st.error("Email already exists.")
+                
+        if st.button("← Back to Login", type="secondary"): nav_to('login')
+        st.markdown("</div>", unsafe_allow_html=True)
+
+# --- DASHBOARD ---
 def show_dashboard():
+    # SIDEBAR
     with st.sidebar:
         st.markdown(f"### 👤 {st.session_state.user_email}")
-        st.markdown("<span style='color:#22c55e; font-size:0.8rem;'>● Online</span>", unsafe_allow_html=True)
+        st.markdown("<div style='font-size:0.8rem; color:#10b981; margin-bottom:20px;'>● Secure Connection</div>", unsafe_allow_html=True)
+        
+        if st.button("New Scan"):
+            st.session_state.current_result = None
+            st.rerun()
+            
         st.markdown("---")
-        if st.button("📊 Dashboard"): nav_to('dashboard')
-        if st.button("ℹ️ About Shinui"): nav_to('about')
-        st.markdown("### Recent History")
+        st.markdown("### History")
         history = get_history(st.session_state.user_email)
         if history:
             for h_type, h_sum, h_date in history:
-                st.markdown(f"<div style='font-size:0.8rem; padding:8px; background:rgba(255,255,255,0.02); border-radius:6px; margin-bottom:6px;'>\n<b>{h_type}</b><br>{h_date}<br><span style='color:#94a3b8'>{h_sum[:60]}...</span>\n</div>", unsafe_allow_html=True)
+                st.markdown(f"""
+                <div style='border-left:2px solid #333; padding-left:10px; margin-bottom:10px;'>
+                    <div style='font-size:0.7rem; color:#71717a'>{h_date}</div>
+                    <div style='font-size:0.85rem; color:#e4e4e7'>{h_type}</div>
+                </div>
+                """, unsafe_allow_html=True)
         else:
-            st.caption("No activity found.")
+            st.caption("No previous scans.")
+            
         st.markdown("---")
         if st.button("Sign Out"): sign_out()
 
-    st.title("Medical Analysis Dashboard")
-    col1, col2 = st.columns([2, 1])
-    with col1:
-        st.markdown("### New Scan")
-        t1, t2 = st.tabs(["📷 Visual Analysis", "📝 Symptom Check"])
-        with t1:
-            st.markdown("<div class='shinui-card'>", unsafe_allow_html=True)
-            img = st.file_uploader("Upload Medical Image", type=['png','jpg','jpeg'])
-            if img and st.button("Analyze Image"):
-                image = Image.open(img)
-                with st.spinner("Shinui AI Analyzing..."):
-                    # Show lottie while processing
-                    if lottie_load:
-                        st_lottie(lottie_load, height=80)
-                    res = get_gemini_insight("Image", image)
-                    st.session_state.current_result = res
-                    add_history(st.session_state.user_email, "Visual Scan", res[:140].replace('\n', ' '))
-            st.markdown("</div>", unsafe_allow_html=True)
-        with t2:
-            st.markdown("<div class='shinui-card'>", unsafe_allow_html=True)
-            txt = st.text_area("Describe Symptoms")
-            if txt and st.button("Analyze Text"):
-                with st.spinner("Shinui AI Analyzing..."):
-                    if lottie_load:
-                        st_lottie(lottie_load, height=80)
-                    res = get_gemini_insight("Text", txt)
-                    st.session_state.current_result = res
-                    add_history(st.session_state.user_email, "Text Analysis", res[:140].replace('\n', ' '))
-            st.markdown("</div>", unsafe_allow_html=True)
-
-    with col2:
-        st.markdown("### Results")
+    # MAIN AREA
+    st.markdown("## Diagnostic Interface")
+    
+    # Layout: Left = Input, Right = Output (Desktop Style)
+    col_input, col_output = st.columns([1, 1.2])
+    
+    with col_input:
+        st.markdown("<div class='meta-card'>", unsafe_allow_html=True)
+        tab_img, tab_txt = st.tabs(["📷 Visual", "📝 Clinical Notes"])
+        
+        analyze_trigger = False
+        input_val = None
+        input_type_str = ""
+        
+        with tab_img:
+            img_file = st.file_uploader("Upload Scan / Image", type=['png','jpg','jpeg'])
+            if img_file and st.button("Analyze Visual Data"):
+                input_val = Image.open(img_file)
+                input_type_str = "Image"
+                analyze_trigger = True
+                
+        with tab_txt:
+            txt_input = st.text_area("Paste Symptoms or Notes", height=150)
+            if txt_input and st.button("Analyze Text Data"):
+                input_val = txt_input
+                input_type_str = "Text"
+                analyze_trigger = True
+        st.markdown("</div>", unsafe_allow_html=True)
+
+        if analyze_trigger:
+            with st.spinner("SHINUI Neural Engine Processing..."):
+                res = get_gemini_analysis(input_type_str, input_val)
+                st.session_state.current_result = res
+                add_history(st.session_state.user_email, f"{input_type_str} Scan", res)
+                st.rerun()
+
+    with col_output:
         if st.session_state.current_result:
-            content = st.session_state.current_result
-            # Try to split into the three headers for nicer display
-            obs = ''
-            risks = ''
-            actions = ''
-            try:
-                parts = content.split('###')
-                for p in parts:
-                    if 'Clinical Observation' in p:
-                        obs = p.replace('1. Clinical Observation','').replace('Clinical Observation','').strip()
-                    if 'Potential Risks' in p:
-                        risks = p.replace('Potential Risks','').strip()
-                    if 'Recommended Actions' in p:
-                        actions = p.replace('Recommended Actions','').strip()
-            except Exception:
-                obs = content
-            # Display neatly
-            st.markdown("<div class='shinui-card' style='border-top:4px solid #7c3aed;'>", unsafe_allow_html=True)
-            if obs:
-                st.markdown(f"<h4>Clinical Observation</h4><div style='padding:8px; background:rgba(255,255,255,0.01); border-radius:8px'>{obs}</div>", unsafe_allow_html=True)
-            if risks:
-                st.markdown(f"<h4>Potential Risks</h4><div style='padding:8px'>{risks}</div>", unsafe_allow_html=True)
-            if actions:
-                st.markdown(f"<h4>Recommended Actions</h4><div style='padding:8px'>{actions}</div>", unsafe_allow_html=True)
-            st.markdown("</div>", unsafe_allow_html=True)
+            # Parsing Logic
+            raw = st.session_state.current_result
+            parts = raw.split("|||")
+            
+            if len(parts) >= 3:
+                obs = parts[0].strip()
+                risks = parts[1].strip()
+                acts = parts[2].strip()
+                
+                st.markdown(f"""
+                <div class='box-obs'>
+                    <div class='res-header' style='color:#3b82f6'>🔍 Clinical Observation</div>
+                    {obs}
+                </div>
+                <div class='box-risk'>
+                    <div class='res-header' style='color:#f59e0b'>⚠️ Risk Assessment</div>
+                    {risks}
+                </div>
+                <div class='box-act'>
+                    <div class='res-header' style='color:#10b981'>✅ Protocol / Action</div>
+                    {acts}
+                </div>
+                """, unsafe_allow_html=True)
+            else:
+                # Fallback for unstructured response
+                st.markdown(f"<div class='meta-card'>{raw}</div>", unsafe_allow_html=True)
         else:
-            st.info("Upload an image or enter text to see analysis results here.")
-
-# --- ABOUT (AUTHENTICATED) ---
+            # Empty State
+            st.markdown("""
+            <div class='meta-card' style='text-align:center; padding:50px; color:#52525b;'>
+                <h4>Ready for Analysis</h4>
+                <p>Upload data on the left to generate insights.</p>
+            </div>
+            """, unsafe_allow_html=True)
 
+# --- ABOUT PAGE ---
 def show_about():
-    with st.sidebar:
-        if st.button("← Back to Dashboard"): nav_to('dashboard')
+    st.button("← Back", on_click=lambda: nav_to('landing'))
+    st.markdown("<br>", unsafe_allow_html=True)
     st.markdown("""
-    <div class='shinui-card'>
-        <h2 style='color:#00f5ff'>About Shinui</h2>
-        <p style='font-size:1.05rem; line-height:1.6'>
-            Shinui is an MVP-stage medical intelligence platform designed to democratize access to health data analysis.
+    <div class='meta-card' style='max-width:800px; margin:0 auto;'>
+        <h1 style='color:#3b82f6'>About SHINUI</h1>
+        <p style='font-size:1.2rem; line-height:1.6; color:#d4d4d8;'>
+            SHINUI is an advanced medical intelligence platform designed to bridge the gap 
+            between raw medical data and actionable human understanding.
         </p>
-        <hr style='border-color:#112233'>
-        <h3>The Problem</h3>
-        <p>Medical data is complex, jargon-heavy, and hard for patients to interpret.</p>
-        <h3>The Solution</h3>
-        <p>Shinui uses Gemini 2.5 to act as a translator, turning images and clinical notes into actionable, understandable insights.</p>
-        <h3>Version Info</h3>
-        <p><b>Build:</b> v1.0 MVP</p>
-        <p><b>Engine:</b> Gemini 2.5 Pro</p>
+        <hr style='border-color:#27272a'>
+        <h3>Technology</h3>
+        <p>Powered by <b>Google Gemini 1.5 Pro</b>, SHINUI utilizes a massive multimodal context window 
+        to understand images, handwriting, and clinical syntax with near-human accuracy.</p>
+        
+        <h3>Mission</h3>
+        <p>To democratize access to high-quality medical data interpretation.</p>
     </div>
     """, unsafe_allow_html=True)
 
 # -----------------------------------------------------------------------------
-# ROUTER: listens to postMessage nav events for buttons in raw HTML
+# 7. ROUTER
 # -----------------------------------------------------------------------------
-
-def handle_post_messages():
-    # Streamlit doesn't give a simple postMessage listener; we can use a workaround
-    # Buttons in HTML above call window.parent.postMessage({type:'nav',page:'...'})
-    # We'll inject JS to listen and set location hash as a signal
-    st.markdown("""
-    <script>
-    window.addEventListener('message', (e) => {
-        try{
-            if(e.data && e.data.type === 'nav'){
-                const page = e.data.page;
-                window.location.hash = page;
-            }
-        }catch(err){}
-    });
-    // On hash change, reload the page so Streamlit picks up
-    window.addEventListener('hashchange', function(){ window.location.reload(); });
-    </script>
-    """, unsafe_allow_html=True)
-
-    # Read location hash and map to streamlit pages
-    hash = st.experimental_get_query_params().get('_', [''])[0]
-    # Also check window.location.hash via JS by reading document.location.hash into query param isn't straightforward
-    # We'll instead read the hash from window.location via a short JS that sets a query param — keep it simple here
-
-# -----------------------------------------------------------------------------
-# MAIN
-# -----------------------------------------------------------------------------
-
-def main():
-    # Map basic routing
-    # Show landing or auth pages when not logged in
-    if st.session_state.page == 'landing':
-        show_landing()
-        show_public_about()
-    elif st.session_state.page == 'auth':
-        show_auth()
-    elif st.session_state.page == 'dashboard':
-        if st.session_state.logged_in:
-            show_dashboard()
-        else:
-            nav_to('auth')
-    elif st.session_state.page == 'about':
-        # If logged in, show authenticated about; else show public about
-        if st.session_state.logged_in:
-            show_about()
-        else:
-            show_public_about()
-
-    # Basic post message handler
-    handle_post_messages()
-
-if __name__ == '__main__':
-    main()
+if st.session_state.page == 'landing': show_landing()
+elif st.session_state.page == 'login': show_auth_login()
+elif st.session_state.page == 'register': show_auth_register()
+elif st.session_state.page == 'dashboard':
+    if st.session_state.logged_in: show_dashboard()
+    else: nav_to('login')
+elif st.session_state.page == 'about': show_about()