Update src/streamlit_app.py

src/streamlit_app.py

@@ -3,106 +3,57 @@ import os
 from google import genai
 from google.genai import types
 
-# --- 1. PAGE CONFIGURATION ---
-st.set_page_config(
-    page_title="Abyssinia Intelligence | Healthcare AI",
-    page_icon="🏥",
-    layout="wide"
-)
+# --- 1. CONFIG ---
+st.set_page_config(page_title="Abyssinia Intelligence 3.0", page_icon="🏥")
 
-# Custom Medical CSS
-st.markdown("""
-    <style>
-    .stChatMessage { border-radius: 10px; margin-bottom: 10px; }
-    .stAlert { border-radius: 10px; }
-    </style>
-    """, unsafe_allow_html=True)
-
-# --- 2. INITIALIZATION & SECURITY ---
 api_key = os.environ.get("GOOGLE_API_KEY")
-
-if not api_key:
-    st.error("🔑 API Key Missing: Please add 'GOOGLE_API_KEY' to your Hugging Face Secrets.")
-    st.stop()
-
-# Initialize the 2026 unified Client
 client = genai.Client(api_key=api_key)
 
-# Initialize Chat History in Session State if it doesn't exist
 if "messages" not in st.session_state:
     st.session_state.messages = []
 
-# Define the System Instruction (The "Brain" of Abyssinia)
-SYSTEM_PROMPT = """
-You are Abyssinia Intelligence, a specialized Healthcare AI assistant.
-Your goal is to assist users with medical inquiries by providing evidence-based, 
-empathetic, and structured health information. 
-
-Rules:
-1. ALWAYS start the first response with a medical disclaimer.
-2. If symptoms sound life-threatening (chest pain, stroke signs), urge immediate emergency care.
-3. Ask follow-up questions to understand the context (e.g., duration, severity).
-4. Do not provide definitive diagnoses; provide 'possible considerations'.
-"""
+# --- 2. UI ---
+st.title("🏥 Abyssinia Intelligence 3.0")
+st.caption("Now powered by Gemini 3.0 Flash with Agentic Reasoning")
 
-# --- 3. UI LAYOUT ---
-with st.sidebar:
-    st.title("🏥 Abyssinia AI")
-    st.info("Your secure, back-and-forth healthcare consultation partner.")
-    if st.button("Clear Consultation History"):
-        st.session_state.messages = []
-        st.rerun()
-    
-    st.markdown("---")
-    st.markdown("**Status:** 🟢 Connected to Gemini 2.0")
-    st.markdown("**Region:** Ethiopia/Global")
+for msg in st.session_state.messages:
+    with st.chat_message(msg["role"]):
+        st.markdown(msg["content"])
 
-# Display previous chat messages
-for message in st.session_state.messages:
-    with st.chat_message(message["role"]):
-        st.markdown(message["content"])
-
-# --- 4. CHAT LOGIC ---
-if prompt := st.chat_input("Describe your symptoms or ask a health question..."):
-    
-    # 1. Add user message to UI and State
+# --- 3. LOGIC ---
+if prompt := st.chat_input("How can I help you today?"):
     st.session_state.messages.append({"role": "user", "content": prompt})
     with st.chat_message("user"):
         st.markdown(prompt)
 
-    # 2. Generate AI Response
     with st.chat_message("assistant"):
-        message_placeholder = st.empty()
-        
         try:
-            # Map history to the format required by the 2026 SDK
+            # Map history for the 2026 SDK
             history_bundle = [
                 types.Content(role=m["role"], parts=[types.Part(text=m["content"])])
-                for m in st.session_state.messages[:-1] # Exclude the current prompt
+                for m in st.session_state.messages[:-1]
             ]
 
-            # Use the new chat interface
             chat = client.chats.create(
-                model="gemini-2.0-flash",
+                model="gemini-3-flash-preview", # <-- UPGRADED MODEL
                 config=types.GenerateContentConfig(
-                    system_instruction=SYSTEM_PROMPT,
-                    temperature=0.4
+                    system_instruction="You are Abyssinia Intelligence. Use advanced medical reasoning.",
+                    # Setting thinking to 'low' helps avoid the 429 quota errors 
+                    # by reducing the computational load for simple chat turns.
+                    thinking_config=types.ThinkingConfig(
+                        thinking_level=types.ThinkingLevel.LOW 
+                    ),
+                    temperature=0.3
                 ),
                 history=history_bundle
             )
             
-            # Request generation
             response = chat.send_message(prompt)
-            full_response = response.text
-            
-            message_placeholder.markdown(full_response)
-            
-            # 3. Add assistant response to State
-            st.session_state.messages.append({"role": "assistant", "content": full_response})
+            st.markdown(response.text)
+            st.session_state.messages.append({"role": "assistant", "content": response.text})
             
         except Exception as e:
-            st.error(f"Consultation Error: {str(e)}")
-
-# --- 5. FOOTER ---
-st.markdown("---")
-st.caption("Disclaimer: Abyssinia Intelligence is an educational AI tool. It does not provide medical diagnoses or prescriptions.")
+            if "429" in str(e):
+                st.error("🚨 **Quota Exceeded.** Gemini 3.0 Flash is in high demand. Please wait 10 seconds.")
+            else:
+                st.error(f"Error: {e}")