Update src/streamlit_app.py

src/streamlit_app.py

@@ -2,76 +2,93 @@ import streamlit as st
 import os
 from google import genai
 from google.genai import types
+from groq import Groq
 
-st.set_page_config(page_title="Gemini 3 Hypothesis Engine", layout="wide")
+# --- Configuration & Styling ---
+st.set_page_config(page_title="Gemini 3 + Groq: Hypothesis Engine", layout="wide")
 st.title("🔬 Advanced Scientific Hypothesis Engine")
+st.caption("Dual-Engine: Gemini 3 for Research | Groq for Speed")
 
-# 1. API Key Check
-API_KEY = os.environ.get("GOOGLE_API_KEY")
-if not API_KEY:
-    st.error("API Key not found! Add GOOGLE_API_KEY to Space Secrets.")
-    st.stop()
-
-# 2. Model Selection (Sidebar)
-# Switch to Flash if you hit 429 errors on Pro!
-with st.sidebar:
-    st.header("Settings")
-    selected_model = st.selectbox(
-        "Select Model", 
-        ["gemini-3-flash-preview", "gemini-3-pro-preview"],
-        help="Flash has higher quota. Use Pro for complex math."
-    )
-    st.divider()
+# --- SECURE API KEYS ---
+GEMINI_KEY = os.environ.get("gemini_api")
+GROQ_KEY = os.environ.get("groq_api")
 
-# 3. Persistent Client & Chat (Fixes NoneType Error)
-if "client" not in st.session_state:
-    st.session_state.client = genai.Client(api_key=API_KEY)
+if not GEMINI_KEY or not GROQ_KEY:
+    st.error("Please add 'gemini_api' and 'groq_api' to your Secrets.")
+    st.stop()
 
-if "chat" not in st.session_state or st.session_state.get("current_model") != selected_model:
-    st.session_state.current_model = selected_model
-    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())]
-    )
-    st.session_state.chat = st.session_state.client.chats.create(model=selected_model, config=config)
+# --- Initialize Clients ---
+if "gemini_client" not in st.session_state:
+    st.session_state.gemini_client = genai.Client(api_key=GEMINI_KEY)
+if "groq_client" not in st.session_state:
+    st.session_state.groq_client = Groq(api_key=GROQ_KEY)
 
-if "messages" not in st.session_state:
-    st.session_state.messages = []
+SYSTEM_INSTRUCTIONS = "You are a Senior Scientific Discovery Agent. Be precise and ground claims in evidence."
 
-# 4. File Ingestion
+# --- Sidebar: Engine Selection & Research Corpus ---
 with st.sidebar:
-    st.header("Research Corpus")
+    st.header("⚙️ Engine Settings")
+    engine = st.radio("Select Primary Brain:", ["Groq (Fast/No Limits)", "Gemini 3 (Deep Search/Code)"])
+    
+    st.header("📚 Research Corpus")
     uploaded_files = st.file_uploader("Upload PDFs", type="pdf", accept_multiple_files=True)
+    
     if st.button("Reset Lab State"):
-        for key in list(st.session_state.keys()): del st.session_state[key]
+        for key in list(st.session_state.keys()):
+            del st.session_state[key]
         st.rerun()
 
-# 5. Chat Logic
+# --- Initialize Messages ---
+if "messages" not in st.session_state:
+    st.session_state.messages = []
+
+# --- Main Interaction Loop ---
 for msg in st.session_state.messages:
-    with st.chat_message(msg["role"]): st.markdown(msg["content"])
+    with st.chat_message(msg["role"]):
+        st.markdown(msg["content"])
 
 if prompt := st.chat_input("Enter your research objective..."):
     st.session_state.messages.append({"role": "user", "content": prompt})
-    with st.chat_message("user"): st.markdown(prompt)
+    with st.chat_message("user"):
+        st.markdown(prompt)
 
     with st.chat_message("assistant"):
-        with st.status("Agent Reasoning...", expanded=True) as status:
-            try:
-                # Execution
-                response = st.session_state.chat.send_message(prompt)
-                
-                if response.candidates[0].thought_summary:
-                    st.info(f"**Thought Path:** {response.candidates[0].thought_summary}")
-                
-                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:
-                if "429" in str(e):
-                    st.error("Quota Exceeded (429)! Switch to the 'Flash' model in the sidebar.")
-                else:
-                    st.error(f"Error: {e}")
-                status.update(label="Failed", state="error")
+        if engine == "Groq (Fast/No Limits)":
+            with st.spinner("Groq is thinking..."):
+                chat_completion = st.session_state.groq_client.chat.completions.create(
+                    messages=[{"role": "user", "content": prompt}],
+                    model="llama-3.3-70b-versatile",
+                )
+                response_text = chat_completion.choices[0].message.content
+                st.markdown(response_text)
+        
+        else:
+            with st.status("Gemini 3 Researching...", expanded=True) as status:
+                try:
+                    config = types.GenerateContentConfig(
+                        system_instruction=SYSTEM_INSTRUCTIONS,
+                        thinking_config=types.ThinkingConfig(include_thoughts=True, thinking_level=types.ThinkingLevel.LOW),
+                        tools=[types.Tool(google_search=types.GoogleSearchRetrieval()), types.Tool(code_execution=types.ToolCodeExecution())]
+                    )
+                    # Send message using Gemini
+                    response = st.session_state.gemini_client.models.generate_content(
+                        model="gemini-3-flash-preview",
+                        contents=prompt,
+                        config=config
+                    )
+                    
+                    # Display Reasoning/Code
+                    for part in response.candidates[0].content.parts:
+                        if part.thought:
+                            st.info(f"**Reasoning:** {part.text}")
+                        if part.executable_code:
+                            st.code(part.executable_code.code, language="python")
+                    
+                    response_text = response.text
+                    st.markdown(response_text)
+                    status.update(label="Research Complete", state="complete")
+                except Exception as e:
+                    st.error(f"Gemini Error: {e}")
+                    response_text = "Error occurred."
+
+        st.session_state.messages.append({"role": "assistant", "content": response_text})