add passed

app.py

@@ -3,19 +3,20 @@ import json
 import gradio as gr
 
 
-with open('problems.json') as f:
+with open("problems.json") as f:
     problems = json.load(f)
-    problem_choices = [q['question_title'] for q in problems]
+    problem_choices = [q["question_title"] for q in problems]
 
-with open('all_outputs.json') as f:
+with open("all_outputs.json") as f:
     all_outputs = json.load(f)
     model_choices = list(all_outputs.keys())
 
+
 def update_view(problem_index, model_name):
-    code = all_outputs[model_name][problem_index]['code_list'][0]
-    
+    code = all_outputs[model_name][problem_index]["code_list"][0]
+    correct = all_outputs[model_name][problem_index]["pass1_list"][0]
     code_viewer = gr.Code(
-        label=f"Code for Model {model_name} on {problems[problem_index]['question_title']}",
+        label=f"Code for Model {model_name} on {problems[problem_index]['question_title']} problem. Passed: {correct}",
         language="python",
         lines=10,
         value=code,
@@ -24,39 +25,34 @@ def update_view(problem_index, model_name):
     return code_viewer
 
 
-
 with gr.Blocks() as demo:
     ## dropdown
     problem_dropdown = gr.Dropdown(
         label="Selected Problem",
         choices=problem_choices,
         value=problem_choices[0],
-        type="index"
+        type="index",
     )
 
     model_dropdown = gr.Dropdown(
         label="Selected Model",
         choices=model_choices,
         value="GPT-4-Turbo-1106",
-        type="value"
+        type="value",
     )
 
-    code_viewer = (update_view(0, model_choices[0]))
+    code_viewer = update_view(0, model_choices[0])
 
     problem_dropdown.change(
-        update_view,
-        inputs=[problem_dropdown, model_dropdown],
-        outputs=[code_viewer]
+        update_view, inputs=[problem_dropdown, model_dropdown], outputs=[code_viewer]
     )
 
     model_dropdown.change(
-        update_view,
-        inputs=[problem_dropdown, model_dropdown],
-        outputs=[code_viewer]
+        update_view, inputs=[problem_dropdown, model_dropdown], outputs=[code_viewer]
     )
 
     ## code viewer
-    
+
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()