adding more models from openai

app.py

@@ -39,18 +39,22 @@ st.markdown(
     unsafe_allow_html=True,
 )
 
-agent = ChemCrow(
-    model='gpt-4-0613',
-    tools_model='gpt-4-turbo-preview',
-    temp=0.1,
-    openai_api_key=ss.get('api_key'),
-    api_keys={
-        'RXN4CHEM_API_KEY': st.secrets['RXN4CHEM_API_KEY'],
-        'CHEMSPACE_API_KEY': st.secrets['CHEMSPACE_API_KEY']
-    }
-).agent_executor
 
-tools = agent.tools
+def instantiate_agent(model):
+    ss.agent = ChemCrow(
+        model=model,
+        tools_model=model,
+        temp=0.1,
+        openai_api_key=ss.get('api_key'),
+        api_keys={
+            'RXN4CHEM_API_KEY': st.secrets['RXN4CHEM_API_KEY'],
+            'CHEMSPACE_API_KEY': st.secrets['CHEMSPACE_API_KEY']
+        }
+    ).agent_executor
+    return ss.agent
+
+instantiate_agent('gpt-4-0613')
+tools = ss.agent.tools
 
 tool_list = pd.Series(
     {f"✅ {t.name}":t.description for t in tools}
@@ -63,8 +67,8 @@ def on_api_key_change():
     if not oai_key_isvalid(api_key):
         st.write("Please input a valid OpenAI API key.")
 
-
 def run_prompt(prompt):
+    agent = instantiate_agent(ss.get('model_select'))
     st.chat_message("user").write(prompt)
     with st.chat_message("assistant"):
         st_callback = StreamlitCallbackHandlerChem(
@@ -113,6 +117,13 @@ with st.sidebar:
         label_visibility="collapsed"
     )
 
+    # Input model to use
+    st.selectbox(
+        'Select model to use',
+        ['gpt-4-0613', 'gpt-3.5-turbo', 'gpt-4o-mini'],
+        key='model_select',
+    )
+
     # Display prompt examples
     st.markdown('# What can I ask?')
     cols = st.columns(2)