Update

app.py

@@ -45,6 +45,7 @@ except Exception:
 total_issues = dataset.num_rows["test"]
 bug_id_to_time = dict()
 bug_id_to_type = dict()
+bug_id_to_patch = dict()
 bug_id_by_cat = {
     "crash": [],
     "miscompilation": [],
@@ -61,6 +62,7 @@ for issue in dataset["test"]:
     for comp in issue["hints"]["components"]:
         comp_bug_count[comp] = comp_bug_count.get(comp, 0) + 1
     bug_id_to_title[issue["bug_id"]] = "Issue " + issue["bug_id"] + ": " + issue["issue"]["title"]
+    bug_id_to_patch[issue["bug_id"]] = issue["patch"]
 timeline_xs = []
 timeline_ys = []
 timeline_cols = []
@@ -211,11 +213,7 @@ with demo:
             inspect_issue = gr.Dropdown(fixed_bug_title_id_pairs, label="Inspct Issue", interactive=True)
             golden_patch = gr.Code("", language="cpp", label="Golden Patch")
             inspect_issue.change(
-                fn=lambda bug_id: (
-                    dataset["test"][bug_id]["patch"]
-                    if bug_id in dataset["test"]
-                    else f"Not Available (bug_id = {bug_id})"
-                ),
+                fn=lambda bug_id: bug_id_to_patch.get(bug_id, f"Not Available (bug_id = {bug_id})"),
                 inputs=inspect_issue,
                 outputs=golden_patch,
             )