Update app.py

app.py

@@ -6,11 +6,15 @@ def echo(message, chat_history):
         chat_history.append((message, bot_message))
         return '', chat_history
 
-def check_openai_api_key(api_key):
+def is_api_key_valid(api_key):
     openai.api_key = api_key
     try:
-        openai.Model.list()
-    except openai.error.AuthenticationError as e:
+        response = openai.Completion.create(
+            engine="davinci",
+            prompt="This is a test.",
+            max_tokens=5
+        )
+    except:
         return False
     else:
         return True
@@ -31,7 +35,7 @@ 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])
+    openai_key_button.click(is_api_key_valid, inputs=[openai_key_textbox], outputs=[openai_key_textbox])
 
     # Titles
     with gr.Row():