Update app.py

app.py

@@ -33,8 +33,8 @@ 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_token_change(user_token):
+    openai.api_key = user_token
 
 def on_prompt_template_change(prompt_template):
     if not isinstance(prompt_template, str): return
@@ -55,7 +55,7 @@ def submit_message( prompt, prompt_template, temperature, max_tokens, context_le
 
     prompt_msg = { "role": "user", "content": prompt }
 
-    if not API:
+    if not user_token:
         history.append(prompt_msg)
         history.append({
             "role": "system",
@@ -125,7 +125,7 @@ with gr.Blocks(css=css) as demo:
                 btn_clear_conversation = gr.Button("🔃 Start New Conversation")
              with gr.Column():
                 #gr.Markdown("Enter your OpenAI API Key.", elem_id="label")
-                #user_token=gr.Textbox(value=os.getenv("api_key"),type='password',visible=False)
+                user_token=gr.Text(value=API)
                 prompt_template = gr.Dropdown(label="Set a custom insruction for the chatbot:", choices=list(prompt_templates.keys()))
                 prompt_template_preview = gr.Markdown(elem_id="prompt_template_preview", visible=False)
                 with gr.Accordion("Advanced parameters", open=False, visible=False):
@@ -135,13 +135,13 @@ with gr.Blocks(css=css) as demo:
 
     print(user_token)  
 
-    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, [ user_token,input_message, prompt_template, temperature, max_tokens, context_length, state], [input_message, chatbot, total_tokens_str, state])
+    input_message.submit(submit_message, [user_token, 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=[])
-    #user_token.submit()
-    openai.api_key = API
+    user_token.change(on_token_change, inputs=[user_token], outputs=[])
+   
+    
     demo.load(download_prompt_templates, inputs=None, outputs=[prompt_template], queur=False)
 
 print(user_token)