Update src/streamlit_app.py

src/streamlit_app.py

@@ -5,9 +5,10 @@ import hashlib
 import google.generativeai as genai
 from PIL import Image
 from datetime import datetime
+import time
 
 # -----------------------------------------------------------------------------
-# 0. AUTO-CONFIG (FIX UPLOAD ERROR)
+# 0. AUTO-CONFIG
 # -----------------------------------------------------------------------------
 config_dir = ".streamlit"
 if not os.path.exists(config_dir):
@@ -16,15 +17,13 @@ with open(os.path.join(config_dir, "config.toml"), "w") as f:
     f.write("[server]\nenableXsrfProtection=false\nenableCORS=false\nmaxUploadSize=200\n")
 
 # -----------------------------------------------------------------------------
-# 1. DATABASE SETUP (REAL USER SYSTEM)
+# 1. DATABASE & AUTH SYSTEM (The part you liked)
 # -----------------------------------------------------------------------------
 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()
@@ -69,90 +68,61 @@ def get_history(email):
     conn.close()
     return data
 
-# Initialize DB on startup
 init_db()
 
 # -----------------------------------------------------------------------------
-# 2. AI CONFIGURATION
+# 2. AI SETUP (Reverted to the SMART Model)
 # -----------------------------------------------------------------------------
-st.set_page_config(page_title="Shinui | Medical MVP", page_icon="✨", layout="wide")
+st.set_page_config(page_title="SHINUI | Intelligent Care", 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.")
+    st.error("⚠️ API Key Missing. Add 'GOOGLE_API_KEY' to Secrets.")
     st.stop()
 
 genai.configure(api_key=GOOGLE_API_KEY)
-model = genai.GenerativeModel('gemini-1.5-flash')
+
+# Reverting to 'gemini-1.5-pro' (High Intelligence) instead of Flash
+model = genai.GenerativeModel('gemini-2.5-pro')
 
 # -----------------------------------------------------------------------------
-# 3. STATE MANAGEMENT
+# 3. STATE & STYLING
 # -----------------------------------------------------------------------------
 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 '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
 
-# -----------------------------------------------------------------------------
-# 4. UI STYLING (PROFESSIONAL DARK MODE)
-# -----------------------------------------------------------------------------
 st.markdown("""
 <style>
     @import url('https://fonts.googleapis.com/css2?family=Inter:wght@300;400;600;800&display=swap');
+    .stApp { background-color: #0f172a; font-family: 'Inter', sans-serif; color: #f8fafc; }
     
-    .stApp {
-        background-color: #0f172a;
-        font-family: 'Inter', sans-serif;
-        color: #f8fafc;
-    }
-    
-    /* Cards */
+    /* Custom 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);
+        background: #1e293b; border: 1px solid #334155; border-radius: 12px;
+        padding: 24px; margin-bottom: 20px;
     }
     
+    /* Result Boxes */
+    .res-box { padding: 15px; border-radius: 8px; margin-bottom: 10px; }
+    .res-obs { background: rgba(59, 130, 246, 0.15); border-left: 4px solid #3b82f6; } /* Blue */
+    .res-risk { background: rgba(245, 158, 11, 0.15); border-left: 4px solid #f59e0b; } /* Orange */
+    .res-act { background: rgba(16, 185, 129, 0.15); border-left: 4px solid #10b981; } /* Green */
+    
     /* 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;
+        background: #3b82f6; color: white; border: none; border-radius: 8px;
+        padding: 10px 20px; font-weight: 600; width: 100%; transition: 0.2s;
     }
+    div.stButton > button:hover { background: #2563eb; }
     
     #MainMenu, footer, header {visibility: hidden;}
 </style>
 """, unsafe_allow_html=True)
 
 # -----------------------------------------------------------------------------
-# 5. LOGIC FUNCTIONS
+# 4. LOGIC (With Structured Formatting)
 # -----------------------------------------------------------------------------
 def nav_to(page):
     st.session_state.page = page
@@ -164,176 +134,169 @@ def sign_out():
     st.session_state.current_result = None
     nav_to('auth')
 
-def get_gemini_insight(input_type, content):
-    # Detailed MVP System Prompt
+def get_ai_analysis(input_type, content):
+    # We force the AI to use a specific separator "|||" so we can split the result nicely
     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)
+    You are SHINUI, a medical AI. Analyze the input.
+    Provide the output in exactly 3 sections separated by '|||'.
     
-    ### 2. Potential Risks
-    (List potential conditions or risks associated)
+    Section 1: Clinical Observation (What is visible/described?)
+    Section 2: Potential Risks (What are the dangers?)
+    Section 3: Next Steps (Actionable advice)
     
