Update src/streamlit_app.py

src/streamlit_app.py

@@ -38,7 +38,7 @@ model = genai.GenerativeModel('gemini-2.5-pro')
 # 1. DATABASE SYSTEM
 # -----------------------------------------------------------------------------
 def init_db():
-    conn = sqlite3.connect('medivio_final_v8.db')
+    conn = sqlite3.connect('medivio_v9.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('medivio_final_v8.db')
+    conn = sqlite3.connect('medivio_v9.db')
     try:
         conn.execute("INSERT INTO users VALUES (?, ?, ?)", (email, hash_pass(password), datetime.now().strftime("%Y-%m-%d")))
         conn.commit()
@@ -57,20 +57,20 @@ def register_user(email, password):
     finally: conn.close()
 
 def login_user(email, password):
-    conn = sqlite3.connect('medivio_final_v8.db')
+    conn = sqlite3.connect('medivio_v9.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('medivio_final_v8.db')
+    conn = sqlite3.connect('medivio_v9.db')
     conn.execute("INSERT INTO history (email, type, summary, risk_level, date) VALUES (?, ?, ?, ?, ?)", 
                  (email, type, summary[:100], risk, datetime.now().strftime("%Y-%m-%d")))
     conn.commit()
     conn.close()
 
 def get_user_stats(email):
-    conn = sqlite3.connect('medivio_final_v8.db')
+    conn = sqlite3.connect('medivio_v9.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()
@@ -99,16 +99,15 @@ st.markdown("""
         color: #f8fafc;
     }
 
-    .feature-card {
+    /* Clean Card Style */
+    .stContainer {
         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;
+        padding: 20px;
     }
     
+    /* Result Cards */
     .result-card {
         background: #0f172a;
         border-radius: 12px;
@@ -227,88 +226,102 @@ def show_landing():
         </div>
         """, unsafe_allow_html=True)
 
+    # PURE MARKDOWN FOOTER (No HTML Tags)
     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> 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)
+    st.divider()
+    st.caption("Disclaimer: Medivio is an AI-powered analysis tool. It does not provide medical diagnosis. Always consult a qualified healthcare professional for medical advice.")
 
-# --- ABOUT (CLEAN TEXT ONLY) ---
+# --- ABOUT (FIXED: PURE STREAMLIT) ---
 def show_about():
     if st.button("← Back"): go_to('landing')
     st.markdown("<br>", unsafe_allow_html=True)
     
-    st.markdown("<h1 style='text-align:center; color:#3b82f6;'>About Medivio</h1>", unsafe_allow_html=True)
+    # Title
+    st.title("About Medivio")
     
+    # Intro Text (Native Markdown)
     st.markdown("""
-    <div style='max-width: 700px; margin: 0 auto; font-size: 1.1rem; line-height: 1.8; color: #cbd5e1;'>
-        <p style='text-align: center;'>
-            Medivio is an intelligent interface for your health. It acts as a secure bridge between 
-            complex medical data—like X-Rays, MRI scans, and doctor's notes—and clear human understanding.
-        </p>
+    Medivio is an intelligent interface for your health. It acts as a secure bridge between 
+    complex medical data—like X-Rays, MRI scans, and doctor's notes—and clear human understanding.
+    """)
+    
+    st.divider()
+    
+    # Mission Section (Native Streamlit Columns)
+    col_a, col_b = st.columns(2)
+    
+    with col_a:
+        st.subheader("Our Mission")
+        st.markdown("""
+        To democratize access to medical information. We believe everyone should be able to 
+        understand their own health records instantly, without waiting days for an appointment.
+        """)
         
-        <hr style='border-color: #334155; margin: 30px 0;'>
+    with col_b:
+        st.subheader("Technology")
+        st.markdown("""
+        Medivio leverages **Google Gemini 1.5 Pro**, a multimodal AI model capable of processing 
+        vision (images) and text simultaneously. This allows for context-aware analysis that 
+        mimics the reasoning of a medical professional.
+        """)
+        
+    st.divider()
+    
+    # Features (Native Columns, No HTML Injection)
+    st.subheader("How It Works")
+    f1, f2, f3 = st.columns(3)
+    
+    with f1:
+        st.info("**1. Upload**\n\nDrag & Drop medical scans or paste clinical notes securely.")
         
-        <h3 style='color:white;'>Mission</h3>
-        <p>
-            To democratize access to medical information. We believe everyone should be able to 
-            understand their own health records instantly, without waiting days for an appointment.
-        </p>
+    with f2:
+        st.info("**2. Analyze**\n\nOur AI engine detects patterns, risks, and anomalies instantly.")
         
-        <h3 style='color:white;'>Technology</h3>
-        <p>
-            Medivio leverages <b>Google Gemini 1.5 Pro</b>, a multimodal AI model capable of processing 
-            vision (images) and text simultaneously. This allows for context-aware analysis that 
-            mimics the reasoning of a medical professional.
-        </p>
-    </div>
-    """, unsafe_allow_html=True)
+    with f3:
+        st.success("**3. Understand**\n\nReceive a clear, jargon-free explanation of your health data.")
 
 # --- AUTH ---
 def show_auth():
     c1, c2, c3 = st.columns([1,1,1])
     with c2:
         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 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()
+        # Simple Container
+        with st.container():
+            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 Credentials")
                 
-        else: 
-            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")
-            st.markdown("<br>", unsafe_allow_html=True)
-            if st.button("Sign Up"):
-                if register_user(re, rp): 
-                    st.success("Account created! Please sign in.")
-                    time.sleep(1.5)
+                st.markdown("---")
+                if st.button("Create Account", type="secondary"):
+                    st.session_state.auth_mode = 'register'
+                    st.rerun()
+                    
+            else: 
+                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")
+                st.markdown("<br>", unsafe_allow_html=True)
+                if st.button("Sign Up"):
+                    if register_user(re, rp): 
+                        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.")
+                
+                st.markdown("---")
+                if st.button("Back to Login", type="secondary"):
                     st.session_state.auth_mode = '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():
@@ -345,31 +358,32 @@ def show_dashboard():
     st.title("Diagnostic Interface")
     
     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)
-        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"])
-        
-        st.markdown("<br>", unsafe_allow_html=True)
-        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..."):
-                    pil_images = [Image.open(x) for x in img_files] if img_files else []
-                    res = get_gemini_analysis(pil_images, txt_context, mode)
-                    st.session_state.analysis_result = res
-                    st.session_state.analysis_images = pil_images
-                    
-                    parts = res.split("|||")
-                    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, "Analysis", res, "Unknown")
-                    st.rerun()
-        st.markdown("</div>", unsafe_allow_html=True)
+        # Native Container for Upload
+        with st.container():
+            st.write("Upload your medical data below.")
+            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"])
+            
+            st.markdown("<br>", unsafe_allow_html=True)
+            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..."):
+                        pil_images = [Image.open(x) for x in img_files] if img_files else []
+                        res = get_gemini_analysis(pil_images, txt_context, mode)
+                        st.session_state.analysis_result = res
+                        st.session_state.analysis_images = pil_images
+                        
+                        parts = res.split("|||")
+                        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, "Analysis", res, "Unknown")
+                        st.rerun()
     
     else:
         res = st.session_state.analysis_result
@@ -384,16 +398,17 @@ def show_dashboard():
         parts = res.split("|||")
         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"
             if "medium" in sev_clean: bar_class = "risk-fill-med"; color = "#eab308"
             if "high" in sev_clean: bar_class = "risk-fill-high"; color = "#ef4444"
             
+            # Severity Bar (HTML is required here for custom CSS visuals, but kept simple)
             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 style='background:rgba(255,255,255,0.05); padding:20px; border-radius:10px; border-left:5px solid {color};'>
+                <h4 style='margin:0; color:{color}'>SEVERITY: {severity.upper()}</h4>
             </div>
             """, unsafe_allow_html=True)