Upload app.py

app.py

@@ -693,52 +693,18 @@ def _rows_to_table(rows: list[dict]) -> list[list[str]]:
         ])
     return table
 
-def _dropdown_choices(rows: list[dict]) -> list[tuple[str, int]]:
-    # Friendly labels mapped to ID values
-    choices = []
-    for r in rows:
-        label = f'#{r["id"]} — {r["label"]} — {_truncate(r["question"], 60)}'
-        choices.append((label, r["id"]))
-    return choices
 
 # ===== Gradio callbacks for examples =====
-def ui_see_more(selector, rows, filter_label):
-    """Append a chunk of examples to the browser."""
-    chunk = selector.next_batch(k=6, filter_label=filter_label)
-    rows = (rows or []) + chunk
-    return (
-        rows,                                            # rows_state
-        gr.update(value=_rows_to_table(rows)),           # examples_df
-        gr.update(choices=_dropdown_choices(rows), value=None),  # row_picker
-    )
-
-def ui_reset_examples():
-    """Reset per-session selector and clear the browser."""
-    sel = new_selector()
-    rows: list[dict] = []
-    return (
-        sel,                                            # selector_state
-        rows,                                           # rows_state
-        gr.update(value=_rows_to_table(rows)),          # examples_df
-        gr.update(choices=[], value=None),              # row_picker
-    )
 
-def ui_load_selected(rows, selected_id):
-    """Load the selected example into the main inputs."""
-    if not rows or selected_id is None:
-        return gr.update(), gr.update()
-    for r in rows:
-        if r["id"] == selected_id:
-            return r["question"], r["solution"]
-    return gr.update(), gr.update()
-
-def ui_surprise(selector, filter_label):
-    """Pick one example and push it straight to inputs."""
+# ---- replace ui_surprise with this ----
+def ui_surprise(selector, filter_label="any"):
+    """Pick one example and push it straight to inputs; persist selector state."""
+    if selector is None or not POOL:
+        return selector, gr.update(), gr.update()
     r = selector.surprise(filter_label=filter_label)
     if not r:
-        # no more examples; keep inputs unchanged
-        return gr.update(), gr.update()
-    return r["question"], r["solution"]
+        return selector, gr.update(), gr.update()
+    return selector, r["question"], r["solution"]
 
 
 # ---------------- UI: add CSV-driven examples ----------------
@@ -751,7 +717,6 @@ with gr.Blocks(title="Math Solution Classifier", theme=gr.themes.Soft()) as app:
 
     # Per-session state
     selector_state = gr.State(new_selector())
-    rows_state     = gr.State([])  # list[dict] rows currently in the browser
 
     with gr.Row():
         # -------- Left: inputs --------
@@ -780,46 +745,18 @@ with gr.Blocks(title="Math Solution Classifier", theme=gr.themes.Soft()) as app:
     # -------- Curated starter examples (static) --------
     gr.Examples(
         examples=[
-            ["Solve for x: 2x + 5 = 13",
-             "2x + 5 = 13\n2x = 13 - 5\n2x = 8\nx = 4"],
-            ["John has three apples and Mary has seven, how many apples do they have together?",
+        ["John has three apples and Mary has seven, how many apples do they have together?",
              "They have 7 + 3 = 11 apples."],
+        ["A tank holds 60 liters of fuel. A generator uses fuel at a rate of 5 liters per hour. After running for 9 hours, how many liters are still in the tank?",
+"The generator uses 5 L/h × 9 h = 45 L of fuel in 9 hours.\n Then, there remain 60 L + 45 L = 105 L in the tank.\n Final answer: 105 L"],
             ["What is 15% of 200?",
              "15% = 15/100 = 0.15\n0.15 × 200 = 30"],
+             ["A 24-meter rope is cut into 6 equal pieces. A climber uses 2 of those pieces. How many meters of rope are still unused?",
+             "The length of each piece is 24 / 6 = 4 m.\n The climber uses 2 × 4 m = 8 m of rope.\n There are 24 m − 8 m = 16 m of rope still unused."]
         ],
         inputs=[question_input, solution_input],
     )
-
-    # -------- Dynamic browser (CSV) --------
-    with gr.Accordion("Browse more examples", open=False):
-        with gr.Row():
-            filter_dd = gr.Dropdown(
-                label="Filter",
-                choices=[
-                    ("Any", "any"),
-                    ("Correct only", "correct"),
-                    ("Conceptual error only", "conceptual_error"),
-                    ("Computational error only", "computational_error"),
-                ],
-                value="any",
-                allow_custom_value=False,
-            )
-            see_more_btn   = gr.Button("See more")
-            reset_list_btn = gr.Button("Reset list")
-
-        examples_df = gr.Dataframe(
-            headers=["ID", "Label", "Question", "Solution"],
-            value=[],
-            interactive=False,
-            row_count=(0, "dynamic"),
-            col_count=4,
-            wrap=True,
-            height=260,
-            label="Examples",
-        )
-        with gr.Row():
-            row_picker = gr.Dropdown(label="Select example to load", choices=[], value=None, scale=2)
-            load_btn   = gr.Button("Load to editor", scale=1)
+    
 
     # ---------- Wiring ----------
     # Main classify (streaming)
@@ -831,37 +768,14 @@ with gr.Blocks(title="Math Solution Classifier", theme=gr.themes.Soft()) as app:
         concurrency_limit=1,
     )
 
-    # Surprise me → fills inputs from the CSV pool
-    surprise_btn.click(
-        fn=ui_surprise,
-        inputs=[selector_state, filter_dd],
-        outputs=[question_input, solution_input],
-        queue=True,
-    )
-
-    # See more → appends rows to the browser
-    see_more_btn.click(
-        fn=ui_see_more,
-        inputs=[selector_state, rows_state, filter_dd],
-        outputs=[rows_state, examples_df, row_picker],
-        queue=False,
-    )
-
-    # Reset list → new selector + clear table
-    reset_list_btn.click(
-        fn=ui_reset_examples,
-        inputs=None,
-        outputs=[selector_state, rows_state, examples_df, row_picker],
-        queue=False,
-    )
+# ---- and replace the Surprise button wiring with this ----
+surprise_btn.click(
+    fn=ui_surprise,
+    inputs=[selector_state],                              # no filter_dd anymore
+    outputs=[selector_state, question_input, solution_input],  # persist selector state
+    queue=True,
+)
 
-    # Load selected row → fills main inputs
-    load_btn.click(
-        fn=ui_load_selected,
-        inputs=[rows_state, row_picker],
-        outputs=[question_input, solution_input],
-        queue=False,
-    )
 
 # Enable queue for streaming
 app.queue()