Update src/streamlit_app.py

src/streamlit_app.py

@@ -3,53 +3,60 @@ import os
 from google import genai
 from google.genai import types
 
-# 1. Page Configuration
-st.set_page_config(page_title="Advanced Scientific Hypothesis Engine", layout="wide")
+# --- 1. Page Config ---
+st.set_page_config(page_title="Gemini 3 Hypothesis Engine", layout="wide")
 st.title("🔬 Advanced Scientific Hypothesis Engine")
-st.caption("Powered by Gemini 3 Pro Reasoning & Action Loops")
 
-# 2. Secure API Key Retrieval
-# Ensure you have 'GOOGLE_API_KEY' in your Hugging Face Space Secrets
+# --- 2. Secure API & Model Selection ---
 API_KEY = os.environ.get("GOOGLE_API_KEY")
 if not API_KEY:
-    st.error("API Key not found. Please add GOOGLE_API_KEY to your Space Secrets.")
+    st.error("API Key missing! Add 'GOOGLE_API_KEY' to Space Secrets.")
     st.stop()
 
-# 3. Persistent Client Initialization
-# We store the client in session_state so it isn't closed between runs
+# ADDED: Model toggle to help you manage your quota (429 errors)
+with st.sidebar:
+    st.header("Settings")
+    model_choice = st.selectbox(
+        "Select Model", 
+        ["gemini-3-flash-preview", "gemini-3-pro-preview"],
+        help="Use Flash for testing to avoid 429 Quota errors. Use Pro for final discovery."
+    )
+    st.divider()
+
+# --- 3. Persistent Session Management ---
+# Storing the client and chat in st.session_state prevents the NoneType error
 if "client" not in st.session_state:
     st.session_state.client = genai.Client(api_key=API_KEY)
 
-# 4. Persistent Chat Session Initialization
-# This prevents the 'NoneType' error by ensuring 'chat' always exists
 if "chat" not in st.session_state:
-    # Define your agent's core 'Plan -> Act -> Verify' instructions here
-    sys_instruct = "You are a senior scientific auditor. Use Python tools to verify all hypotheses."
-    
+    # Action Era Configuration
+    config = types.GenerateContentConfig(
+        system_instruction="You are a senior scientific auditor. Use Python tools to verify all hypotheses.",
+        thinking_config=types.ThinkingConfig(include_thoughts=True, thinking_level="HIGH"),
+        tools=[
+            types.Tool(google_search=types.GoogleSearchRetrieval()),
+            types.Tool(code_execution=types.ToolCodeExecution())
+        ],
+        temperature=1.0
+    )
     st.session_state.chat = st.session_state.client.chats.create(
-        model="gemini-3-flash-preview",
-        config=types.GenerateContentConfig(
-            system_instruction=sys_instruct,
-            thinking_config=types.ThinkingConfig(include_thoughts=True, thinking_level="HIGH"),
-            tools=[types.Tool(google_search=types.GoogleSearchRetrieval()),
-                   types.Tool(code_execution=types.ToolCodeExecution())]
-        )
+        model=model_choice,
+        config=config
     )
 
 if "messages" not in st.session_state:
     st.session_state.messages = []
 
-# 5. Sidebar for File Management
+# --- 4. Sidebar File Ingestion ---
 with st.sidebar:
     st.header("Research Corpus")
-    uploaded_files = st.file_uploader("Upload PDFs (Max 1M Tokens)", type="pdf", accept_multiple_files=True)
+    uploaded_files = st.file_uploader("Upload PDFs", type="pdf", accept_multiple_files=True)
     if st.button("Reset Lab State"):
-        # Clear state and rerun to fix any 'Closed Client' issues
         for key in list(st.session_state.keys()):
             del st.session_state[key]
         st.rerun()
 
-# 6. Chat Interaction Logic
+# --- 5. Chat Interface ---
 for msg in st.session_state.messages:
     with st.chat_message(msg["role"]):
         st.markdown(msg["content"])
@@ -62,21 +69,28 @@ if prompt := st.chat_input("Enter your research objective..."):
     with st.chat_message("assistant"):
         with st.status("Agent Reasoning...", expanded=True) as status:
             try:
-                # Use the persistent chat session
+                # Execution loop
                 response = st.session_state.chat.send_message(prompt)
                 
-                # Show thoughts and tool usage
+                # Show Thought Signatures (The "Thinking" process)
                 if response.candidates[0].thought_summary:
-                    st.info(f"**Thought Signature:** {response.candidates[0].thought_summary}")
+                    st.info(f"**Reasoning Path:**\n{response.candidates[0].thought_summary}")
+                
+                # Show Tool usage (Python Code)
+                for part in response.candidates[0].content.parts:
+                    if part.executable_code:
+                        st.code(part.executable_code.code, language="python", label="Generated Script")
+                    if part.code_execution_result:
+                        st.success(f"Execution Result: {part.code_execution_result.output}")
                 
-                status.update(label="Analysis Complete", state="complete")
                 st.markdown(response.text)
                 st.session_state.messages.append({"role": "assistant", "content": response.text})
+                status.update(label="Analysis Complete", state="complete")
                 
             except Exception as e:
-                st.error(f"Reasoning Error: {e}")
-                # If the client is closed, trigger a reset
-                if "closed" in str(e).lower():
-                    st.warning("Client session expired. Re-initializing...")
-                    del st.session_state.client
-                    st.rerun()
+                # Handle the 429 Quota error gracefully
+                if "429" in str(e):
+                    st.error("Quota Exceeded (429)! Switch to the 'Flash' model in the sidebar to continue.")
+                else:
+                    st.error(f"Error: {e}")
+                status.update(label="Process Failed", state="error")