Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,7 +1,6 @@
 import streamlit as st
 import os
-from google import genai
-from google.genai import types
+import google.generativeai as genai  # <--- Fixed import for your environment
 from PIL import Image
 
 # -----------------------------------------------------------------------------
@@ -14,7 +13,7 @@ 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 (NEW SDK)
+# 1. SETUP & GOOGLE API
 # -----------------------------------------------------------------------------
 st.set_page_config(page_title="Shinui | Medical AI", page_icon="✨", layout="wide")
 
@@ -23,14 +22,16 @@ if not GOOGLE_API_KEY:
     st.error("⚠️ Google API Key Missing in Secrets.")
     st.stop()
 
-# Initialize the new GenAI Client
-client = genai.Client(api_key=GOOGLE_API_KEY)
+# Configure the standard library
+genai.configure(api_key=GOOGLE_API_KEY)
 
-# You can switch this string to 'gemma-3-27b-it' if you prefer Gemma
-MODEL_ID = "gemini-2.5-pro" 
+# -----------------------------------------------------------------------------
+# ⚠️ MODEL CONFIGURATION
+# -----------------------------------------------------------------------------
+MODEL_ID = "gemini-2.5-pro"
 
 # -----------------------------------------------------------------------------
-# 2. STATE
+# 2. STATE MANAGEMENT
 # -----------------------------------------------------------------------------
 if 'page' not in st.session_state: st.session_state.page = 'landing'
 if 'logged_in' not in st.session_state: st.session_state.logged_in = False
@@ -42,27 +43,29 @@ if 'user_email' not in st.session_state: st.session_state.user_email = ""
 # 3. LOGIC
 # -----------------------------------------------------------------------------
 def get_insight(input_type, content):
-    system_instruction = "You are Shinui, a medical AI assistant. Analyze the input. Provide: 1. Observation, 2. Potential Risks, 3. Recommended Actions. Keep it professional."
+    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."
+    )
     
     try:
+        # Initialize the specific model requested
+        model = genai.GenerativeModel(MODEL_ID)
+
         if input_type == "Image":
-            # The new SDK handles PIL images directly in the contents list
-            response = client.models.generate_content(
-                model=MODEL_ID,
-                contents=[system_instruction, content]
-            )
+            # 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: {content}"
-            response = client.models.generate_content(
-                model=MODEL_ID,
-                contents=full_prompt
-            )
+            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"Error: {str(e)}"
+        return f"System Error ({MODEL_ID}): {str(e)}"
 
 def nav_to(page):
     st.session_state.page = page
@@ -84,6 +87,7 @@ st.markdown("""
     .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)
@@ -102,23 +106,24 @@ def show_landing():
     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 {MODEL_ID.title()}.</span>
+            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;'>
-            Shinui analyzes medical data instantly using multimodal AI.
+            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 Shinui"): nav_to('about')
+            if st.button("About System"): nav_to('about')
     
     with c2:
         st.markdown(f"""
         <div class='shinui-card'>
-            <h3>🧬 Powered by Google GenAI</h3>
-            <p style='color:#94a3b8;'>Running on {MODEL_ID}</p>
+            <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)
 
@@ -130,14 +135,15 @@ def show_about():
     <div class='shinui-card'>
         <h2 style='color:#38bdf8'>About Shinui</h2>
         <p style='font-size:1.1rem; line-height:1.6'>
-            Shinui creates tools that bridge the gap between complex medical data and human understanding.
-            By leveraging Google's <b>{MODEL_ID}</b> architecture, we provide instant analysis for images and text.
+            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>Core Features</h3>
+        <h3>Capabilities</h3>
         <ul>
-            <li><b>Visual Analysis:</b> Interpretation of scans, labels, and skin conditions.</li>
-            <li><b>Clinical NLP:</b> Processing of complex medical notes and symptoms.</li>
+            <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)
@@ -161,46 +167,46 @@ def show_login():
 def show_dashboard():
     with st.sidebar:
         st.markdown(f"### 👤 {st.session_state.user_email}")
-        if st.button("About System"): nav_to('about_internal')
+        if st.button("System Status"): nav_to('about_internal')
         st.markdown("---")
-        st.write("HISTORY")
+        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)
         else:
-            st.caption("No scans yet.")
+            st.caption("No analysis performed yet.")
         st.markdown("---")
         if st.button("Sign Out"): sign_out()
 
-    st.title(f"{MODEL_ID.title()} Interface")
-    t1, t2 = st.tabs(["📷 Image Scan", "📝 Text Analysis"])
+    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 Image", type=['png','jpg','jpeg'])
-        if img_file and st.button("Analyze Visual"):
+        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"{MODEL_ID} Processing..."):
+            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"Image: {res[:30]}...")
+                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")
-        if txt and st.button("Analyze Notes"):
-            with st.spinner(f"{MODEL_ID} Processing..."):
+        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"Text: {res[:30]}...")
+                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</h3>
+            <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)
@@ -212,7 +218,8 @@ def show_about_internal():
     st.markdown(f"""
     <div class='shinui-card'>
         <h2 style='color:#38bdf8'>System Status</h2>
-        <p><b>Model:</b> {MODEL_ID}</p>
+        <p><b>Active Model:</b> {MODEL_ID}</p>
+        <p><b>Library:</b> google-generativeai (Standard)</p>
         <p><b>Status:</b> Online</p>
     </div>
     """, unsafe_allow_html=True)
@@ -228,4 +235,5 @@ elif st.session_state.page == '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('login')
+