Update app.py

app.py

@@ -47,6 +47,7 @@ def parse_notebook(score_options, output_options, index):
         sample = ds_out[score_options][index]
     
     notebook_string = sample["text"]
+    notebook_response = sample["response"]
     notebook_id = sample["id"].split("/")[-1]
 
     out_path = os.path.join(TMP_DIR, notebook_id)
@@ -59,16 +60,18 @@ def parse_notebook(score_options, output_options, index):
     (notebook_body, resources) = html_exporter.from_notebook_node(notebook_parsed)
     notebook_body = embed_figures(notebook_body, resources)
 
-    return notebook_body, out_path
+    return notebook_body, out_path, notebook_response
 
 
 with gr.Blocks() as demo:
     gr.Markdown("# Kaggle Notebooks")
-    score_options = gr.Dropdown(["error","0", "1", "2", "3", "4", "5"], value="5", label="Notebook score", info="Select the assigned notebook score.")
-    output_options = gr.Radio(["Outputs only", "All"], value="Outputs only", label="Output filter", info="Many notebooks contain no outputs.")
-    index_slider = gr.Slider(minimum=0, maximum=len(ds_out["5"]), step=1, value=0, label="Index")
+    with gr.Row():
+        score_options = gr.Dropdown(["error","0", "1", "2", "3", "4", "5"], value="5", label="Notebook score", info="Select the assigned notebook score.")
+        output_options = gr.Radio(["Outputs only", "All"], value="Outputs only", label="Output filter", info="Many notebooks contain no outputs.")
+        index_slider = gr.Slider(minimum=0, maximum=len(ds_out["5"]), step=1, value=0, label="Index")
     
     file = gr.File()
+    response = gr.Markdown("", label="LLM score justification")
     html = gr.HTML("")
 
 
@@ -83,7 +86,7 @@ with gr.Blocks() as demo:
         outputs=[index_slider, index_slider]
     )
     
-    index_slider.change(fn=parse_notebook, inputs=[score_options, output_options, index_slider], outputs=[html, file])
-    demo.load(fn=parse_notebook, inputs=[score_options, output_options, index_slider], outputs=[html, file])
+    index_slider.change(fn=parse_notebook, inputs=[score_options, output_options, index_slider], outputs=[html, file, response])
+    demo.load(fn=parse_notebook, inputs=[score_options, output_options, index_slider], outputs=[html, file, response])
 
 demo.launch()