test

app.py

@@ -11,7 +11,10 @@ model = AutoModelForCausalLM.from_pretrained(model_name)
 
 template_single = '''Please output any <{}> in the following sentence one per line without any additional text: "{}"'''
 
-global api_key
+def update_api_key(new_key):
+    global api_key
+    os.environ['OPENAI_API_TOKEN'] = new_key.value
+    openai.api_key = os.environ['OPENAI_API_TOKEN']
 
 def chat(system_prompt, user_prompt, model = 'gpt-3.5-turbo', temperature = 0, verbose = False):
     ''' Normal call of OpenAI API '''
@@ -62,13 +65,11 @@ def interface():
         gr.Markdown(" Description ")
 
         textbox_prompt = gr.Textbox(show_label=False, placeholder="Write a prompt and press enter")
-        api_key_input = gr.Textbox(label="Open AI Key", placeholder="Enter your Openai key here", type="password")
+        with gr.Row():
+            api_key_input = gr.Textbox(label="Open AI Key", placeholder="Enter your Openai key here", type="password")
+            api_key_btn = gr.Button(label="Submit Api Key", scale=0)
         tab_name = gr.Dropdown(["Noun", "Determiner", "Noun phrase", "Verb phrase", "Dependent clause", "T-units"], label="Linguistic Entity")
-        btn = gr.Button("Submit")
-
-        #API Keys
-        os.environ['OPENAI_API_TOKEN'] = api_key_input.value
-        openai.api_key = os.environ['OPENAI_API_TOKEN']
+        btn = gr.Button(label="Submit")
 
         # prompt = template_single.format(tab_name, textbox_prompt)
 
@@ -95,6 +96,7 @@ def interface():
         textbox_prompt.submit(respond, inputs=[textbox_prompt, vicuna_S2_chatbot], outputs=[textbox_prompt, vicuna_S2_chatbot])
         textbox_prompt.submit(respond, inputs=[textbox_prompt, vicuna_S3_chatbot], outputs=[textbox_prompt, vicuna_S3_chatbot])
 
+        api_key_btn.click(update_api_key, inputs=api_key_input)
         btn.click(respond_gpt, inputs=[tab_name, textbox_prompt, gpt_S1_chatbot], outputs=[tab_name, textbox_prompt, gpt_S1_chatbot])
 
 with gr.Blocks() as demo: