Update app.py

app.py

@@ -35,14 +35,12 @@ def download_prompt_templates():
     choices = choices[:1] + sorted(choices[1:])
     return gr.update(value=choices[0], choices=choices)
 
-#def on_token_change(user_token):
-#    openai.api_key = user_token
 
 def on_prompt_template_change(prompt_template):
     if not isinstance(prompt_template, str): return
     return prompt_templates[prompt_template]
 
-def submit_message(user_token, prompt, prompt_template, temperature, max_tokens, context_length, state):
+def submit_message(key, prompt, prompt_template, temperature, max_tokens, context_length, state):
 
     history = state['messages']
 
@@ -57,14 +55,7 @@ def submit_message(user_token, prompt, prompt_template, temperature, max_tokens,
 
     prompt_msg = { "role": "user", "content": prompt }
 
-    if not user_token:
-        history.append(prompt_msg)
-        history.append({
-            "role": "system",
-            "content": "Error: OpenAI API Key is not set."
-        })
-        return '', [(history[i]['content'], history[i+1]['content']) for i in range(0, len(history)-1, 2)], f"Total tokens used: 0", state
-    
+
     try:
         completion = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=system_prompt + history[-context_length*2:] + [prompt_msg], temperature=temperature, max_tokens=max_tokens)
 
@@ -130,8 +121,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, [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=[])