Update app.py

app.py

@@ -6,6 +6,8 @@ import os
 import langchain
 import faiss
 
+
+
 prompt_templates = {"All Needs Gurus": "I want you to act as a needs assessment expert."}
 
 def get_empty_state():
@@ -31,16 +33,17 @@ 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
     return prompt_templates[prompt_template]
 
-def submit_message(key, prompt, prompt_template, temperature, max_tokens, context_length, state):
+def submit_message(prompt, prompt_template, temperature, max_tokens, context_length, state):
 
+    openai.api_key = os.environ['openai_key']
     history = state['messages']
 
     if not prompt:
@@ -108,7 +111,7 @@ with gr.Blocks(css=css) as demo:
                 total_tokens_str = gr.Markdown(elem_id="total_tokens_str")
                 btn_clear_conversation = gr.Button("🔃 Start New Conversation")
             with gr.Column():
-                key = os.environ['openai_key']
+           
                 #gr.Markdown(key, elem_id="label")
                 #user_token = gr.Textbox(value='', placeholder="OpenAI API Key", type="password", show_label=False)
                 prompt_template = gr.Dropdown(label="Choose a guru:", choices=list(prompt_templates.keys()))
@@ -121,7 +124,7 @@ 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>''')
     
-    btn_submit.click(submit_message, [key, input_message, prompt_template, temperature, max_tokens, context_length, state], [input_message, chatbot, total_tokens_str, state])
+    btn_submit.click(submit_message, [ input_message, prompt_template, temperature, max_tokens, context_length, state], [input_message, chatbot, total_tokens_str, state])
     input_message.submit(submit_message, [ 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])