Update app.py

app.py

@@ -2,14 +2,14 @@ import openai
 import gradio as gr
 
 # Define the chat function
-def chat(api_key, message):
+def chat(api_key, model, message):
     # Check if an API key has been provided
     if api_key is None:
         return "Please enter your OpenAI API key and try again."
     
     # Set up the OpenAI API request
     response = openai.Completion.create(
-        engine="text-davinci-003",
+        engine=model,
         prompt=message,
         max_tokens=1024,
         n=1,
@@ -25,12 +25,17 @@ def chat(api_key, message):
 
 # Define the Gradio interface
 api_key_input = gr.inputs.Textbox(label="OpenAI API Key", default=None)
+model_input = gr.inputs.Dropdown(
+    label="Select model",
+    choices=["text-davinci-003", "text-davinci-002"],
+    default="text-davinci-003",
+)
 message_input = gr.inputs.Textbox(label="Enter your message here")
 output = gr.outputs.Textbox(label="Bot response")
 
 chat_button = gr.Interface(
     fn=chat,
-    inputs=[api_key_input, message_input],
+    inputs=[api_key_input, model_input, message_input],
     outputs=output,
     title="OpenAI Chatbot",
     description="Enter your message below to chat with an AI",