Make clear reset the example

app.py

@@ -717,10 +717,6 @@ def solve_problem(problem, temperature, progress=gr.Progress()):
     yield sample["gen_texts"]
 
 
-def clear_inputs():
-    return "", 0.1, ""
-
-
 example_data = datasets.load_dataset(
     "AI-MO/kaggle-validation-set-medium-extended",
     split="train",
@@ -748,6 +744,11 @@ def update_problems():
     )
 
 
+def clear_inputs():
+    problem_1_display, problem_2_display, problem_1_full, problem_2_full = update_problems()
+    return "", 0.1, "", problem_1_display, problem_2_display, problem_1_full, problem_2_full
+
+
 css = """
 .top-margin { 
     margin-top: 20px; 
@@ -805,7 +806,6 @@ with gr.Blocks(css=css) as demo:
     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])
 
     problem_1_full_hidden = gr.Markdown(value=problem_1_full, visible=False)
     problem_2_full_hidden = gr.Markdown(value=problem_2_full, visible=False)
@@ -813,6 +813,20 @@ with gr.Blocks(css=css) as demo:
     copy_btn_1.click(fn=copy_problem_to_input, inputs=[problem_1_full_hidden], outputs=[inp])
     copy_btn_2.click(fn=copy_problem_to_input, inputs=[problem_2_full_hidden], outputs=[inp])
 
+    btn_clear.click(
+        fn=clear_inputs,
+        inputs=[],
+        outputs=[
+            inp,
+            temperature,
+            out,
+            problem_example_1,
+            problem_example_2,
+            problem_1_full_hidden,
+            problem_2_full_hidden,
+        ],
+    )
+
     demo.load(
         update_problems,
         inputs=None,