Update app.py

app.py

@@ -7,7 +7,6 @@ import langchain
 import faiss
 
 
-openai.api_key = os.environ['openai_key']
 
 
 prompt_templates = {"All Needs Gurus": "I want you to act as a needs assessment expert."}
@@ -35,6 +34,9 @@ def download_prompt_templates():
     choices = choices[:1] + sorted(choices[1:])
     return gr.update(value=choices[0], choices=choices)
 
+def key_change(key):
+    openai.api_key = key
+
 
 def on_prompt_template_change(prompt_template):
     if not isinstance(prompt_template, str): return
@@ -111,6 +113,7 @@ with gr.Blocks(css=css) as demo:
             with gr.Column():
                 #gr.Markdown("Enter your OpenAI API Key. You can get one [here](https://platform.openai.com/account/api-keys).", elem_id="label")
                 #user_token = gr.Textbox(value='', placeholder="OpenAI API Key", type="password", show_label=False)
+                key = os.environ['openai_key']
                 prompt_template = gr.Dropdown(label="Choose a guru:", choices=list(prompt_templates.keys()))
                 prompt_template_preview = gr.Markdown(elem_id="prompt_template_preview")
                 with gr.Accordion("Advanced parameters", open=False):
@@ -120,12 +123,12 @@ with gr.Blocks(css=css) as demo:
 
     gr.HTML('''<br><br><br><center>You can duplicate this Space to skip the queue:<a href="https://huggingface.co/spaces/anzorq/chatgpt-demo?duplicate=true"><img src="https://bit.ly/3gLdBN6" alt="Duplicate Space"></a><br>
             <p><img src="https://visitor-badge.glitch.me/badge?page_id=anzorq.chatgpt_api_demo_hf" alt="visitors"></p></center>''')
-    key = os.environ['openai_key']
+    
     btn_submit.click(submit_message, [key, input_message, prompt_template, temperature, max_tokens, context_length, state], [input_message, chatbot, total_tokens_str, state])
     input_message.submit(submit_message, [key, input_message, prompt_template, temperature, max_tokens, context_length, state], [input_message, chatbot, total_tokens_str, state])
     btn_clear_conversation.click(clear_conversation, [], [input_message, chatbot, total_tokens_str, state])
     prompt_template.change(on_prompt_template_change, inputs=[prompt_template], outputs=[prompt_template_preview])
-    #user_token.change(on_token_change, inputs=[user_token], outputs=[])
+    key.change(key_change, inputs=[key], outputs=[])
 
     
     demo.load(download_prompt_templates, inputs=None, outputs=[prompt_template], queur=False)