api key as internal var

app.py

@@ -1,9 +1,5 @@
 import os
 
-# Initialize with fake api key
-if 'OPENAI_API_KEY' not in os.environ:
-    os.environ['OPENAI_API_KEY'] = 'sk-'
-
 import pandas as pd
 import requests
 import streamlit as st
@@ -27,19 +23,6 @@ load_dotenv()
 ss = st.session_state
 
 
-agent = ChemCrow(
-    #tools,
-    model='gpt-4',
-    temp=0.1,
-).agent_executor
-
-
-#tool_list = pd.Series(
-#    {f"✅ {t.name}":t.description for t in tools}
-#).reset_index()
-#tool_list.columns = ['Tool', 'Description']
-
-
 icon = Image.open('assets/logo0.png')
 st.set_page_config(
     page_title="ChemCrow",
@@ -60,9 +43,6 @@ st.markdown(
 
 # Session state
 # st.session_state['molecule'] = "CCO"
-def on_api_key_change():
-    api_key = ss.get('api_key') or os.getenv('OPENAI_API_KEY')
-    os.environ["OPENAI_API_KEY"] = api_key
 
 
 # sidebar
@@ -72,7 +52,7 @@ with st.sidebar:
 
     # Input OpenAI api key
     st.markdown('Input your OpenAI API key.')
-    st.text_input('OpenAI API key', type='password', key='api_key', on_change=on_api_key_change, label_visibility="collapsed")
+    st.text_input('OpenAI API key', type='password', key='api_key', label_visibility="collapsed")
 
     # Display available tools
     #st.markdown(f"# Available tools: {len(tools)}")
@@ -85,6 +65,22 @@ with st.sidebar:
     #)
 
 
+agent = ChemCrow(
+    #tools,
+    model='gpt-4',
+    temp=0.1,
+    api_key=ss.get('api_key')
+).agent_executor
+
+
+#tool_list = pd.Series(
+#    {f"✅ {t.name}":t.description for t in tools}
+#).reset_index()
+#tool_list.columns = ['Tool', 'Description']
+
+
+
+
 print(st.session_state)
 # Agent execution
 if prompt := st.chat_input():

requirements.txt

@@ -2,5 +2,5 @@ ipython
 paper-scraper @ git+https://github.com/blackadad/paper-scraper.git
 streamlit
 python-dotenv
-chemcrow==0.3.1
+chemcrow==0.3.2