test

app.py

@@ -3,8 +3,6 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
 import time
 import openai
 
-openai.api_key = "OPENAI_API_KEY"
-
 # Load the Vicuna 7B v1.3 LMSys model and tokenizer
 model_name = "lmsys/vicuna-7b-v1.3"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
@@ -26,41 +24,43 @@ def respond(message, chat_history):
     time.sleep(2)
     return "", chat_history
 
-def interface():
+def interface(tab_name):
         gr.Markdown(" Description ")
 
-        prompt_POS = gr.Textbox(show_label=False, placeholder="Write a prompt and press enter")
-        openai_key = gr.Textbox(label="Open AI Key", placeholder="Enter your Openai key here", type="password")
+        textbox_prompt = gr.Textbox(show_label=False, placeholder="Write a prompt and press enter")
+        openai.api_key = gr.Textbox(label="Open AI Key", placeholder="Enter your Openai key here", type="password")
+
+        prompt = template_single.format(tab_name, textbox_prompt)
 
         gr.Markdown("Strategy 1 QA-Based Prompting")
         with gr.Row():
-            vicuna_S1_chatbot_POS = gr.Chatbot(label="vicuna-7b")
-            llama_S1_chatbot_POS = gr.Chatbot(label="llama-7b")
-            gpt_S1_chatbot_POS = gr.Chatbot(label="gpt-3.5")
-        clear = gr.ClearButton([prompt_POS, vicuna_S1_chatbot_POS])
+            vicuna_S1_chatbot = gr.Chatbot(label="vicuna-7b")
+            llama_S1_chatbot = gr.Chatbot(label="llama-7b")
+            gpt_S1_chatbot = gr.Chatbot(label="gpt-3.5")
+        clear = gr.ClearButton([prompt, vicuna_S1_chatbot])
         gr.Markdown("Strategy 2 Instruction-Based Prompting")
         with gr.Row():
-            vicuna_S2_chatbot_POS = gr.Chatbot(label="vicuna-7b")
-            llama_S2_chatbot_POS = gr.Chatbot(label="llama-7b")
-            gpt_S2_chatbot_POS = gr.Chatbot(label="gpt-3.5")
-        clear = gr.ClearButton([prompt_POS, vicuna_S2_chatbot_POS])
+            vicuna_S2_chatbot = gr.Chatbot(label="vicuna-7b")
+            llama_S2_chatbot = gr.Chatbot(label="llama-7b")
+            gpt_S2_chatbot = gr.Chatbot(label="gpt-3.5")
+        clear = gr.ClearButton([prompt, vicuna_S2_chatbot])
         gr.Markdown("Strategy 3 Structured Prompting")
         with gr.Row():
-            vicuna_S3_chatbot_POS = gr.Chatbot(label="vicuna-7b")
-            llama_S3_chatbot_POS = gr.Chatbot(label="llama-7b")
-            gpt_S3_chatbot_POS = gr.Chatbot(label="gpt-3.5")
-        clear = gr.ClearButton([prompt_POS, vicuna_S3_chatbot_POS])
+            vicuna_S3_chatbot = gr.Chatbot(label="vicuna-7b")
+            llama_S3_chatbot = gr.Chatbot(label="llama-7b")
+            gpt_S3_chatbot = gr.Chatbot(label="gpt-3.5")
+        clear = gr.ClearButton([prompt, vicuna_S3_chatbot])
 
-        prompt_POS.submit(respond, [prompt_POS, vicuna_S1_chatbot_POS], [prompt_POS, vicuna_S1_chatbot_POS])
-        prompt_POS.submit(respond, [prompt_POS, vicuna_S2_chatbot_POS], [prompt_POS, vicuna_S2_chatbot_POS])
-        prompt_POS.submit(respond, [prompt_POS, vicuna_S3_chatbot_POS], [prompt_POS, vicuna_S3_chatbot_POS])
+        prompt.submit(respond, [prompt, vicuna_S1_chatbot], [prompt, vicuna_S1_chatbot])
+        prompt.submit(respond, [prompt, vicuna_S2_chatbot], [prompt, vicuna_S2_chatbot])
+        prompt.submit(respond, [prompt, vicuna_S3_chatbot], [prompt, vicuna_S3_chatbot])
         
 
 with gr.Blocks() as demo:
     gr.Markdown("# LLM Evaluator With Linguistic Scrutiny")
 
     with gr.Tab("Noun"):
-        interface()
+        interface("Noun")
     
     with gr.Tab("Determiner"):
         gr.Markdown(" Description ")
@@ -87,13 +87,13 @@ with gr.Blocks() as demo:
         clear = gr.ClearButton([prompt_CHUNK, vicuna_S3_chatbot_CHUNK])
     
     with gr.Tab("Noun phrase"):
-        interface()
+        interface("Noun phrase")
     with gr.Tab("Verb phrase"):
-        interface()
+        interface("Verb phrase")
     with gr.Tab("Dependent clause"):
-        interface()
+        interface("Dependent clause")
     with gr.Tab("T-units"):
-        interface()
+        interface("T-units")
 
     prompt_CHUNK.submit(respond, [prompt_CHUNK, vicuna_S1_chatbot_CHUNK], [prompt_CHUNK, vicuna_S1_chatbot_CHUNK])
     prompt_CHUNK.submit(respond, [prompt_CHUNK, vicuna_S2_chatbot_CHUNK], [prompt_CHUNK, vicuna_S2_chatbot_CHUNK])