Update src/ui/streamlit_app.py

src/ui/streamlit_app.py

@@ -86,6 +86,12 @@ def run_app():
     # Sidebar configuration
     st.sidebar.header("Configuration")
     
+    openai_api_key = st.sidebar.text_input(
+        "OpenAI API Key",
+        type="password",
+        help="Paste your OpenAI API key here. It will not be stored."
+    )
+    
     model = st.sidebar.selectbox(
         "LLM Model",
         options=[
@@ -142,11 +148,15 @@ def run_app():
     
     # Run button
     if col1.button("Generate Hypotheses", type="primary", use_container_width=True):
+        # Set the OpenAI API key in the environment if provided
+        if openai_api_key:
+            os.environ["OPENAI_API_KEY"] = openai_api_key
         # Initialize AI Co-Scientist
         config = {
             "model": model if model else "gpt-3.5-turbo",  # Ensure we never pass None as the model
             "temperature": temperature,
-            "max_iterations": iterations
+            "max_iterations": iterations,
+            "openai_api_key": openai_api_key,  # Pass the key for future-proofing
         }
         
         st.session_state.acs = AICoScientist(config=config)