Add clear and copy examples problem feature

app.py

@@ -8,6 +8,8 @@ import re
 import subprocess
 import tempfile
 import json
+import datasets
+import random
 from typing import Tuple, Dict, Any, List
 from sympy import N, simplify
 from sympy.parsing.latex import parse_latex
@@ -541,7 +543,7 @@ def generate(message, temperature):
         max_tokens=1024,
         stop=["```output\n"],
         temperature=temperature,
-        timeout=30
+        timeout=30,
     )
 
     response = stream.response
@@ -715,17 +717,81 @@ def solve_problem(problem, temperature, progress=gr.Progress()):
     yield sample["gen_texts"]
 
 
-with gr.Blocks(css=".top-margin { margin-top: 20px; }") as demo:
+def clear_inputs():
+    return "", 0.1, ""
+
+
+example_data = datasets.load_dataset("AI-MO/kaggle-validation-set-medium-extended", split="train")
+
+
+def get_random_problems():
+    examples = random.sample(list(example_data), 2)
+    problems = [ex["problem"] for ex in examples]
+    return problems[0], problems[1]
+
+
+def copy_problem_to_input(problem):
+    return problem
+
+
+css = """
+.top-margin { 
+    margin-top: 20px; 
+}
+
+.problem-container {
+    display: flex;
+    align-items: center;
+    margin-bottom: 10px;
+    border-radius: 8px;
+    border: 1px solid rgb(229, 231, 235);
+    padding: 10px 12px;
+}
+
+# .problem-text {
+#     flex: 1;
+#     padding: 10px;
+#     background-color: #f9f9f9;
+#     border: 1px solid #ccc;
+#     border-radius: 5px;
+#     margin-right: 10px;
+# }
+
+.copy-button {
+    flex: none;
+}
+"""
+
+with gr.Blocks(css=css) as demo:
+    latex_delimiters = [
+        {"left": "[", "right": "]", "display": True},
+    ]
+
+    problem_1_text, problem_2_text = get_random_problems()
+
+    with gr.Row(elem_classes="problem-container"):
+        problem_1_text_contract = problem_1_text[:120] + "..." if len(problem_1_text) > 100 else problem_1_text
+        problem_1 = gr.Markdown(value=problem_1_text_contract, latex_delimiters=latex_delimiters)
+        copy_btn_1 = gr.Button("Copy", elem_classes="copy-button")
+    with gr.Row(elem_classes="problem-container"):
+        problem_2_text_contract = problem_2_text[:120] + "..." if len(problem_2_text) > 100 else problem_2_text
+        problem_2 = gr.Markdown(value=problem_2_text_contract, latex_delimiters=latex_delimiters)
+        copy_btn_2 = gr.Button("Copy", elem_classes="copy-button")
+
     with gr.Row():
         inp = gr.Textbox(placeholder="Problem", label="Problem", lines=5)
     with gr.Accordion("Advanced Options", open=False):
         temperature = gr.Slider(minimum=0.0, maximum=1.0, value=0.1, step=0.1, label="Temperature")
     with gr.Row():
-        out = gr.Markdown(elem_classes=["top-margin"], latex_delimiters=[{"left": "[", "right": "]", "display": True}])
+        out = gr.Markdown(elem_classes=["top-margin"], latex_delimiters=latex_delimiters)
 
     btn = gr.Button("Run")
+    btn_clear = gr.Button("Clear")
 
     btn.click(fn=solve_problem, inputs=[inp, temperature], outputs=out)
+    btn_clear.click(fn=clear_inputs, inputs=[], outputs=[inp, temperature, out])
+    copy_btn_1.click(fn=copy_problem_to_input, inputs=[gr.Markdown(value=problem_1_text, visible=False)], outputs=[inp])
+    copy_btn_2.click(fn=copy_problem_to_input, inputs=[gr.Markdown(value=problem_2_text, visible=False)], outputs=[inp])
 
 
 if __name__ == "__main__":