testing gpt generation

app.py

@@ -88,30 +88,31 @@ def llama_respond(tab_name, message, chat_history):
     time.sleep(2)
     return tab_name, "", chat_history
 
-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
+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}'''
+        
+        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 = ""
@@ -303,13 +304,12 @@ 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)
-            if (have_key == "Yes"):
-                task_btn.click(gpt_strategies_respond, inputs=[strategy1, task, task_linguistic_entities, gpt_S1_chatbot], 
-                                    outputs=[task, task_prompt, gpt_S1_chatbot])
-                task_btn.click(gpt_strategies_respond, inputs=[strategy1, task, task_linguistic_entities, gpt_S2_chatbot], 
-                                    outputs=[task, task_prompt, gpt_S2_chatbot])
-                task_btn.click(gpt_strategies_respond, inputs=[strategy1, task, task_linguistic_entities, gpt_S3_chatbot], 
-                                    outputs=[task, task_prompt, gpt_S3_chatbot])
+            task_btn.click(gpt_strategies_respond, inputs=[have_key, strategy1, task, task_linguistic_entities, gpt_S1_chatbot], 
+                                outputs=[task, task_prompt, gpt_S1_chatbot])
+            task_btn.click(gpt_strategies_respond, inputs=[have_key, strategy1, task, task_linguistic_entities, gpt_S2_chatbot], 
+                                outputs=[task, task_prompt, gpt_S2_chatbot])
+            task_btn.click(gpt_strategies_respond, inputs=[have_key, strategy1, task, task_linguistic_entities, gpt_S3_chatbot], 
+                                outputs=[task, task_prompt, gpt_S3_chatbot])