Update app.py

app.py

@@ -6,6 +6,16 @@ def echo(message, chat_history):
         chat_history.append((message, bot_message))
         return '', chat_history
 
+def check_openai_api_key(api_key):
+    openai.api_key = api_key
+    try:
+        openai.Model.list()
+    except openai.error.AuthenticationError as e:
+        return False
+    else:
+        return True
+
+
 with gr.Blocks() as demo:
     overview = gr.Markdown("""
     # CentaurSock 
@@ -21,6 +31,8 @@ with gr.Blocks() as demo:
     openai_key_textbox = gr.Textbox(label='OpenAI Key')
     openai_key_button = gr.Button(value='Test OpenAI Key')
 
+    openai_key_button.click(check_openai_api_key, inputs=[openai_key_textbox], outputs=[openai_key_textbox])
+
     # Titles
     with gr.Row():
         ally_title = gr.Markdown("""<center><h2> ALLY </h2></center>""")