Update app.py

app.py

@@ -2,6 +2,10 @@ import gradio as gr
 import openai
 import requests
 import csv
+import os
+
+
+openai_key = os.environ['openai_key']
 
 prompt_templates = {"All Needs Gurus": "I want you to act as a needs assessment expert."}
 
@@ -123,8 +127,8 @@ 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, [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_submit.click(submit_message, [openai_key, input_message, prompt_template, temperature, max_tokens, context_length, state], [input_message, chatbot, total_tokens_str, state])
+    input_message.submit(submit_message, [openai_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=[])