test

app.py

@@ -67,19 +67,22 @@ with gr.Blocks() as demo:
         return response
 
     # Define the Gradio interface
-    def chatbot_interface_POS(prompt_POS):
+    def chatbot_interface_POS(input_dict):
+        prompt_POS = input_dict["prompt_POS"]
         vicuna_response_POS = generate_response(prompt_POS)
         # Add responses from other chatbots if needed
         return {"Vicuna-7B": vicuna_response_POS}
 
-    def chatbot_interface_Chunk(prompt_chunk):
+    def chatbot_interface_Chunk(input_dict):
+        prompt_chunk = input_dict["prompt_chunk"]
         vicuna_response_chunk = generate_response(prompt_chunk)
         # Add responses from other chatbots if needed
         return {"Vicuna-7B": vicuna_response_chunk}
 
     # Connect the interfaces to the functions
-    send_button_POS.click(chatbot_interface_POS, [prompt_POS, vicuna_chatbot1_POS])
-    send_button_Chunk.click(chatbot_interface_Chunk, [prompt_chunk, vicuna_chatbot1_chunk])
+    send_button_POS.click(chatbot_interface_POS, {"prompt_POS": prompt_POS, "vicuna_chatbot1_POS": vicuna_chatbot1_POS})
+    send_button_Chunk.click(chatbot_interface_Chunk, {"prompt_chunk": prompt_chunk, "vicuna_chatbot1_chunk": vicuna_chatbot1_chunk})
+
 
 
 demo.launch()