test

app.py

@@ -90,7 +90,7 @@ with gr.Blocks() as demo:
         
         chat_history.append((message, bot_message))
         time.sleep(2)
-        return bot_message
+        return entity, message, bot_message
 
     submit_btn.click(respond_entities, [entity_dropdown, prompt_POS, vicuna_S1_chatbot_POS], [entity_dropdown, prompt_POS, vicuna_S1_chatbot_POS])
     submit_btn.click(respond_entities, [entity_dropdown, prompt_POS, vicuna_S2_chatbot_POS], [entity_dropdown, prompt_POS, vicuna_S2_chatbot_POS])