llm error handling + buttons.

app.py

@@ -4,6 +4,7 @@ import os
 if 'OPENAI_API_KEY' not in os.environ:
     os.environ['OPENAI_API_KEY'] = 'none'
 
+import openai
 import pandas as pd
 import streamlit as st
 from IPython.core.display import HTML
@@ -43,7 +44,7 @@ agent = ChemCrow(
     temp=0.1,
     openai_api_key=ss.get('api_key'),
     api_keys={
-        'rxn4chem':st.secrets['RXN4CHEM_API_KEY']
+        'rxn4chem': st.secrets['RXN4CHEM_API_KEY']
     }
 ).agent_executor
 
@@ -69,7 +70,6 @@ pre_prompts = [
 
 
 def run_prompt(prompt):
-
     st.chat_message("user").write(prompt)
     with st.chat_message("assistant"):
         st_callback = StreamlitCallbackHandlerChem(
@@ -78,9 +78,12 @@ def run_prompt(prompt):
             collapse_completed_thoughts = False,
             output_placeholder=ss
         )
-        with wandb_tracing_enabled():
-            response = agent.run(prompt, callbacks=[st_callback])
-            st.write(response)
+        try:
+            with wandb_tracing_enabled():
+                response = agent.run(prompt, callbacks=[st_callback])
+                st.write(response)
+        except openai.error.AuthenticationError:
+            st.write("Please input a valid OpenAI API key")
 
 
 # sidebar