Update app.py

app.py

@@ -4,11 +4,9 @@ import requests
 import csv
 import os
 
-
-
-API = os.environ.get("api_key")
-openai.api_key = API
-print(API)
+API= os.environ.get("api_key")
+user_token = os.environ.get("api_key")
+print(user_token)
 
 prompt_templates = {"Default ChatGPT": ""}
 
@@ -35,14 +33,14 @@ 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 = API
+#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( prompt, prompt_template, temperature, max_tokens, context_length, state):
 
     history = state['messages']
 
@@ -57,7 +55,7 @@ def submit_message(user_token, prompt, prompt_template, temperature, max_tokens,
 
     prompt_msg = { "role": "user", "content": prompt }
 
-    if not user_token:
+    if not API:
         history.append(prompt_msg)
         history.append({
             "role": "system",
@@ -84,7 +82,7 @@ def submit_message(user_token, prompt, prompt_template, temperature, max_tokens,
     chat_messages = [(history[i]['content'], history[i+1]['content']) for i in range(0, len(history)-1, 2)]
 
     return '', chat_messages, total_tokens_used_msg, state
-#print(user_token)
+print(user_token)
 
 def clear_conversation():
     return gr.update(value=None, visible=True), None, "", get_empty_state()
@@ -107,7 +105,7 @@ css = """
      
       }
       """
-#print(user_token)
+print(user_token)
 with gr.Blocks(css=css) as demo:
     
     state = gr.State(get_empty_state())
@@ -126,28 +124,26 @@ 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():
-                gr.Markdown("Enter your OpenAI API Key.", elem_id="label",visible=False)
-                user_token=gr.Textbox(value="",type='password',visible=False)
-                user_token.update(API)
+                #gr.Markdown("Enter your OpenAI API Key.", elem_id="label")
+                #user_token=gr.Textbox(value=os.getenv("api_key"),type='password',visible=False)
                 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):
                     temperature = gr.Slider(minimum=0, maximum=2.0, value=0.7, step=0.1, label="Temperature", info="Higher = more creative/chaotic")
                     max_tokens = gr.Slider(minimum=100, maximum=4096, value=1000, step=1, label="Max tokens per response")
                     context_length = gr.Slider(minimum=1, maximum=10, value=2, step=1, label="Context length", info="Number of previous messages to send to the chatbot. Be careful with high values, it can blow up the token budget quickly.")
-                    
+
     print(user_token)  
-    
-    
-    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_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])
-    user_token.change(on_token_change, inputs=[user_token], outputs=[])
-    
-    
+    #user_token.change(on_token_change, inputs=[user_token], outputs=[])
+    #user_token.submit()
+    openai.api_key = API
     demo.load(download_prompt_templates, inputs=None, outputs=[prompt_template], queur=False)
-openai.api_key = API
-print(openai.api_key)
+
+print(user_token)
 demo.queue(concurrency_count=10)
 demo.launch(height='1000px',width='800px')