Update src/streamlit_app.py

src/streamlit_app.py

@@ -38,7 +38,7 @@ model = genai.GenerativeModel('gemini-1.5-pro')
 # 1. DATABASE SYSTEM
 # -----------------------------------------------------------------------------
 def init_db():
-    conn = sqlite3.connect('shinui_v3.db')
+    conn = sqlite3.connect('shinui_v5.db')
     c = conn.cursor()
     c.execute('''CREATE TABLE IF NOT EXISTS users (email TEXT PRIMARY KEY, password TEXT, joined_date TEXT)''')
     c.execute('''CREATE TABLE IF NOT EXISTS history (id INTEGER PRIMARY KEY AUTOINCREMENT, email TEXT, type TEXT, summary TEXT, risk_level TEXT, date TEXT)''')
@@ -48,7 +48,7 @@ def init_db():
 def hash_pass(password): return hashlib.sha256(str.encode(password)).hexdigest()
 
 def register_user(email, password):
-    conn = sqlite3.connect('shinui_v3.db')
+    conn = sqlite3.connect('shinui_v5.db')
     try:
         conn.execute("INSERT INTO users VALUES (?, ?, ?)", (email, hash_pass(password), datetime.now().strftime("%Y-%m-%d")))
         conn.commit()
@@ -57,31 +57,34 @@ def register_user(email, password):
     finally: conn.close()
 
 def login_user(email, password):
-    conn = sqlite3.connect('shinui_v3.db')
+    conn = sqlite3.connect('shinui_v5.db')
     data = conn.execute("SELECT * FROM users WHERE email=? AND password=?", (email, hash_pass(password))).fetchall()
     conn.close()
     return data
 
 def add_history(email, type, summary, risk):
-    conn = sqlite3.connect('shinui_v3.db')
+    conn = sqlite3.connect('shinui_v5.db')
     conn.execute("INSERT INTO history (email, type, summary, risk_level, date) VALUES (?, ?, ?, ?, ?)", 
                  (email, type, summary[:100], risk, datetime.now().strftime("%Y-%m-%d %H:%M")))
     conn.commit()
     conn.close()
 
 def get_user_stats(email):
-    conn = sqlite3.connect('shinui_v3.db')
-    total = conn.execute("SELECT COUNT(*) FROM history WHERE email=?", (email,)).fetchone()[0]
-    last = conn.execute("SELECT date FROM history WHERE email=? ORDER BY id DESC LIMIT 1", (email,)).fetchone()
+    conn = sqlite3.connect('shinui_v5.db')
+    # Get last 5 items for history list
+    history = conn.execute("SELECT type, date FROM history WHERE email=? ORDER BY id DESC LIMIT 5", (email,)).fetchall()
+    # Get last login date (simplified logic: using last scan date or join date)
+    last_scan = history[0][1] if history else "New User"
     conn.close()
-    return total, (last[0] if last else "N/A")
+    return history, last_scan
 
 init_db()
 
 # -----------------------------------------------------------------------------
-# 2. SESSION STATE & UI
+# 2. SESSION STATE & CSS
 # -----------------------------------------------------------------------------
 if 'page' not in st.session_state: st.session_state.page = 'landing'
+if 'auth_mode' not in st.session_state: st.session_state.auth_mode = 'login' # Toggles Login/Register view
 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 'analysis_result' not in st.session_state: st.session_state.analysis_result = None
