Reset to recommended settings button

app.py

@@ -180,6 +180,22 @@ def on_clear(state, gamma):
             "Focal points cleared. Click the restored image or the depth map to focus.")
 
 
+RECOMMENDED = {"long_side": 1024, "omega": 1.25, "gamma": 1.0, "band": 0.06, "strength": 0.5, "mode": SINGLE,
+               "show_marker": False}
+
+
+def on_reset(state):
+    """Recommended settings (the paper's defaults), the estimated depth again; focal points are kept."""
+    r = RECOMMENDED
+    controls = [r["long_side"], r["omega"], r["gamma"], r["band"], r["strength"], r["mode"], r["show_marker"], None]
+    if not state:
+        return controls + [gr.update(), None, gr.update(), state, "Settings reset to the recommended values."]
+    state = dict(state, user_depth=None, pts=state["pts"][:1])
+    out, depth_view, msg = render(state, r["strength"], r["mode"], r["gamma"], r["band"], r["show_marker"])
+    note = "Settings reset to the recommended values (ω applies on the next Restore). "
+    return controls + [aif_view(state, r["mode"]), out, depth_view, state, note + msg]
+
+
 HEADER = """
 <div style="text-align:center">
 <h1 style="margin-bottom:0.2em">FlashReFocus: Interactive Image Refocusing in Seconds</h1>
@@ -226,6 +242,7 @@ with gr.Blocks(title="FlashReFocus", theme=gr.themes.Soft()) as demo:
             with gr.Row():
                 show_marker = gr.Checkbox(value=False, label="Show focal markers on the result")
                 clear = gr.Button("Clear focal points")
+                reset = gr.Button("Reset to recommended settings")
             out = gr.Image(label="3. Refocused", type="pil", interactive=False, height=420, format="png")
     status = gr.Markdown("")
     controls = [state, strength, mode, gamma, band, show_marker]
@@ -239,6 +256,7 @@ with gr.Blocks(title="FlashReFocus", theme=gr.themes.Soft()) as demo:
     show_marker.change(on_change, controls, outputs)
     user_depth.change(on_user_depth, [state, user_depth, strength, mode, gamma, band, show_marker], outputs)
     clear.click(on_clear, [state, gamma], outputs)
+    reset.click(on_reset, [state], [long_side, omega, gamma, band, strength, mode, show_marker, user_depth] + outputs)
 
 if __name__ == "__main__":
     demo.queue(max_size=20).launch(show_api=False)