test another way to make gpt optional

app.py

@@ -89,31 +89,30 @@ def llama_respond(tab_name, message, chat_history):
     time.sleep(2)
     return tab_name, "", chat_history
 
-def gpt_strategies_respond(have_key, strategy, task_name, task_ling_ent, message, chat_history, max_convo_length = 10):
-    if (have_key == "Yes"):
-        formatted_system_prompt = ""
-        if (task_name == "POS Tagging"):
-            if (strategy == "S1"):
-                formatted_system_prompt = f'''Generate the output only for the assistant. Output any {task_ling_ent} in the following sentence one per line without any additional text: {message}'''
-            elif (strategy == "S2"):
-                formatted_system_prompt = f'''POS tag the following sentence using Universal POS tag set without generating any additional text: {message}'''
-            elif (strategy == "S3"):
-                formatted_system_prompt = f'''POS tag the following sentence using Universal POS tag set without generating any additional text: {message}'''
-        elif (task_name == "Chunking"):
-            if (strategy == "S1"):
-                formatted_system_prompt = f'''Generate the output only for the assistant. Output any {task_ling_ent} in the following sentence one per line without any additional text: {message}'''
-            elif (strategy == "S2"):
-                formatted_system_prompt = f'''Chunk the following sentence in CoNLL 2000 format with BIO tags without outputing any additional text: {message}'''
-            elif (strategy == "S3"):
-                formatted_system_prompt = f'''Chunk the following sentence in CoNLL 2000 format with BIO tags without outputing any additional text: {message}'''
+def gpt_strategies_respond(strategy, task_name, task_ling_ent, message, chat_history, max_convo_length = 10):
+    formatted_system_prompt = ""
+    if (task_name == "POS Tagging"):
+        if (strategy == "S1"):
+            formatted_system_prompt = f'''Generate the output only for the assistant. Output any {task_ling_ent} in the following sentence one per line without any additional text: {message}'''
+        elif (strategy == "S2"):
+            formatted_system_prompt = f'''POS tag the following sentence using Universal POS tag set without generating any additional text: {message}'''
+        elif (strategy == "S3"):
+            formatted_system_prompt = f'''POS tag the following sentence using Universal POS tag set without generating any additional text: {message}'''
+    elif (task_name == "Chunking"):
+        if (strategy == "S1"):
+            formatted_system_prompt = f'''Generate the output only for the assistant. Output any {task_ling_ent} in the following sentence one per line without any additional text: {message}'''
+        elif (strategy == "S2"):
+            formatted_system_prompt = f'''Chunk the following sentence in CoNLL 2000 format with BIO tags without outputing any additional text: {message}'''
+        elif (strategy == "S3"):
+            formatted_system_prompt = f'''Chunk the following sentence in CoNLL 2000 format with BIO tags without outputing any additional text: {message}'''
         
-        formatted_prompt = format_chat_prompt(message, chat_history, max_convo_length)
-        print('Prompt + Context:')
-        print(formatted_prompt)
-        bot_message = chat(system_prompt = formatted_system_prompt,
-                        user_prompt = formatted_prompt)
-        chat_history.append((message, bot_message))
-        return "", chat_history
+    formatted_prompt = format_chat_prompt(message, chat_history, max_convo_length)
+    print('Prompt + Context:')
+    print(formatted_prompt)
+    bot_message = chat(system_prompt = formatted_system_prompt,
+                    user_prompt = formatted_prompt)
+    chat_history.append((message, bot_message))
+    return "", chat_history
 
 def vicuna_strategies_respond(strategy, task_name, task_ling_ent, message, chat_history):
     formatted_prompt = ""
@@ -246,7 +245,7 @@ def interface():
             task_prompt = gr.Textbox(show_label=False, placeholder="Write a prompt and press enter")
             with gr.Row():
                 have_key = gr.Dropdown(["Yes", "No"], label="Do you own an API Key?", scale=0)
-                task_apikey_input = gr.Textbox(label="Open AI Key", placeholder="Enter your OpenAI key here", type="password", visible=False)
+                task_apikey_input = gr.Textbox(label="Open AI Key", placeholder="Enter your OpenAI key here", type="password", visible=True)
             task = gr.Dropdown(["POS Tagging", "Chunking"], label="Task")
             task_linguistic_entities = gr.Dropdown(["Noun", "Determiner", "Noun phrase", "Verb phrase", "Dependent clause", "T-units"], label="Linguistic Entity For Strategy 1")
             task_btn = gr.Button(value="Submit")
@@ -276,15 +275,16 @@ def interface():
                                                vicuna_S3_chatbot, llama_S3_chatbot, gpt_S3_chatbot])
             
             # Event Handler for API Key
-            def toggle_api_key_input(value):
-                print("toggle_api_key_input ran")
-                if value == "Yes":
-                    task_apikey_input.visible = True
-                else:
-                    task_apikey_input.visible = False
+            # def toggle_api_key_input(value):
+            #     print("toggle_api_key_input ran")
+            #     if value == "Yes":
+            #         task_apikey_input.visible = True
+            #     else:
+            #         task_apikey_input.visible = False
             
-            have_key.input(toggle_api_key_input, inputs=have_key)
-            task_apikey_input.input(update_api_key, inputs=task_apikey_input)
+            # have_key.input(toggle_api_key_input, inputs=have_key)
+            # task_apikey_input.input(update_api_key, inputs=task_apikey_input)
+            task_btn.click(update_api_key, inputs=task_apikey_input)
             
             # vicuna_strategies_respond(strategy, task_name, task_ling_ent, message, chat_history):
             # Event Handlers for Vicuna Chatbot POS/Chunk
@@ -306,12 +306,13 @@ def interface():
             # Event Handler for GPT 3.5 Chatbot POS/Chunk, user must submit api key before submitting the prompt
             # Will activate after getting API key
             # task_apikey_btn.click(update_api_key, inputs=ling_ents_apikey_input)
-            task_btn.click(gpt_strategies_respond, inputs=[have_key, strategy1, task, task_linguistic_entities, task_prompt, gpt_S1_chatbot], 
-                                outputs=[task, task_prompt, gpt_S1_chatbot])
-            task_btn.click(gpt_strategies_respond, inputs=[have_key, strategy1, task, task_linguistic_entities, task_prompt, gpt_S2_chatbot], 
-                                outputs=[task, task_prompt, gpt_S2_chatbot])
-            task_btn.click(gpt_strategies_respond, inputs=[have_key, strategy1, task, task_linguistic_entities, task_prompt, gpt_S3_chatbot], 
-                                outputs=[task, task_prompt, gpt_S3_chatbot])
+            if (have_key.value == "Yes"):
+                task_btn.click(gpt_strategies_respond, inputs=[strategy1, task, task_linguistic_entities, task_prompt, gpt_S1_chatbot], 
+                                    outputs=[task, task_prompt, gpt_S1_chatbot])
+                task_btn.click(gpt_strategies_respond, inputs=[strategy1, task, task_linguistic_entities, task_prompt, gpt_S2_chatbot], 
+                                    outputs=[task, task_prompt, gpt_S2_chatbot])
+                task_btn.click(gpt_strategies_respond, inputs=[strategy1, task, task_linguistic_entities, task_prompt, gpt_S3_chatbot], 
+                                    outputs=[task, task_prompt, gpt_S3_chatbot])