Update src/streamlit_app.py

src/streamlit_app.py

@@ -12,7 +12,7 @@ st.caption("Powered by Gemini 3 Pro Reasoning & Action Loops")
 API_KEY = os.environ.get("GOOGLE_API_KEY")
 
 if not API_KEY:
-    st.error("Please add your GOOGLE_API_KEY to the Hugging Face Space Secrets.")
+    st.error("Error: GOOGLE_API_KEY not found. Add it to your Space Secrets.")
     st.stop()
 
 # --- Initialize Gen AI Client ---
@@ -20,44 +20,48 @@ client = genai.Client(api_key=API_KEY)
 
 # --- Define Advanced System Instructions ---
 SYSTEM_INSTRUCTIONS = """
-You are a Senior Scientific Discovery Agent specializing in cross-disciplinary synthesis.
-Your core objective: Find contradictions, missing links, or novel hypotheses in massive research datasets.
-
-STRATEGIC PROTOCOL:
-1. ANALYSIS: Scan the 1M token context for conflicting claims between papers.
-2. PLANNING: Explicitly state your reasoning path before taking any action.
-3. VERIFICATION: Use the 'code_execution' tool to run Python simulations or statistical checks.
-4. GROUNDING: Use 'google_search' to verify if your discovery is already public.
-5. PERSISTENCE: If a tool fails, analyze the error and try a different Python approach.
-DO NOT provide medical diagnoses. Focus on chemistry, physics, and materials science.
+You are a Senior Scientific Research Agent. 
+Your goal is to find contradictions and novel links across multiple research papers.
+When a user asks a question, scan the provided context, propose a hypothesis, 
+and use your Python tool to verify the mathematical feasibility of your theory.
+Use Thought Signatures to maintain your plan across multi-step tool calls.
 """
 
-# --- Stateful Session Management ---
-if "chat" not in st.session_state:
-    # Official SDK handles Thought Signatures automatically in Chat sessions
-    st.session_state.chat = client.chats.create(
-        model="gemini-3-pro-preview",
-        config=types.GenerateContentConfig(
-            system_instruction=SYSTEM_INSTRUCTIONS,
-            thinking_config=types.ThinkingConfig(
-                include_thoughts=True, 
-                thinking_level=types.ThinkingLevel.HIGH  # Mandatory for Marathon Agents
-            ),
-            tools=[
-                types.Tool(google_search=types.GoogleSearchRetrieval()),
-                types.Tool(code_execution=types.ToolCodeExecution())
-            ],
-            temperature=1.0  # Gemini 3 reasoning is optimized for 1.0
-        )
-    )
+# --- Configuration for Action Era Reasoning ---
+gen_config = types.GenerateContentConfig(
+    system_instruction=SYSTEM_INSTRUCTIONS,
+    thinking_config=types.ThinkingConfig(
+        include_thoughts=True, 
+        thinking_level=types.ThinkingLevel.HIGH
+    ),
+    tools=[
+        types.Tool(google_search=types.GoogleSearchRetrieval()),
+        types.Tool(code_execution=types.ToolCodeExecution())
+    ],
+    temperature=1.0
+)
+
+# --- Robust Session Management (Fixes the NoneType Error) ---
+if "messages" not in st.session_state:
     st.session_state.messages = []
 
-# --- UI Sidebar: Multi-Paper Ingestion ---
+# Initialize or Re-initialize Chat Session if it is missing
+if "chat" not in st.session_state or st.session_state.chat is None:
+    try:
+        st.session_state.chat = client.chats.create(
+            model="gemini-3-pro-preview",
+            config=gen_config
+        )
+    except Exception as e:
+        st.error(f"Failed to initialize Gemini 3 Chat: {e}")
+        st.stop()
+
+# --- UI Sidebar: Multi-Paper Ingest ---
 with st.sidebar:
     st.header("Research Corpus")
     uploaded_files = st.file_uploader("Upload PDFs (Max 1M Tokens)", type="pdf", accept_multiple_files=True)
     if st.button("Reset Lab State"):
-        st.session_state.chat = None # Resetting will trigger re-initialization
+        st.session_state.chat = None # This will trigger re-initialization on rerun
         st.session_state.messages = []
         st.rerun()
 
@@ -72,22 +76,30 @@ if prompt := st.chat_input("Enter your research objective..."):
         st.markdown(prompt)
 
     with st.chat_message("assistant"):
-        # We use st.status to show the "Thought Signatures" and Action Loops live
-        with st.status("Agent Reasoning...", expanded=True) as status:
-            response = st.session_state.chat.send_message(prompt)
-            
-            # 1. Display Internal Reasoning (Thought Summary)
-            if response.candidates[0].thought_summary:
-                st.info(f"**Thought Signature Path:**\n{response.candidates[0].thought_summary}")
-            
-            # 2. Display Action Loop: Code Execution & Search
-            for part in response.candidates[0].content.parts:
-                if part.executable_code:
-                    st.code(part.executable_code.code, language="python", label="Agent-Generated Script")
-                if part.code_execution_result:
-                    st.success(f"Execution Output: {part.code_execution_result.output}")
-            
-            status.update(label="Discovery Finalized", state="complete")
+        with st.status("Agent Reasoning & Verifying...", expanded=True) as status:
+            try:
+                # Double-check initialization before calling
+                if st.session_state.chat is None:
+                     st.session_state.chat = client.chats.create(model="gemini-3-pro-preview", config=gen_config)
+                
+                response = st.session_state.chat.send_message(prompt)
+                
+                # Display Internal Reasoning (Thought Summary)
+                if response.candidates[0].thought_summary:
+                    st.info(f"**Reasoning Path:**\n{response.candidates[0].thought_summary}")
+                
+                # Display Action Loop: Code & Search
+                for part in response.candidates[0].content.parts:
+                    if part.executable_code:
+                        st.code(part.executable_code.code, language="python", label="Agent Script")
+                    if part.code_execution_result:
+                        st.success(f"Output: {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})
 
-        st.markdown(response.text)
-        st.session_state.messages.append({"role": "assistant", "content": response.text})
+            except Exception as e:
+                st.error(f"An error occurred during reasoning: {e}")
+                status.update(label="Error Occurred", state="error")