Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,10 +1,13 @@
 import streamlit as st
 import os
-import google.generativeai as genai  # <--- Fixed import for your environment
+import sqlite3
+import hashlib
+import google.generativeai as genai
 from PIL import Image
+from datetime import datetime
 
 # -----------------------------------------------------------------------------
-# 0. AUTO-CONFIG
+# 0. AUTO-CONFIG (FIX UPLOAD ERROR)
 # -----------------------------------------------------------------------------
 config_dir = ".streamlit"
 if not os.path.exists(config_dir):
@@ -13,227 +16,324 @@ with open(os.path.join(config_dir, "config.toml"), "w") as f:
     f.write("[server]\nenableXsrfProtection=false\nenableCORS=false\nmaxUploadSize=200\n")
 
 # -----------------------------------------------------------------------------
-# 1. SETUP & GOOGLE API
+# 1. DATABASE SETUP (REAL USER SYSTEM)
 # -----------------------------------------------------------------------------
-st.set_page_config(page_title="Shinui | Medical AI", page_icon="✨", layout="wide")
+def init_db():
+    conn = sqlite3.connect('users.db')
+    c = conn.cursor()
+    # Create users table
+    c.execute('''CREATE TABLE IF NOT EXISTS users
+                 (email TEXT PRIMARY KEY, password TEXT, joined_date TEXT)''')
+    # Create history table
+    c.execute('''CREATE TABLE IF NOT EXISTS history
+                 (id INTEGER PRIMARY KEY AUTOINCREMENT, email TEXT, type TEXT, summary TEXT, date TEXT)''')
+    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')
+    c = conn.cursor()
+    try:
+        c.execute("INSERT INTO users VALUES (?, ?, ?)", (email, hash_pass(password), datetime.now().strftime("%Y-%m-%d")))
+        conn.commit()
+        return True
+    except sqlite3.IntegrityError:
+        return False
+    finally:
+        conn.close()
+
+def login_user(email, password):
+    conn = sqlite3.connect('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')
+    c = conn.cursor()
+    c.execute("INSERT INTO history (email, type, summary, date) VALUES (?, ?, ?, ?)", 
+              (email, type, summary, datetime.now().strftime("%Y-%m-%d %H:%M")))
+    conn.commit()
+    conn.close()
+
+def get_history(email):
+    conn = sqlite3.connect('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
+
+# Initialize DB on startup
+init_db()
+
+# -----------------------------------------------------------------------------
+# 2. AI CONFIGURATION
+# -----------------------------------------------------------------------------
+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("⚠️ Google API Key Missing in Secrets.")
+    st.error("⚠️ System Error: Google API Key Missing.")
     st.stop()
 
-# Configure the standard library
 genai.configure(api_key=GOOGLE_API_KEY)
+model = genai.GenerativeModel('gemini-1.5-flash')
 
 # -----------------------------------------------------------------------------
-# ⚠️ MODEL CONFIGURATION
+# 3. STATE MANAGEMENT
 # -----------------------------------------------------------------------------
-MODEL_ID = "gemini-2.5-pro"
-
-# -----------------------------------------------------------------------------
-# 2. STATE MANAGEMENT
-# -----------------------------------------------------------------------------
-if 'page' not in st.session_state: st.session_state.page = 'landing'
+if 'page' not in st.session_state: st.session_state.page = 'auth'
 if 'logged_in' not in st.session_state: st.session_state.logged_in = False
-if 'history' not in st.session_state: st.session_state.history = []
-if 'result' not in st.session_state: st.session_state.result = None
 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
 
 # -----------------------------------------------------------------------------
-# 3. LOGIC
+# 4. UI STYLING (PROFESSIONAL DARK MODE)
 # -----------------------------------------------------------------------------
-def get_insight(input_type, content):
-    system_instruction = (
-        "You are Shinui, an advanced medical AI assistant. "
-        "Analyze the input rigorously. Provide: "
-        "1. Detailed Observation, 2. Potential Risks/Differential Diagnosis, 3. Recommended Clinical Actions. "
-        "Maintain a highly professional, clinical tone."
-    )
+st.markdown("""
+<style>
+    @import url('https://fonts.googleapis.com/css2?family=Inter:wght@300;400;600;800&display=swap');
     
-    try:
-        # Initialize the specific model requested
-        model = genai.GenerativeModel(MODEL_ID)
-
-        if input_type == "Image":
-            # The standard SDK handles [text, image] lists natively
-            response = model.generate_content([system_instruction, content])
-            return response.text
-            
-        elif input_type == "Text":
-            full_prompt = f"{system_instruction}\n\nPatient Notes/Input: {content}"
-            response = model.generate_content(full_prompt)
-            return response.text
-            
-    except Exception as e:
-        return f"System Error ({MODEL_ID}): {str(e)}"
+    .stApp {
+        background-color: #0f172a;
+        font-family: 'Inter', sans-serif;
+        color: #f8fafc;
+    }
+    
+    /* Cards */
+    .shinui-card {
+        background: #1e293b;
+        border: 1px solid #334155;
+        border-radius: 12px;
+        padding: 24px;
+        margin-bottom: 20px;
+        box-shadow: 0 4px 6px -1px rgba(0, 0, 0, 0.1);
+    }
+    
+    /* Buttons */
+    div.stButton > button {
+        background-color: #3b82f6;
+        color: white;
+        border-radius: 8px;
+        padding: 10px 24px;
+        font-weight: 600;
+        border: none;
+        width: 100%;
+        transition: all 0.2s;
+    }
+    div.stButton > button:hover {
+        background-color: #2563eb;
+    }
+    
+    /* Results */
+    .result-section {
+        padding: 15px;
+        border-left: 4px solid #3b82f6;
+        background: rgba(59, 130, 246, 0.1);
+        margin-bottom: 10px;
+        border-radius: 0 8px 8px 0;
+    }
+    
+    /* Inputs */
+    .stTextInput input {
+        background-color: #334155;
+        color: white;
+        border: 1px solid #475569;
+    }
+    
+    #MainMenu, footer, header {visibility: hidden;}
+</style>
+""", unsafe_allow_html=True)
 
+# -----------------------------------------------------------------------------
+# 5. LOGIC FUNCTIONS
+# -----------------------------------------------------------------------------
 def nav_to(page):
     st.session_state.page = page
     st.rerun()
 
 def sign_out():
     st.session_state.logged_in = False
-    st.session_state.history = []
-    st.session_state.result = None
     st.session_state.user_email = ""
-    nav_to('landing')
+    st.session_state.current_result = None
+    nav_to('auth')
 
-# -----------------------------------------------------------------------------
-# 4. UI STYLING
-# -----------------------------------------------------------------------------
-st.markdown("""
-<style>
-    @import url('https://fonts.googleapis.com/css2?family=Plus+Jakarta+Sans:wght@300;400;600;800&display=swap');
-    .stApp { background-color: #020617; font-family: 'Plus Jakarta Sans', sans-serif; color: #f8fafc; }
-    .shinui-card { background: rgba(30, 41, 59, 0.4); border: 1px solid rgba(148, 163, 184, 0.1); border-radius: 16px; padding: 25px; margin-bottom: 20px; }
-    div.stButton > button { background: #38bdf8; color: #0f172a; border: none; font-weight: 700; padding: 12px 20px; border-radius: 8px; width: 100%; }
-    h1, h2, h3 { letter-spacing: -0.02em; }
-    #MainMenu, footer, header {visibility: hidden;}
-</style>
-""", unsafe_allow_html=True)
+def get_gemini_insight(input_type, content):
+    # Detailed MVP System Prompt
+    prompt = """
+    You are Shinui, a professional medical AI assistant. 
+    Analyze the input carefully. 
+    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.)
+    """
+    
+    try:
+        if input_type == "Image":
+            response = model.generate_content([prompt, content])
+            return response.text
+        elif input_type == "Text":
+            response = model.generate_content(f"{prompt}\n\nInput Data: {content}")
+            return response.text
+    except Exception as e:
+        return f"Error: {str(e)}"
 
 # -----------------------------------------------------------------------------
-# 5. PAGES
+# 6. PAGES
 # -----------------------------------------------------------------------------
 
-# --- LANDING ---
-def show_landing():
-    c1, c2 = st.columns([1, 8])
-    with c1: st.markdown("### ✨ Shinui")
-    st.markdown("<br><br>", unsafe_allow_html=True)
-    
-    c1, c2 = st.columns([1.5, 1])
-    with c1:
-        st.markdown(f"""
-        <h1 style='font-size: 4rem; line-height: 1.1; margin-bottom: 20px;'>
-            Medical Intelligence.<br><span style='color:#38bdf8;'>Powered by Gemini 2.5 Pro.</span>
-        </h1>
-        <p style='font-size: 1.2rem; color: #94a3b8; margin-bottom: 40px;'>
-            High-fidelity multimodal analysis for complex clinical data.
-        </p>
-        """, unsafe_allow_html=True)
-        b1, b2 = st.columns([1, 2])
-        with b1: 
-            if st.button("Sign In"): nav_to('login')
-        with b2:
-            if st.button("About System"): nav_to('about')
-    
-    with c2:
-        st.markdown(f"""
-        <div class='shinui-card'>
-            <h3>🧬 Advanced Model Active</h3>
-            <p style='color:#94a3b8;'>Engine: <code>{MODEL_ID}</code></p>
-            <p style='color:#94a3b8; font-size: 0.9rem'>Optimized for deep reasoning & visual processing.</p>
-        </div>
-        """, unsafe_allow_html=True)
-
-# --- ABOUT ---
-def show_about():
-    if st.button("← Back Home"): nav_to('landing')
-    st.markdown("<br>", unsafe_allow_html=True)
-    st.markdown(f"""
-    <div class='shinui-card'>
-        <h2 style='color:#38bdf8'>About Shinui</h2>
-        <p style='font-size:1.1rem; line-height:1.6'>
-            Shinui is configured to use the high-performance <b>{MODEL_ID}</b> model. 
-            Unlike lightweight "Flash" models, this engine is designed for complex reasoning tasks, 
-            handling intricate medical imagery and detailed clinical notes with higher accuracy.
-        </p>
-        <hr style='border-color:#333'>
-        <h3>Capabilities</h3>
-        <ul>
-            <li><b>Deep Visual Reasoning:</b> Analyzing medical imaging with higher parameter counts.</li>
-            <li><b>Complex Context Window:</b> Processing long-form patient histories.</li>
-        </ul>
-    </div>
-    """, unsafe_allow_html=True)
-
-# --- LOGIN ---
-def show_login():
-    c1, c2, c3 = st.columns([1,1,1])
+# --- AUTH PAGE (LOGIN / REGISTER) ---
+def show_auth():
+    c1, c2, c3 = st.columns([1, 2, 1])
     with c2:
         st.markdown("<br><br>", unsafe_allow_html=True)
-        st.markdown("<div class='shinui-card' style='text-align:center;'><h2>Member Access</h2></div>", unsafe_allow_html=True)
-        email = st.text_input("Email")
-        password = st.text_input("Password", type="password")
-        if st.button("Authenticate"):
-            if email:
-                st.session_state.logged_in = True
-                st.session_state.user_email = email
-                nav_to('dashboard')
-        if st.button("Back"): nav_to('landing')
+        st.markdown("<h1 style='text-align:center; color:#3b82f6;'>✨ Shinui</h1>", unsafe_allow_html=True)
+        st.markdown("<p style='text-align:center; color:#94a3b8;'>Intelligent Medical Analysis MVP</p>", unsafe_allow_html=True)
+        
+        tab1, tab2 = st.tabs(["Sign In", "Create Account"])
+        
+        with tab1:
+            st.markdown("<div class='shinui-card'>", unsafe_allow_html=True)
+            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.5) # Smooth transition
+                    nav_to('dashboard')
+                else:
+                    st.error("Invalid email or password")
+            st.markdown("</div>", unsafe_allow_html=True)
+            
+        with tab2:
+            st.markdown("<div class='shinui-card'>", unsafe_allow_html=True)
+            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")
+            st.markdown("</div>", unsafe_allow_html=True)
 
 # --- DASHBOARD ---
 def show_dashboard():
