~UI

app.py

@@ -29,7 +29,7 @@ def download_prompt_templates():
     return gr.update(value=choices[0], choices=choices)
 
 def on_token_change(user_token):
-    openai.api_key = user_token or os.environ.get("OPENAI_API_KEY")
+    openai.api_key = user_token
 
 def on_prompt_template_change(prompt_template):
     if not isinstance(prompt_template, str): return
@@ -49,6 +49,14 @@ def submit_message(user_token, prompt, prompt_template, temperature, max_tokens,
         system_prompt = [{ "role": "system", "content": prompt_template }]
 
     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 + [prompt_msg], temperature=temperature, max_tokens=max_tokens)
@@ -68,7 +76,7 @@ def submit_message(user_token, prompt, prompt_template, temperature, max_tokens,
     total_tokens_used_msg = f"Total tokens used: {state['total_tokens']}"
     chat_messages = [(history[i]['content'], history[i+1]['content']) for i in range(0, len(history)-1, 2)]
 
-    return gr.update(value=''), chat_messages, total_tokens_used_msg, state
+    return '', chat_messages, total_tokens_used_msg, state
 
 def clear_conversation():
     return gr.update(value=None, visible=True), None, "", get_empty_state()
@@ -97,7 +105,7 @@ with gr.Blocks(css=css) as demo:
         with gr.Column():
             with gr.Row():
                 gr.Markdown("Enter your own OpenAI API Key. You can get one [here](https://platform.openai.com/account/api-keys).", elem_id="label")
-                user_token = gr.Textbox(value='stub', placeholder="OpenAI API Key", type="password", show_label=False)
+                user_token = gr.Textbox(value='', placeholder="OpenAI API Key", type="password", show_label=False)
             with gr.Row():
                 with gr.Column():
                     chatbot = gr.Chatbot(elem_id="chatbox")