extra checks + new cc version

app.py

@@ -11,6 +11,7 @@ from PIL import Image
 from chemcrow.agents import ChemCrow, make_tools
 from chemcrow.frontend.streamlit_callback_handler import \
     StreamlitCallbackHandlerChem
+from utils import oai_key_isvalid
 
 from dotenv import load_dotenv
 
@@ -45,8 +46,6 @@ agent = ChemCrow(
     }
 ).agent_executor
 
-print(st.secrets)
-
 tools = agent.tools
 
 tool_list = pd.Series(
@@ -54,6 +53,14 @@ tool_list = pd.Series(
 ).reset_index()
 tool_list.columns = ['Tool', 'Description']
 
+def on_api_key_change():
+    api_key = ss.get('api_key') or os.getenv('OPENAI_API_KEY')
+    # Check if key is valid
+    if oai_key_isvalid(api_key):
+        os.environ["OPENAI_API_KEY"] = api_key
+    else:
+        st.write("Please input a valid OpenAI API key.")
+
 # sidebar
 with st.sidebar:
     chemcrow_logo = Image.open('assets/chemcrow-logo-bold-new.png')
@@ -61,7 +68,13 @@ 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', label_visibility="collapsed")
+    st.text_input(
+        'OpenAI API key',
+        type='password',
+        key='api_key',
+        on_change=on_api_key_change,
+        label_visibility="collapsed"
+    )
 
     # Display available tools
     st.markdown(f"# Available tools: {len(tool_list)}")

requirements.txt

@@ -2,4 +2,6 @@ ipython
 paper-scraper @ git+https://github.com/blackadad/paper-scraper.git
 streamlit
 python-dotenv
-chemcrow==0.3.6
+duckduckgo-search
+wikipedia
+chemcrow==0.3.7

utils.py

@@ -0,0 +1,11 @@
+from langchain.llms import OpenAI
+
+def oai_key_isvalid(api_key):
+    """Check if a given OpenAI key is valid"""
+    try:
+        llm = OpenAI(openai_api_key = api_key)
+        out = llm("This is a test")
+        print(out)
+        return True
+    except:
+        return False