Update app.py

app.py

@@ -25,8 +25,27 @@ def build_question_selector_map(questions):
 
     return question_selector_map
 
-def display_question_answer(question, cot, request: gr.Request):
-    q = question_selector_map[question]
+def math_display_question_answer(question, cot, request: gr.Request):
+    if cot:
+        q = math_cot_question_selector_map[question]
+    else:
+        q = math_question_selector_map[question]
+
+    return q["agent_response"]["llama"][0], q["agent_response"]["wizardlm"][0], q["agent_response"]["orca"][0], q["agent_response"]["llama"][1], q["agent_response"]["wizardlm"][1], q["agent_response"]["orca"][1], q["agent_response"]["llama"][2], q["agent_response"]["wizardlm"][2], q["agent_response"]["orca"][2]
+
+def gsm_display_question_answer(question, cot, request: gr.Request):
+    if cot:
+        q = gsm_cot_question_selector_map[question]
+    else:
+        q = gsm_question_selector_map[question]
+
+    return q["agent_response"]["llama"][0], q["agent_response"]["wizardlm"][0], q["agent_response"]["orca"][0], q["agent_response"]["llama"][1], q["agent_response"]["wizardlm"][1], q["agent_response"]["orca"][1], q["agent_response"]["llama"][2], q["agent_response"]["wizardlm"][2], q["agent_response"]["orca"][2]
+
+def mmlu_display_question_answer(question, cot, request: gr.Request):
+    if cot:
+        q = mmlu_cot_question_selector_map[question]
+    else:
+        q = mmlu_question_selector_map[question]
 
     return q["agent_response"]["llama"][0], q["agent_response"]["wizardlm"][0], q["agent_response"]["orca"][0], q["agent_response"]["llama"][1], q["agent_response"]["wizardlm"][1], q["agent_response"]["orca"][1], q["agent_response"]["llama"][2], q["agent_response"]["wizardlm"][2], q["agent_response"]["orca"][2]
     
@@ -110,8 +129,9 @@ math_result, math_cot_result, gsm_result, gsm_cot_result = load_responses()
 math_questions, gsm_questions = load_questions(math_result, gsm_result)
 
 math_question_selector_map = build_question_selector_map(math_result)
-cot_math_question_selector_map = build_question_selector_map(cot_math_result)
+math_cot_question_selector_map = build_question_selector_map(math_cot_result)
 gsm_question_selector_map = build_question_selector_map(gsm_result)
+gsm_cot_question_selector_map = build_question_selector_map(gsm_cot_result)
 
 TITLE = """<h1 align="center">LLM Agora 🗣️🏦</h1>"""
 
@@ -237,14 +257,14 @@ with gr.Blocks() as demo:
             gr.HTML("""<h1 align="center"> The result of Math </h1>""")
             # gr.Image(value="result/Math/math_result.png")
 
-        math_cot.change(
-            display_question_answer,
-            [math_cot, math_question_list],
+        math_cot.select(
+            math_display_question_answer,
+            [math_question_list, math_cot],
             [math_model1_output1, math_model2_output1, math_model3_output1, math_model1_output2, math_model2_output2, math_model3_output2, math_model1_output3, math_model2_output3, math_model3_output3]
         )
         math_question_list.change(
-            display_question_answer,
-            [math_cot, math_question_list],
+            math_display_question_answer,
+            [math_question_list, math_cot],
             [math_model1_output1, math_model2_output1, math_model3_output1, math_model1_output2, math_model2_output2, math_model3_output2, math_model1_output3, math_model2_output3, math_model3_output3]
         )