-    ### 3. Recommended Actions
-    (Suggest next steps, e.g., see a specialist, apply ice, etc.)
+    Do not add Markdown headers like '##'. Just the content.
     """
     
     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
+        else:
+            response = model.generate_content(f"{prompt}\n\nInput: {content}")
+            
+        return response.text
     except Exception as e:
-        return f"Error: {str(e)}"
+        return f"Error|||System Failure|||{str(e)}"
 
 # -----------------------------------------------------------------------------
-# 6. PAGES
+# 5. PAGES
 # -----------------------------------------------------------------------------
 
-# --- AUTH PAGE (LOGIN / REGISTER) ---
+# --- AUTH ---
 def show_auth():
-    c1, c2, c3 = st.columns([1, 2, 1])
+    c1, c2, c3 = st.columns([1, 1.5, 1])
     with c2:
-        st.markdown("<br><br>", unsafe_allow_html=True)
-        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)
+        st.markdown("<br><br><h1 style='text-align:center; color:#3b82f6;'>SHINUI</h1>", unsafe_allow_html=True)
+        st.markdown("<p style='text-align:center; color:#94a3b8;'>Advanced Medical Intelligence</p>", unsafe_allow_html=True)
         
-        tab1, tab2 = st.tabs(["Sign In", "Create Account"])
+        tab1, tab2 = st.tabs(["Login", "Register"])
         
         with tab1:
             st.markdown("<div class='shinui-card'>", unsafe_allow_html=True)
-            email = st.text_input("Email Address", key="l_email")
+            email = st.text_input("Email", key="l_email")
             password = st.text_input("Password", type="password", key="l_pass")
-            if st.button("Login"):
+            if st.button("Sign In"):
                 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
+                    st.success("Success")
+                    time.sleep(0.5)
                     nav_to('dashboard')
                 else:
-                    st.error("Invalid email or password")
+                    st.error("Invalid credentials")
             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.")
+            r_email = st.text_input("Email", key="r_email")
+            r_pass = st.text_input("Password", type="password", key="r_pass")
+            if st.button("Create Account"):
+                if register_user(r_email, r_pass):
+                    st.success("Account created. Please log in.")
                 else:
-                    st.error("Email already exists")
+                    st.error("Email taken.")
             st.markdown("</div>", unsafe_allow_html=True)
 
 # --- DASHBOARD ---
 def show_dashboard():
-    # SIDEBAR NAVIGATION
+    # 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("🟢 **System Online**")
         st.markdown("---")
+        if st.button("New Scan"): 
+            st.session_state.current_result = None
+            st.rerun()
         
-        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:#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 activity found.")
-            
-        st.markdown("---")
+                st.markdown(f"<div style='font-size:0.75rem; color:#94a3b8; border-bottom:1px solid #333; padding:5px;'>{h_date}<br><b style='color:white'>{h_type}</b></div>", unsafe_allow_html=True)
+        
+        st.markdown("<br><br>", unsafe_allow_html=True)
         if st.button("Sign Out"): sign_out()
 
-    # MAIN CONTENT
-    st.title("Medical Analysis Dashboard")
+    # MAIN
+    st.title("Diagnostic Interface")
     
-    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', ' '))
+    # INPUT SECTION
+    st.markdown("<div class='shinui-card'>", unsafe_allow_html=True)
+    t1, t2 = st.tabs(["📷 Visual", "📝 Text"])
+    
+    input_content = None
+    input_type = None
+    
+    with t1:
+        img_file = st.file_uploader("Upload Image", type=['png','jpg','jpeg'])
+        if img_file and st.button("Analyze Image"):
+            input_content = Image.open(img_file)
+            input_type = "Image"
+            
+    with t2:
+        txt_val = st.text_area("Enter Symptoms")
+        if txt_val and st.button("Analyze Text"):
+            input_content = txt_val
+            input_type = "Text"
+            
+    if input_content:
+        with st.spinner("SHINUI is thinking..."):
+            # Call AI
+            raw_res = get_ai_analysis(input_type, input_content)
+            st.session_state.current_result = raw_res
+            # Save simple log
+            add_history(st.session_state.user_email, f"{input_type} Analysis", "Analysis Complete")
+            st.rerun()
+    st.markdown("</div>", unsafe_allow_html=True)
 
-    # 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
+    # RESULT DISPLAY (THE FIXED PART)
+    if st.session_state.current_result:
+        # Split the result by our secret separator |||
+        parts = st.session_state.current_result.split("|||")
+        
+        # Ensure we have 3 parts, otherwise just show text
+        if len(parts) >= 3:
+            obs = parts[0].strip()
+            risks = parts[1].strip()
+            actions = parts[2].strip()
+            
+            col1, col2 = st.columns([1, 1])
+            
+            st.markdown("### Analysis Results")
+            
+            # Observation (Full Width)
             st.markdown(f"""
-            <div class='shinui-card' style='border-top: 4px solid #3b82f6;'>
-                {content}
+            <div class='res-box res-obs'>
+                <h4 style='color:#3b82f6; margin:0;'>🔍 Clinical Observation</h4>
+                <p style='color:#e2e8f0;'>{obs}</p>
             </div>
             """, unsafe_allow_html=True)
+            
+            # Split Risks and Actions
+            with col1:
+                st.markdown(f"""
+                <div class='res-box res-risk'>
+                    <h4 style='color:#f59e0b; margin:0;'>⚠️ Potential Risks</h4>
+                    <p style='color:#e2e8f0;'>{risks}</p>
+                </div>
+                """, unsafe_allow_html=True)
+            with col2:
+                st.markdown(f"""
+                <div class='res-box res-act'>
+                    <h4 style='color:#10b981; margin:0;'>✅ Recommended Actions</h4>
+                    <p style='color:#e2e8f0;'>{actions}</p>
+                </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 to Dashboard"): nav_to('dashboard')
-    
-    st.markdown("""
-    <div class='shinui-card'>
-        <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)
+            # Fallback if AI format varies
+            st.info(st.session_state.current_result)
 
 # -----------------------------------------------------------------------------
-# 7. ROUTER
+# 6. ROUTING
 # -----------------------------------------------------------------------------
-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('auth')
-elif st.session_state.page == 'about':
-    if st.session_state.logged_in: show_about()
-    else: nav_to('auth')  
+    if st.session_state.logged_in:
+        show_dashboard()
+    else:
+        nav_to('auth')