removed POS/Chunk tab and enabled llama and vicuna

app.py

@@ -182,55 +182,7 @@ def llama_strategies_respond(strategy, task_name, task_ling_ent, message, chat_h
     return task_name, "", chat_history
 
 def interface():
-
-        # prompt = template_single.format(tab_name, textbox_prompt)
-
         with gr.Tab("Linguistic Entities"):
-            gr.Markdown(""" 
-                        ## 📜 Step-By-Step Instructions
-
-                        - Enter a sentence for three models to process (Vicuna-7b, LLaMA-7b and GPT-3.5).
-                        - Enter your OpenAI Api Key and click on 'Submit Key'.
-                        - Select a Linguistic Entity from the Dropdown.
-                        - Click 'Submit' to send your inputs to the models.
-                        - Scroll to the bottom and click 'Clear' to start again.
-
-                        ### 🤖 Now the models will output the linguistic entities found in your prompt based on your selections!
-                        """)
-
-            # Inputs
-            ling_ents_prompt = gr.Textbox(show_label=False, placeholder="Write a prompt and press enter")
-            ling_ents_apikey_input = gr.Textbox(label="Open AI Key", placeholder="Enter your Openai key here", type="password")
-            # ling_ents_apikey_btn = gr.Button(value="Submit Key", scale=0)
-            linguistic_entities = gr.Dropdown(["Noun", "Determiner", "Noun phrase", "Verb phrase", "Dependent clause", "T-units"], label="Linguistic Entity")
-            ling_ents_btn = gr.Button(value="Submit")
-
-            # Outputs
-            gr.Markdown("### Strategy 1 QA-Based Prompting")
-
-            linguistic_features_textbox = gr.Textbox(label="Linguistic Features", disabled=True)
-
-            with gr.Row():
-                vicuna_ling_ents_chatbot = gr.Chatbot(label="vicuna-7b")
-                llama_ling_ents_chatbot = gr.Chatbot(label="llama-7b")
-                gpt_ling_ents_chatbot = gr.Chatbot(label="gpt-3.5")
-            clear = gr.ClearButton(components=[ling_ents_prompt, ling_ents_apikey_input, vicuna_ling_ents_chatbot, llama_ling_ents_chatbot, gpt_ling_ents_chatbot])
-
-            # Event Handler for Vicuna Chatbot
-            ling_ents_btn.click(vicuna_respond, inputs=[linguistic_entities, ling_ents_prompt, vicuna_ling_ents_chatbot], 
-                      outputs=[linguistic_entities, ling_ents_prompt, vicuna_ling_ents_chatbot])
-            
-            # Event Handler for LLaMA Chatbot
-            ling_ents_btn.click(llama_respond, inputs=[linguistic_entities, ling_ents_prompt, llama_ling_ents_chatbot], 
-                      outputs=[linguistic_entities, ling_ents_prompt, llama_ling_ents_chatbot])
-            
-            # Event Handler for GPT 3.5 Chatbot, user must submit api key before submitting the prompt
-            # Will activate after getting API key
-            # ling_ents_apikey_btn.click(update_api_key, inputs=ling_ents_apikey_input)
-            # ling_ents_btn.click(gpt_respond, inputs=[linguistic_entities, ling_ents_prompt, gpt_ling_ents_chatbot], 
-            #                     outputs=[linguistic_entities, ling_ents_prompt, gpt_ling_ents_chatbot])
-
-        with gr.Tab("POS/Chunking"):
             gr.Markdown("""
                         ## 📜 Step-By-Step Instructions
 
@@ -278,33 +230,24 @@ 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
-            
-            # 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
-            # task_btn.click(vicuna_strategies_respond, inputs=[strategy1, task, task_linguistic_entities, task_prompt, vicuna_S1_chatbot], 
-            #           outputs=[task, task_prompt, vicuna_S1_chatbot])
-            # task_btn.click(vicuna_strategies_respond, inputs=[strategy2, task, task_linguistic_entities, task_prompt, vicuna_S2_chatbot], 
-            #           outputs=[task, task_prompt, vicuna_S2_chatbot])
-            # task_btn.click(vicuna_strategies_respond, inputs=[strategy3, task, task_linguistic_entities, task_prompt, vicuna_S3_chatbot], 
-            #           outputs=[task, task_prompt, vicuna_S3_chatbot])
+            task_btn.click(vicuna_strategies_respond, inputs=[strategy1, task, task_linguistic_entities, task_prompt, vicuna_S1_chatbot], 
+                      outputs=[task, task_prompt, vicuna_S1_chatbot])
+            task_btn.click(vicuna_strategies_respond, inputs=[strategy2, task, task_linguistic_entities, task_prompt, vicuna_S2_chatbot], 
+                      outputs=[task, task_prompt, vicuna_S2_chatbot])
+            task_btn.click(vicuna_strategies_respond, inputs=[strategy3, task, task_linguistic_entities, task_prompt, vicuna_S3_chatbot], 
+                      outputs=[task, task_prompt, vicuna_S3_chatbot])
             
             # Event Handler for LLaMA Chatbot POS/Chunk
-            # task_btn.click(llama_strategies_respond, inputs=[strategy1, task, task_linguistic_entities, task_prompt, llama_S1_chatbot], 
-            #           outputs=[task, task_prompt, llama_S1_chatbot])
-            # task_btn.click(llama_strategies_respond, inputs=[strategy2, task, task_linguistic_entities, task_prompt, llama_S2_chatbot], 
-            #           outputs=[task, task_prompt, llama_S2_chatbot])
-            # task_btn.click(llama_strategies_respond, inputs=[strategy3, task, task_linguistic_entities, task_prompt, llama_S3_chatbot], 
-            #           outputs=[task, task_prompt, llama_S3_chatbot])
+            task_btn.click(llama_strategies_respond, inputs=[strategy1, task, task_linguistic_entities, task_prompt, llama_S1_chatbot], 
+                      outputs=[task, task_prompt, llama_S1_chatbot])
+            task_btn.click(llama_strategies_respond, inputs=[strategy2, task, task_linguistic_entities, task_prompt, llama_S2_chatbot], 
+                      outputs=[task, task_prompt, llama_S2_chatbot])
+            task_btn.click(llama_strategies_respond, inputs=[strategy3, task, task_linguistic_entities, task_prompt, llama_S3_chatbot], 
+                      outputs=[task, task_prompt, llama_S3_chatbot])
             
             # Event Handler for GPT 3.5 Chatbot POS/Chunk, user must submit api key before submitting the prompt
             # Will activate after getting API key