@@ -90,16 +93,51 @@ if 'chat_history' not in st.session_state: st.session_state.chat_history = []
 
 st.markdown("""
 <style>
-    @import url('https://fonts.googleapis.com/css2?family=Inter:wght@300;400;600;800&display=swap');
-    .stApp { background-color: #09090b; font-family: 'Inter', sans-serif; color: #e4e4e7; }
-    .meta-card { background-color: #18181b; border: 1px solid #27272a; border-radius: 12px; padding: 24px; margin-bottom: 20px; }
-    div.stButton > button { background: #3b82f6; color: white; border-radius: 8px; font-weight: 600; width: 100%; }
-    .risk-bar { height: 10px; border-radius: 5px; margin-top: 5px; background: #333; }
-    .risk-fill-low { background: #22c55e; width: 33%; height: 100%; border-radius: 5px; }
-    .risk-fill-med { background: #eab308; width: 66%; height: 100%; border-radius: 5px; }
-    .risk-fill-high { background: #ef4444; width: 100%; height: 100%; border-radius: 5px; }
-    .chat-user { background: #27272a; padding: 10px; border-radius: 10px; margin-bottom: 5px; text-align: right; color: #e4e4e7; }
-    .chat-ai { background: #1e3a8a; padding: 10px; border-radius: 10px; margin-bottom: 5px; text-align: left; color: white; }
+    @import url('https://fonts.googleapis.com/css2?family=Plus+Jakarta+Sans:wght@300;400;600;800&display=swap');
+    
+    .stApp {
+        background: radial-gradient(circle at 50% 0%, #1e293b 0%, #020617 100%);
+        font-family: 'Plus Jakarta Sans', sans-serif;
+        color: #f8fafc;
+    }
+
+    .feature-card {
+        background: rgba(255, 255, 255, 0.03);
+        border: 1px solid rgba(255, 255, 255, 0.05);
+        border-radius: 16px;
+        padding: 30px;
+        text-align: center;
+        backdrop-filter: blur(10px);
+        margin-bottom: 20px;
+    }
+    
+    .result-card {
+        background: #0f172a;
+        border-radius: 12px;
+        padding: 20px;
+        margin-bottom: 15px;
+        border-left: 4px solid #334155;
+    }
+
+    div.stButton > button {
+        background: linear-gradient(135deg, #3b82f6 0%, #2563eb 100%);
+        color: white; border: none; border-radius: 8px; font-weight: 600;
+        padding: 12px 20px; width: 100%; transition: transform 0.2s;
+    }
+    div.stButton > button:hover { transform: scale(1.02); }
+
+    .chat-user { background: #334155; padding: 12px; border-radius: 12px 12px 0 12px; margin-bottom: 10px; text-align: right; margin-left: 20%; }
+    .chat-ai { background: #1e293b; border: 1px solid #3b82f6; padding: 12px; border-radius: 12px 12px 12px 0; margin-bottom: 10px; text-align: left; margin-right: 20%; }
+
+    .history-item {
+        font-size: 0.85rem;
+        padding: 8px;
+        border-bottom: 1px solid rgba(255,255,255,0.05);
+        display: flex;
+        justify-content: space-between;
+    }
+    .history-date { color: #64748b; font-size: 0.75rem; }
+
     #MainMenu, footer, header {visibility: hidden;}
 </style>
 """, unsafe_allow_html=True)
@@ -111,11 +149,10 @@ def get_gemini_analysis(images, text_context, mode):
     role = "a helpful medical assistant"
     if mode == "Radiologist Expert": role = "a senior radiologist. Use precise technical terminology."
     elif mode == "Simple Explanation": role = "a compassionate doctor explaining to a patient. Use simple analogies."
