Update src/streamlit_app.py

src/streamlit_app.py

@@ -13,8 +13,8 @@ from streamlit_lottie import st_lottie
 # 0. CONFIGURATION
 # -----------------------------------------------------------------------------
 st.set_page_config(
-    page_title="SHINUI | Advanced Medical AI", 
-    page_icon="🧬", 
+    page_title="Medivio | Medical Clarity", 
+    page_icon="⚕️", 
     layout="wide", 
     initial_sidebar_state="expanded"
 )
@@ -35,10 +35,10 @@ genai.configure(api_key=GOOGLE_API_KEY)
 model = genai.GenerativeModel('gemini-2.5-pro')
 
 # -----------------------------------------------------------------------------
-# 1. DATABASE SYSTEM
+# 1. DATABASE SYSTEM (Renamed to Medivio)
 # -----------------------------------------------------------------------------
 def init_db():
-    conn = sqlite3.connect('shinui_v5.db')
+    conn = sqlite3.connect('medivio.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_v5.db')
+    conn = sqlite3.connect('medivio.db')
     try:
         conn.execute("INSERT INTO users VALUES (?, ?, ?)", (email, hash_pass(password), datetime.now().strftime("%Y-%m-%d")))
         conn.commit()
@@ -57,23 +57,21 @@ def register_user(email, password):
     finally: conn.close()
 
 def login_user(email, password):
-    conn = sqlite3.connect('shinui_v5.db')
+    conn = sqlite3.connect('medivio.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_v5.db')
+    conn = sqlite3.connect('medivio.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")))
+                 (email, type, summary[:100], risk, datetime.now().strftime("%Y-%m-%d")))
     conn.commit()
     conn.close()
 
 def get_user_stats(email):
-    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)
+    conn = sqlite3.connect('medivio.db')
+    history = conn.execute("SELECT type, date FROM history WHERE email=? ORDER BY id DESC LIMIT 10", (email,)).fetchall()
     last_scan = history[0][1] if history else "New User"
     conn.close()
     return history, last_scan
@@ -84,7 +82,7 @@ init_db()
 # 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 'auth_mode' not in st.session_state: st.session_state.auth_mode = 'login'
 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
@@ -131,10 +129,11 @@ st.markdown("""
 
     .history-item {
         font-size: 0.85rem;
-        padding: 8px;
+        padding: 10px;
         border-bottom: 1px solid rgba(255,255,255,0.05);
         display: flex;
         justify-content: space-between;
+        color: #cbd5e1;
     }
     .history-date { color: #64748b; font-size: 0.75rem; }
 
@@ -150,14 +149,16 @@ def get_gemini_analysis(images, text_context, mode):
     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."
     
+    # Prompt Updated for Medivio
     prompt = [
-        f"You are SHINUI, {role}.",
+        f"You are Medivio, {role}.",
         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)",
-        "Part 3: Severity Score (Just the word: Low, Medium, or High)",
-        "Part 4: Recommended Actions",
+        "Strictly format your output into 5 parts separated by '|||'.",
+        "Part 1: A Short, Descriptive Title (Max 4 words, e.g. 'Chest X-Ray Normal', 'Skin Rash Check')",
+        "Part 2: Clinical Findings (What is seen?)",
+        "Part 3: Risk Assessment (Low/Medium/High)",
+        "Part 4: Severity Score (Just the word: Low, Medium, or High)",
+        "Part 5: Recommended Actions",
         "Do not use Markdown headers. Just raw text for each section."
     ]
     content = prompt + images
@@ -165,15 +166,14 @@ def get_gemini_analysis(images, text_context, mode):
         response = model.generate_content(content)
         return response.text
     except Exception as e:
-        return f"Error|||System Error|||Low|||{str(e)}"
+        return f"Error|||System Error|||Low|||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. You have ALREADY analyzed this patient's scan.",
+        "You are Medivio. 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.",
@@ -201,16 +201,17 @@ def do_sign_out():
 # 4. PAGES
 # -----------------------------------------------------------------------------
 
-# --- LANDING ---
+# --- LANDING (UPDATED TEXT) ---
 def show_landing():
     c1, c2 = st.columns([1, 6])
-    with c1: st.markdown("<h3 style='color:#3b82f6; margin:0;'>SHINUI</h3>", unsafe_allow_html=True)
+    with c1: st.markdown("<h3 style='color:#3b82f6; margin:0;'>MEDIVIO</h3>", unsafe_allow_html=True)
     st.markdown("<br><br>", unsafe_allow_html=True)
     
     col1, col2 = st.columns([1.3, 1])
     with col1:
-        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)
+        # Option 4 Text Implementation
+        st.markdown("<h1 style='font-size: 4rem; line-height: 1.1; margin-bottom:20px;'>Upload your medical report.<br><span style='background: -webkit-linear-gradient(0deg, #3b82f6, #a855f7); -webkit-background-clip: text; -webkit-text-fill-color: transparent;'>Get answers in seconds.</span></h1>", unsafe_allow_html=True)
+        st.markdown("<p style='color:#94a3b8; font-size:1.2rem; line-height:1.6;'>AI that translates scans, labs, and clinical notes into language you understand.</p>", unsafe_allow_html=True)
         
         b1, b2 = st.columns([1,1.5])
         with b1: 
@@ -229,22 +230,22 @@ def show_landing():
     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. 
+        <b>Disclaimer:</b> Medivio 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)
 
-# --- ABOUT ---
+# --- ABOUT (REMOVED TECH JARGON) ---
 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>
+        <h1 style='color:#3b82f6;'>About Medivio</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) 
+            Medivio 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>
         
@@ -268,7 +269,7 @@ def show_about():
     </div>
     """, unsafe_allow_html=True)
 
