Update app.py

app.py

@@ -11,8 +11,11 @@ from langchain.llms import OpenAI
 
 def set_api_key(key):
     os.environ["OPENAI_API_KEY"] = key
-    time.sleep(600)
+    return f"Your API Key has been set to {key}"
+
+def reset_api_key():
     os.environ["OPENAI_API_KEY"] = ""
+    return api_key_status: gr.update(placeholder="Your API Key has been reset")
     
 def get_api_key():
     api_key = os.getenv("OPENAI_API_KEY")
@@ -45,10 +48,11 @@ with gr.Blocks() as demo:
     with gr.Tab("OpenAI API Key Submission"):
         api_input = gr.Textbox(label = "API Key - Your key will be active for 10 minutes. The bar below shows the time left.",
                                placeholder = "Please provide your OpenAI API key here.")
-        api_submission_conf = gr.Textbox(label = "API Key Status",
-                                         placeholder = "Your API Key has been reset or has not be set yet. Please re-enter your key.",
+        api_key_status = gr.Textbox(label = "API Key Status",
+                                         placeholder = "Your API Key has been reset or has not be set yet. Please enter your key.",
                                          interactive = False)
         api_submit_button = gr.Button("Submit")
+        api_reset_button = gr.Button("Clear API Key from session")
     
     with gr.Tab("Coding Assistant"):
         api_check_button = gr.Button("Get API Key")
@@ -57,7 +61,8 @@ with gr.Blocks() as demo:
         msg = gr.Textbox(label="User Prompt", placeholder="Your Query Here")
         clear = gr.Button("Clear")
         
-    api_submit_button.click(set_api_key, inputs=api_input, outputs=api_submission_conf, queue=False)
+    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])
     clear.click(lambda: None, None, chatbot, queue=False)