-    elif mode == "Medical Student": role = "a medical professor teaching a student. Explain the 'why'."
     
     prompt = [
         f"You are SHINUI, {role}.",
-        f"Analyze these medical images and this clinical context: '{text_context}'.",
+        f"Analyze these medical images and context: '{text_context}'.",
         "Strictly format your output into 4 parts separated by '|||'.",
         "Part 1: Clinical Findings (What is seen?)",
         "Part 2: Risk Assessment (Low/Medium/High)",
@@ -131,21 +168,23 @@ def get_gemini_analysis(images, text_context, mode):
         return f"Error|||System Error|||Low|||{str(e)}"
 
 def chat_with_scan(user_query):
+    # Context-Aware Prompt
     prev_findings = st.session_state.analysis_result
     images = st.session_state.analysis_images
+    
     context_prompt = [
-        "You are SHINUI. We have already analyzed the patient's scan. Here are the previous findings:",
-        f"--- START FINDINGS ---\n{prev_findings}\n--- END FINDINGS ---",
-        "The user has a follow-up question based on these findings. Answer directly and clearly.",
+        "You are SHINUI. You have ALREADY analyzed this patient's scan.",
+        f"Here is your previous analysis summary: {prev_findings}",
+        "The user is asking a follow-up question. Do NOT re-analyze the whole image from scratch unless asked.",
+        "Answer the specific question based on the findings above.",
         f"User Question: {user_query}"
     ]
     full_content = context_prompt + images
     try:
         response = model.generate_content(full_content)
         return response.text
-    except: return "I could not process that follow-up. Please try rephrasing."
+    except: return "I couldn't process that. Please try again."
 
-# Navigation Helper (NO CALLBACKS)
 def go_to(page):
     st.session_state.page = page
     st.rerun()
@@ -155,6 +194,7 @@ def do_sign_out():
     st.session_state.analysis_result=None
     st.session_state.analysis_images=[]
     st.session_state.chat_history=[]
+    st.session_state.auth_mode = 'login'
     go_to('landing')
 
 # -----------------------------------------------------------------------------
@@ -164,71 +204,159 @@ def do_sign_out():
 # --- LANDING ---
 def show_landing():
     c1, c2 = st.columns([1, 6])
-    with c1: st.markdown("<h3 style='color:#3b82f6'>SHINUI</h3>", unsafe_allow_html=True)
+    with c1: st.markdown("<h3 style='color:#3b82f6; margin:0;'>SHINUI</h3>", unsafe_allow_html=True)
     st.markdown("<br><br>", unsafe_allow_html=True)
-    col1, col2 = st.columns([1.2, 1])
+    
+    col1, col2 = st.columns([1.3, 1])
     with col1:
-        st.markdown("<h1 style='font-size: 3.5rem'>Medical Intelligence<br><span style='color:#3b82f6'>Redefined.</span></h1>", unsafe_allow_html=True)
-        st.markdown("<p style='color:#a1a1aa; font-size:1.1rem'>Multi-image analysis, expert modes, and realtime AI chat.</p>", unsafe_allow_html=True)
-        b1, b2 = st.columns([1,1])
+        st.markdown("<h1 style='font-size: 4rem; line-height: 1.1; margin-bottom:20px;'>Medical Clarity.<br><span style='background: -webkit-linear-gradient(0deg, #3b82f6, #a855f7); -webkit-background-clip: text; -webkit-text-fill-color: transparent;'>Simplified.</span></h1>", unsafe_allow_html=True)
+        st.markdown("<p style='color:#94a3b8; font-size:1.2rem; line-height:1.6;'>The advanced AI assistant that translates complex medical scans and notes into clear, actionable insights.</p>", unsafe_allow_html=True)
         
-        # FIXED: Normal buttons, no on_click callback
+        b1, b2 = st.columns([1,1.5])
         with b1: 
-            if st.button("Launch App"): go_to('login')
+            if st.button("Start Now"): go_to('login')
         with b2: 
-            if st.button("About"): go_to('about')
-            
+            if st.button("How it Works"): go_to('about')
+
     with col2:
-        st.markdown("<div class='meta-card' style='height:300px; display:flex; justify-content:center; align-items:center;'><h1>🧬</h1></div>", unsafe_allow_html=True)
+        st.markdown("""
+        <div style='text-align:center; animation: float 6s ease-in-out infinite;'>
+            <div style='font-size: 7rem;'>🧬</div>
+        </div>
+        """, unsafe_allow_html=True)
+
+    # Footer Disclaimer
+    st.markdown("<br><br><br><br>", unsafe_allow_html=True)
+    st.markdown("""
+    <div style='text-align:center; font-size:0.8rem; color:#475569; border-top:1px solid #1e293b; padding-top:20px;'>
+        <b>Disclaimer:</b> SHINUI is an AI-powered analysis tool. It does not provide medical diagnosis. 
+        Always consult a qualified healthcare professional for medical advice.
+    </div>
+    """, unsafe_allow_html=True)
 
-# --- AUTH ---
+# --- ABOUT ---
+def show_about():
+    if st.button("← Back"): go_to('landing')
+    st.markdown("<br>", unsafe_allow_html=True)
+    
+    st.markdown("""
+    <div style='max-width: 700px; margin: 0 auto; text-align: center;'>
+        <h1 style='color:#3b82f6;'>About SHINUI</h1>
+        <p style='font-size:1.1rem; line-height:1.8; color:#cbd5e1; margin-bottom:30px;'>
+            SHINUI is an AI interface designed to interpret medical data. 
+            It acts as a bridge between raw diagnostic information (like X-rays and MRI scans) 
+            and human understanding.
+        </p>
+        
+        <div class='feature-card'>
+            <h3>How It Works</h3>
+            <div style='display:flex; justify-content:space-around; margin-top:20px; text-align:left;'>
+                <div>
+                    <b>1. Upload</b><br>
+                    <span style='color:#94a3b8; font-size:0.9rem'>Drag & Drop medical scans.</span>
+                </div>
+                <div>
+                    <b>2. Analyze</b><br>
+                    <span style='color:#94a3b8; font-size:0.9rem'>AI detects patterns & risks.</span>
+                </div>
+                <div>
+                    <b>3. Understand</b><br>
+                    <span style='color:#94a3b8; font-size:0.9rem'>Get simple explanations.</span>
+                </div>
+            </div>
+        </div>
+    </div>
+    """, unsafe_allow_html=True)
+
+# --- AUTH (Combined Login/Register with State Switch) ---
 def show_auth():
     c1, c2, c3 = st.columns([1,1,1])
     with c2:
-        st.markdown("<br><br><div class='meta-card'><h2 style='text-align:center'>Welcome</h2>", unsafe_allow_html=True)
-        tab1, tab2 = st.tabs(["Login", "Register"])
-        with tab1:
+        st.markdown("<br><br>", unsafe_allow_html=True)
+        st.markdown("<div class='feature-card' style='text-align:left; padding:40px;'>", unsafe_allow_html=True)
+        
+        if st.session_state.auth_mode == 'login':
+            st.markdown("<h2 style='text-align:center; margin-bottom:10px;'>Member Login</h2>", unsafe_allow_html=True)
             email = st.text_input("Email", key="l_e")
             pw = st.text_input("Password", type="password", key="l_p")
+            st.markdown("<br>", unsafe_allow_html=True)
             if st.button("Sign In"):
                 if login_user(email, pw):
                     st.session_state.logged_in=True
                     st.session_state.user_email=email
                     go_to('dashboard')
-                else: st.error("Invalid")
-        with tab2:
+                else: st.error("Invalid Credentials")
+            
+            st.markdown("<hr style='border-color:rgba(255,255,255,0.1)'>", unsafe_allow_html=True)
+            if st.button("Create Account", type="secondary"):
+                st.session_state.auth_mode = 'register'
+                st.rerun()
+                
+        else: # Register Mode
+            st.markdown("<h2 style='text-align:center; margin-bottom:10px;'>Create Account</h2>", unsafe_allow_html=True)
             re = st.text_input("Email", key="r_e")
             rp = st.text_input("Password", type="password", key="r_p")
-            if st.button("Register"):
-                if register_user(re, rp): st.success("Created! Login now.")
-                else: st.error("Taken.")
+            st.markdown("<br>", unsafe_allow_html=True)
+            if st.button("Sign Up"):
+                if register_user(re, rp): 
+                    st.success("Account created!")
+                    time.sleep(1)
+                    st.session_state.auth_mode = 'login' # AUTO-SWITCH TO LOGIN
+                    st.rerun()
+                else: st.error("Email already used.")
+            
+            st.markdown("<hr style='border-color:rgba(255,255,255,0.1)'>", unsafe_allow_html=True)
+            if st.button("Back to Login", type="secondary"):
+                st.session_state.auth_mode = 'login'
+                st.rerun()
+        
         st.markdown("</div>", unsafe_allow_html=True)
 
 # --- DASHBOARD ---
 def show_dashboard():
-    total_scans, last_scan = get_user_stats(st.session_state.user_email)
+    history_data, last_active = get_user_stats(st.session_state.user_email)
+    
     with st.sidebar:
         st.markdown(f"### 👤 {st.session_state.user_email}")
-        st.markdown("<div style='color:#10b981; margin-bottom:10px'>● Online</div>", unsafe_allow_html=True)
-        st.markdown("---")
-        c1, c2 = st.columns(2)
-        c1.metric("Total Scans", total_scans)
-        c2.metric("Last Active", last_scan if last_scan != "N/A" else "New")
-        st.markdown("---")
+        st.markdown("<div style='color:#10b981; margin-bottom:20px'>● Online</div>", unsafe_allow_html=True)
+        
         if st.button("Start New Analysis"):
             st.session_state.analysis_result = None
             st.session_state.analysis_images = []
             st.session_state.chat_history = []
             st.rerun()
+            
+        st.markdown("---")
+        # FIX: Small "Last active" text on top of history
+        st.markdown(f"<div style='font-size:0.75rem; color:#64748b; margin-bottom:10px;'>Last active: {last_active}</div>", unsafe_allow_html=True)
+        st.markdown("##### Recent Scans")
+        
+        if history_data:
+            for h_type, h_date in history_data:
+                # FIX: Clean list - Title and Date only
+                st.markdown(f"""
+                <div class='history-item'>
+                    <span>{h_type}</span>
+                    <span class='history-date'>{h_date.split(' ')[0]}</span>
+                </div>
+                """, unsafe_allow_html=True)
+        else:
+            st.caption("No scans yet.")
+            
+        st.markdown("---")
         if st.button("Sign Out"): do_sign_out()
 
-    st.title("Advanced Diagnostic Interface")
+    st.title("Diagnostic Interface")
+    
+    # INPUT SECTION
     if st.session_state.analysis_result is None:
-        st.markdown("<div class='meta-card'>", unsafe_allow_html=True)
+        st.markdown("<div class='feature-card' style='text-align:left;'>", unsafe_allow_html=True)
         img_files = st.file_uploader("Upload Medical Scans (X-Ray, MRI, CT)", type=['png','jpg','jpeg'], accept_multiple_files=True)
         c1, c2 = st.columns([2, 1])
         with c1: txt_context = st.text_area("Patient Symptoms / Clinical Context", height=100, placeholder="E.g. Patient has chest pain for 3 weeks...")
-        with c2: mode = st.selectbox("Analysis Mode", ["Radiologist Expert", "Simple Explanation", "Medical Student", "Research Deep Dive"])
+        with c2: mode = st.selectbox("Analysis Mode", ["Radiologist Expert", "Simple Explanation"])
+        
+        st.markdown("<br>", unsafe_allow_html=True)
         if st.button("Run SHINUI Analysis"):
             if not img_files and not txt_context: st.error("Please upload an image or provide text.")
             else:
@@ -242,12 +370,13 @@ def show_dashboard():
                     add_history(st.session_state.user_email, "Multi-Scan", res, risk_lvl)
                     st.rerun()
         st.markdown("</div>", unsafe_allow_html=True)
+    
+    # RESULT SECTION
     else:
         res = st.session_state.analysis_result
         images = st.session_state.analysis_images
-        col_act1, col_act2 = st.columns([1, 4])
         
-        # FIXED: Normal button logic
+        col_act1, col_act2 = st.columns([1, 4])
         with col_act1:
             if st.button("💬 Chat with Scan"): go_to('chat')
         
@@ -260,18 +389,24 @@ def show_dashboard():
             color = "#22c55e"
             if "medium" in sev_clean: bar_class = "risk-fill-med"; color = "#eab308"
             if "high" in sev_clean: bar_class = "risk-fill-high"; color = "#ef4444"
-            st.markdown(f"<div class='meta-card'><h4 style='margin:0; color:{color}'>SEVERITY SCORE: {severity.upper()}</h4><div class='risk-bar'><div class='{bar_class}'></div></div></div>", unsafe_allow_html=True)
+            
+            st.markdown(f"""
+            <div class='feature-card' style='text-align:left; border-color:{color};'>
+                <h4 style='margin:0; color:{color}'>SEVERITY SCORE: {severity.upper()}</h4>
+                <div class='risk-bar' style='background:rgba(255,255,255,0.1)'><div class='{bar_class}'></div></div>
+            </div>
+            """, unsafe_allow_html=True)
+            
             c1, c2 = st.columns(2)
             with c1:
-                st.markdown(f"<div class='meta-card'><h4 style='color:#3b82f6'>🔍 Findings</h4>{obs}</div>", unsafe_allow_html=True)
-                st.markdown(f"<div class='meta-card'><h4 style='color:#f59e0b'>⚠️ Risks</h4>{risks}</div>", unsafe_allow_html=True)
+                st.markdown(f"<div class='result-card' style='border-color:#3b82f6'><h4 style='color:#3b82f6'>🔍 Findings</h4>{obs}</div>", unsafe_allow_html=True)
+                st.markdown(f"<div class='result-card' style='border-color:#f59e0b'><h4 style='color:#f59e0b'>⚠️ Risks</h4>{risks}</div>", unsafe_allow_html=True)
             with c2:
-                st.markdown(f"<div class='meta-card'><h4 style='color:#10b981'>✅ Protocol</h4>{actions}</div>", unsafe_allow_html=True)
+                st.markdown(f"<div class='result-card' style='border-color:#10b981'><h4 style='color:#10b981'>✅ Protocol</h4>{actions}</div>", unsafe_allow_html=True)
                 if images: st.image(images[0], caption="Primary Scan Reference", use_container_width=True)
 
 # --- CHAT PAGE ---
 def show_chat():
-    # FIXED: Normal button logic
     if st.button("← Back to Results"): go_to('dashboard')
     st.title("💬 Chat with Scan")
     if st.session_state.analysis_images:
@@ -279,28 +414,26 @@ def show_chat():
             cols = st.columns(len(st.session_state.analysis_images))
             for idx, img in enumerate(st.session_state.analysis_images):
                 with cols[idx]: st.image(img, width=150)
+    
     chat_container = st.container()
     user_input = st.chat_input("Ask a follow-up question about the findings...")
+    
     if user_input:
         st.session_state.chat_history.append({"role": "user", "content": user_input})
         with st.spinner("Consulting Analysis..."):
             ai_reply = chat_with_scan(user_input)
             st.session_state.chat_history.append({"role": "ai", "content": ai_reply})
+            
     with chat_container:
         for msg in st.session_state.chat_history:
             if msg['role'] == 'user': st.markdown(f"<div class='chat-user'>{msg['content']}</div>", unsafe_allow_html=True)
             else: st.markdown(f"<div class='chat-ai'>{msg['content']}</div>", unsafe_allow_html=True)
 
-# --- ABOUT ---
-def show_about():
-    if st.button("← Back"): go_to('landing')
-    st.markdown("<br><div class='meta-card'><h2>About SHINUI</h2><p>Advanced Medical AI.</p></div>", unsafe_allow_html=True)
-
 # -----------------------------------------------------------------------------
 # 5. ROUTING
 # -----------------------------------------------------------------------------
 if st.session_state.page == 'landing': show_landing()
-elif st.session_state.page == 'login': show_auth()
+elif st.session_state.page == 'login': show_auth() # Single Auth Page with logic
 elif st.session_state.page == 'dashboard': 
     if st.session_state.logged_in: show_dashboard()
     else: go_to('login')