-# --- AUTH (Combined Login/Register with State Switch) ---
+# --- AUTH ---
 def show_auth():
     c1, c2, c3 = st.columns([1,1,1])
     with c2:
@@ -299,9 +300,9 @@ def show_auth():
             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.success("Account created! Please sign in.")
+                    time.sleep(1.5)
+                    st.session_state.auth_mode = 'login'
                     st.rerun()
                 else: st.error("Email already used.")
             
@@ -327,17 +328,15 @@ def show_dashboard():
             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")
+        st.markdown("##### History")
         
         if history_data:
-            for h_type, h_date in history_data:
-                # FIX: Clean list - Title and Date only
+            for h_title, h_date in history_data:
                 st.markdown(f"""
                 <div class='history-item'>
-                    <span>{h_type}</span>
-                    <span class='history-date'>{h_date.split(' ')[0]}</span>
+                    <span>{h_title}</span>
+                    <span class='history-date'>{h_date}</span>
                 </div>
                 """, unsafe_allow_html=True)
         else:
@@ -348,7 +347,6 @@ def show_dashboard():
 
     st.title("Diagnostic Interface")
     
-    # INPUT SECTION
     if st.session_state.analysis_result is None:
         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)
@@ -357,7 +355,7 @@ def show_dashboard():
         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 st.button("Run Analysis"):
             if not img_files and not txt_context: st.error("Please upload an image or provide text.")
             else:
                 with st.spinner("Processing Multi-Modal Data..."):
@@ -365,13 +363,20 @@ def show_dashboard():
                     res = get_gemini_analysis(pil_images, txt_context, mode)
                     st.session_state.analysis_result = res
                     st.session_state.analysis_images = pil_images
+                    
+                    # PARSE (Title is Part 1)
                     parts = res.split("|||")
-                    risk_lvl = parts[2].strip() if len(parts) > 2 else "Unknown"
-                    add_history(st.session_state.user_email, "Multi-Scan", res, risk_lvl)
+                    
+                    if len(parts) >= 5:
+                        title = parts[0].strip().replace("Part 1:", "").replace("**", "").strip()
+                        risk_lvl = parts[3].strip()
+                        add_history(st.session_state.user_email, title, res, risk_lvl)
+                    else:
+                        add_history(st.session_state.user_email, "Scan Analysis", res, "Unknown")
+                        
                     st.rerun()
         st.markdown("</div>", unsafe_allow_html=True)
     
-    # RESULT SECTION
     else:
         res = st.session_state.analysis_result
         images = st.session_state.analysis_images
@@ -381,9 +386,11 @@ def show_dashboard():
             if st.button("💬 Chat with Scan"): go_to('chat')
         
         st.markdown("---")
+        
         parts = res.split("|||")
-        if len(parts) >= 4:
-            obs, risks, severity, actions = parts[0], parts[1], parts[2], parts[3]
+        if len(parts) >= 5:
+            title, obs, risks, severity, actions = parts[0], parts[1], parts[2], parts[3], parts[4]
+            
             sev_clean = severity.strip().lower()
             bar_class = "risk-fill-low"
             color = "#22c55e"
@@ -397,6 +404,8 @@ def show_dashboard():
             </div>
             """, unsafe_allow_html=True)
             
+            st.markdown(f"### {title}")
+            
             c1, c2 = st.columns(2)
             with c1:
                 st.markdown(f"<div class='result-card' style='border-color:#3b82f6'><h4 style='color:#3b82f6'>🔍 Findings</h4>{obs}</div>", unsafe_allow_html=True)
@@ -433,7 +442,7 @@ def show_chat():
 # 5. ROUTING
 # -----------------------------------------------------------------------------
 if st.session_state.page == 'landing': show_landing()
-elif st.session_state.page == 'login': show_auth() # Single Auth Page with logic
+elif st.session_state.page == 'login': show_auth()
 elif st.session_state.page == 'dashboard': 
     if st.session_state.logged_in: show_dashboard()
     else: go_to('login')