update

app.py

@@ -26,6 +26,10 @@ def gradio_chat_completion(prompt, difficulty):
     # produce table
     df = pd.DataFrame(
         {
+            "Correct": [
+                predicted == target["response"]
+                for predicted, target in zip(predictions, targets)
+            ],
             "Input": [input_["message"] for input_ in inputs],
             "Prediction": predictions,
             "Target": [target["response"] for target in targets],
@@ -82,9 +86,10 @@ with gr.Blocks() as demo:
         )
         accuracy_box = gr.Textbox(label="Accuracy", interactive=False)
         results_table = gr.Dataframe(
-            headers=["Input", "Target", "Prediction"],
-            col_count=(3, "fixed"),
+            headers=["Correct", "Input", "Target", "Prediction"],
+            col_count=(4, "fixed"),
             interactive=False,
+            wrap=True,
         )
 
         btn.click(