test

app.py

@@ -13,20 +13,50 @@ with gr.Blocks() as demo:
     with gr.Tab("POS"):
         gr.Markdown(" Description ")
 
-        msg = gr.Textbox(show_label=False, placeholder="Write a prompt and press enter")
+        prompt_POS = gr.Textbox(show_label=False, placeholder="Write a prompt and press enter")
 
         gr.Markdown("Strategy 1 QA")
         with gr.Row():
             vicuna_S1_chatbot_POS = gr.Chatbot(label="vicuna-7b")
-            clear = gr.ClearButton([msg, vicuna_S1_chatbot_POS])
+            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])
         gr.Markdown("Strategy 2 Instruction")
         with gr.Row():
             vicuna_S2_chatbot_POS = gr.Chatbot(label="vicuna-7b")
-            clear = gr.ClearButton([msg, vicuna_S2_chatbot_POS])
-        gr.Markdown("Strategy 1 Structured Prompting")
+            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])
+        gr.Markdown("Strategy 3 Structured Prompting")
         with gr.Row():
             vicuna_S3_chatbot_POS = gr.Chatbot(label="vicuna-7b")
-            clear = gr.ClearButton([msg, vicuna_S3_chatbot_POS])
+            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])
+    
+    with gr.Tab("Chunk"):
+        gr.Markdown(" Description ")
+
+        prompt_CHUNK = gr.Textbox(show_label=False, placeholder="Write a prompt and press enter")
+
+        gr.Markdown("Strategy 1 QA")
+        with gr.Row():
+            vicuna_S1_chatbot_CHUNK = gr.Chatbot(label="vicuna-7b")
+            llama_S1_chatbot_CHUNK = gr.Chatbot(label="llama-7b")
+            gpt_S1_chatbot_CHUNK = gr.Chatbot(label="gpt-3.5")
+        clear = gr.ClearButton([msg, vicuna_S1_chatbot_CHUNK])
+        gr.Markdown("Strategy 2 Instruction")
+        with gr.Row():
+            vicuna_S2_chatbot_CHUNK = gr.Chatbot(label="vicuna-7b")
+            llama_S2_chatbot_CHUNK = gr.Chatbot(label="llama-7b")
+            gpt_S2_chatbot_CHUNK = gr.Chatbot(label="gpt-3.5")
+        clear = gr.ClearButton([msg, vicuna_S2_chatbot_CHUNK])
+        gr.Markdown("Strategy 3 Structured Prompting")
+        with gr.Row():
+            vicuna_S3_chatbot_CHUNK = gr.Chatbot(label="vicuna-7b")
+            llama_S3_chatbot_CHUNK = gr.Chatbot(label="llama-7b")
+            gpt_S3_chatbot_CHUNK = gr.Chatbot(label="gpt-3.5")
+        clear = gr.ClearButton([msg, vicuna_S3_chatbot_CHUNK])
 
     def respond(message, chat_history):
         input_ids = tokenizer.encode(message, return_tensors="pt")
@@ -37,8 +67,12 @@ with gr.Blocks() as demo:
         time.sleep(2)
         return "", chat_history
 
-    msg.submit(respond, [msg, vicuna_S1_chatbot_POS], [msg, vicuna_S1_chatbot_POS])
-    msg.submit(respond, [msg, vicuna_S2_chatbot_POS], [msg, vicuna_S2_chatbot_POS])
-    msg.submit(respond, [msg, vicuna_S3_chatbot_POS], [msg, vicuna_S3_chatbot_POS])
+    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_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])
+    prompt_CHUNK.submit(respond, [prompt_CHUNK, vicuna_S3_chatbot_CHUNK], [prompt_CHUNK, vicuna_S3_chatbot_CHUNK])
 
 demo.launch()