Update app.py

app.py

@@ -22,7 +22,6 @@ def get_api_key():
 
 def set_model(model):
     os.environ["model_type"] = model
-    return f"Your API Key has been set to: {model}"
 
 def get_model():
     model = os.getenv("model_type")
@@ -72,17 +71,17 @@ with gr.Blocks() as demo:
     
     with gr.Tab("Coding Assistant"):
         api_check_button = gr.Button("Get API Key")
+        api_print = gr.Textbox(label = "OpenAI API Key - Please ensure the API Key is set correctly")
         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!"
         ),
         model_submit_button = gr.Button("Submit Model Selection")
-        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")
         clear = gr.Button("Clear")
         
-    api_submit_button.click(set_api_key, inputs=model_selection)
-    model_submit_button.click(set_model, inputs=api_input, outputs=api_key_status)
+    api_submit_button.click(set_api_key, inputs=api_input)
+    model_submit_button.click(set_model, inputs=model_selection)
     api_reset_button.click(reset_api_key, outputs=api_key_status)
     api_check_button.click(get_api_key, outputs=api_print)
     msg.submit(respond, inputs = [msg, chatbot], outputs = [msg, chatbot])