+    # SIDEBAR NAVIGATION
     with st.sidebar:
         st.markdown(f"### 👤 {st.session_state.user_email}")
-        if st.button("System Status"): nav_to('about_internal')
+        st.markdown("<span style='color:#22c55e; font-size:0.8rem;'>● Online</span>", unsafe_allow_html=True)
         st.markdown("---")
-        st.write("SESSION HISTORY")
-        if st.session_state.history:
-            for h in reversed(st.session_state.history):
-                st.markdown(f"<div style='font-size:0.8rem; padding:5px; border-left:2px solid #38bdf8; margin-bottom:5px;'>{h[:50]}...</div>", unsafe_allow_html=True)
+        
+        if st.button("📊 Dashboard"): nav_to('dashboard')
+        if st.button("ℹ️ About Shinui"): nav_to('about')
+        
+        st.markdown("### Recent 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:#334155; border-radius:6px; margin-bottom:6px;'>
+                    <b>{h_type}</b><br>{h_date}<br><span style='color:#94a3b8'>{h_sum[:40]}...</span>
+                </div>
+                """, unsafe_allow_html=True)
         else:
-            st.caption("No analysis performed yet.")
+            st.caption("No activity found.")
+            
         st.markdown("---")
         if st.button("Sign Out"): sign_out()
 
-    st.title(f"Shinui Interface ({MODEL_ID})")
-    t1, t2 = st.tabs(["📷 Advanced Image Scan", "📝 Clinical Text Analysis"])
-
-    with t1:
-        st.markdown("<div class='shinui-card'>", unsafe_allow_html=True)
-        img_file = st.file_uploader("Upload Medical Imaging", type=['png','jpg','jpeg'])
-        if img_file and st.button("Run 2.5 Pro Analysis"):
-            image = Image.open(img_file)
-            st.image(image, width=300)
-            with st.spinner(f"Running deep analysis with {MODEL_ID}..."):
-                res = get_insight("Image", image)
-                st.session_state.result = res
-                st.session_state.history.append(f"Img: {res[:30]}...")
-        st.markdown("</div>", unsafe_allow_html=True)
-
-    with t2:
-        st.markdown("<div class='shinui-card'>", unsafe_allow_html=True)
-        txt = st.text_area("Clinical Notes / Symptoms")
-        if txt and st.button("Run 2.5 Pro Analysis"):
-            with st.spinner(f"Running deep analysis with {MODEL_ID}..."):
-                res = get_insight("Text", txt)
-                st.session_state.result = res
-                st.session_state.history.append(f"Txt: {res[:30]}...")
-        st.markdown("</div>", unsafe_allow_html=True)
-
-    if st.session_state.result:
-        st.markdown(f"""
-        <div class='shinui-card' style='border-left: 5px solid #38bdf8;'>
-            <h3 style='margin-top:0; color:#38bdf8;'>Analysis Result ({MODEL_ID})</h3>
-            <div style='white-space: pre-wrap; color: #e2e8f0; line-height: 1.6;'>{st.session_state.result}</div>
-        </div>
-        """, unsafe_allow_html=True)
-
-# --- INTERNAL ABOUT ---
-def show_about_internal():
+    # MAIN CONTENT
+    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..."):
+                    res = get_gemini_insight("Image", image)
+                    st.session_state.current_result = res
+                    # Save detailed result to DB
+                    add_history(st.session_state.user_email, "Visual Scan", res[:100].replace('\n', ' '))
+        
+        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..."):
+                    res = get_gemini_insight("Text", txt)
+                    st.session_state.current_result = res
+                    add_history(st.session_state.user_email, "Text Analysis", res[:100].replace('\n', ' '))
+
+    # RESULTS SECTION (Right Side / Bottom)
+    with col2:
+        st.markdown("### Results")
+        if st.session_state.current_result:
+            # Parsing Markdown for pretty display
+            content = st.session_state.current_result
+            st.markdown(f"""
+            <div class='shinui-card' style='border-top: 4px solid #3b82f6;'>
+                {content}
+            </div>
+            """, unsafe_allow_html=True)
+        else:
+            st.info("Upload an image or enter text to see analysis results here.")
+
+# --- ABOUT PAGE ---
+def show_about():
     with st.sidebar:
-        if st.button("← Back"): nav_to('dashboard')
-    st.markdown(f"""
+        if st.button("← Back to Dashboard"): nav_to('dashboard')
+    
+    st.markdown("""
     <div class='shinui-card'>
-        <h2 style='color:#38bdf8'>System Status</h2>
-        <p><b>Active Model:</b> {MODEL_ID}</p>
-        <p><b>Library:</b> google-generativeai (Standard)</p>
-        <p><b>Status:</b> Online</p>
+        <h2 style='color:#3b82f6'>About Shinui</h2>
+        <p style='font-size:1.1rem; line-height:1.6'>
+            Shinui is an MVP-stage medical intelligence platform designed to democratize access to health data analysis.
+        </p>
+        <hr style='border-color:#334155'>
+        <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 Google's Gemini 1.5 Flash model 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 1.5 Flash</p>
     </div>
     """, unsafe_allow_html=True)
 
 # -----------------------------------------------------------------------------
-# 6. ROUTER
+# 7. ROUTER
 # -----------------------------------------------------------------------------
-if st.session_state.page == 'landing': show_landing()
-elif st.session_state.page == 'about': show_about()
-elif st.session_state.page == 'login': show_login()
+import time # Import needed for sleep
+
+if st.session_state.page == 'auth':
+    show_auth()
 elif st.session_state.page == 'dashboard':
     if st.session_state.logged_in: show_dashboard()
-    else: nav_to('login')
-elif st.session_state.page == 'about_internal':
-    if st.session_state.logged_in: show_about_internal()
-    else: nav_to('login')
-
+    else: nav_to('auth')
+elif st.session_state.page == 'about':
+    if st.session_state.logged_in: show_about()
+    else: nav_to('auth')