Update app.py

app.py

@@ -20,7 +20,7 @@ def get_api_key():
     api_key = os.getenv("OPENAI_API_KEY")
     return api_key
 
-def respond(message, chat_history):
+def respond(message, chat_history, model_type):
 
     # Get embeddings
     embeddings = OpenAIEmbeddings()
@@ -37,7 +37,7 @@ def respond(message, chat_history):
         chat_history = [(chat_history[0][0], chat_history[0][1])]
     
     # Create ChatOpenAI and ConversationalRetrievalChain
-    model = ChatOpenAI(model='gpt-3.5-turbo')
+    model = ChatOpenAI(model=model_type)
     qa = ConversationalRetrievalChain.from_llm(model, retriever)
 
     bot_message = qa({"question": message, "chat_history": chat_history})
@@ -64,6 +64,9 @@ with gr.Blocks() as demo:
     
     with gr.Tab("Coding Assistant"):
         api_check_button = gr.Button("Get API Key")
+        model_selection = gr.Dropdown(
+            ["gpt-3.5-turbo", "gpt-4"], label="Model Selection", info="Please ensure you provide the API Key that corresponds to the Model you select!"
+        ),
         api_print = gr.Textbox(label = "OpenAI API Key - Please ensure the API Key is set correctly")
         chatbot = gr.Chatbot(label="ChatGPT Powered Coding Assistant")
         msg = gr.Textbox(label="User Prompt", placeholder="Your Query Here")
@@ -72,7 +75,7 @@ with gr.Blocks() as demo:
     api_submit_button.click(set_api_key, inputs=api_input, outputs=api_key_status)
     api_reset_button.click(reset_api_key, outputs=api_key_status)
     api_check_button.click(get_api_key, outputs=api_print)
-    msg.submit(respond, [msg, chatbot], [msg, chatbot])
+    msg.submit(respond, [msg, chatbot, model_selection], [msg, chatbot])
     clear.click(lambda: None, None, chatbot, queue=False)
     
 demo.launch()