move decorators to main function

app.py

@@ -34,12 +34,8 @@ DIFFICULY = {"easy": 0.18, "medium": 0.1, "hard": 0.05}
 DIFFICULY_LEVELS = list(DIFFICULY.keys())
 
 
-@interactive(
-    seed=(45, [0, 100], "Puzzle seed"),
-    difficulty=(DIFFICULY_LEVELS[0], DIFFICULY_LEVELS, "Difficulty")
-)
 def generate_random_puzzle(
-    seed: int = 45,
+    seed: int = 43,
     difficulty: str = DIFFICULY_LEVELS[0],
     context: dict = {}
 ):
@@ -72,10 +68,6 @@ def create_puzzle(
     return out, crop
 
 
-@interactive(
-    flip=(False, "Flip Image"),
-    mirror=(False, "Mirror Image"),
-)
 def flip_mirror_piece(
     piece: np.ndarray,
     flip: bool = False,
@@ -87,10 +79,6 @@ def flip_mirror_piece(
     return flip_image(piece.copy(), flip=flip, mirror=mirror)
 
 
-@interactive(
-    pos_x=(0.5, [0.1, 0.9, 0.005], "Position X", ["left", "right"]),
-    pos_y=(0.5, [0.1, 0.9, 0.005], "Position Y", ["up", "down"]),
-)
 def place_puzzle(
     puzzle: np.ndarray,
     piece: np.ndarray,
@@ -111,11 +99,9 @@ TOLERANCES = {"low": 0.01, "medium": 0.02, "high": 0.05}
 TOLERANCE_LEVELS = list(TOLERANCES.keys())
 
 
-@interactive(
-    tolerance=(TOLERANCE_LEVELS[0], TOLERANCE_LEVELS, "Tolerance")
-)
 def check_puzzle(tolerance: str = "low", context: dict = {}) -> None:
-    """Check if the user placed the puzzle piece correctly."""
+    """Check if the user placed the puzzle piece correctly.
+    Store the result in the context."""
     x_gt, y_gt = context["puzzle_pos"]
     flip_gt, mirror_gt = context["puzzle_flip_mirror"]
     x, y = context["user_pos"]
@@ -133,7 +119,8 @@ def display_feedback(
     nok_ribbon: np.ndarray,
     context: dict = {}
 ) -> np.ndarray:
-    success = context["success"]
+    """Display green/red ribbon on the right side of the puzzle."""
+    success = context.get("success", False)
     ribbon = ok_ribbon if success else nok_ribbon
     out = np.hstack([puzzle, ribbon[:puzzle.shape[0], ...]])
     return out
@@ -152,19 +139,51 @@ def captcha_pipe(inp):
     puzzle = display_feedback(puzzle, ok_ribbon, nok_ribbon)
     return puzzle
 
+# add interactivity
+# -----------------
+
+
+def main(img: np.ndarray, backend="gradio", debug: bool = False):
+    # If debug mode, add interactive sliders to tune the puzzle generation
+    # and help the "game master" design a feasible puzzle.
+    if debug:
+        interactive(
+            tolerance=(TOLERANCE_LEVELS[0], TOLERANCE_LEVELS, "Tolerance")
+        )(check_puzzle)
+        interactive(
+            seed=(43, [0, 100], "Puzzle seed"),
+            difficulty=(DIFFICULY_LEVELS[0], DIFFICULY_LEVELS, "Difficulty")
+        )(generate_random_puzzle)
+    interactive(
+        pos_x=(0.5, [0.1, 0.9, 0.005], "Position X", ["left", "right"]),
+        pos_y=(0.5, [0.1, 0.9, 0.005], "Position Y", ["up", "down"]),
+    )(place_puzzle)
+    # left, right, up, down will only supported when using the Qt backend
+    interactive(
+        flip=(False, "Flip Image"),
+        mirror=(False, "Mirror Image"),
+    )(flip_mirror_piece)
+    captcha_pipe_interactive = interactive_pipeline(
+        gui=backend,
+        cache=True,
+        markdown_description=markdown_description
+    )(captcha_pipe)
+    captcha_pipe_interactive(img)
+
 
 if __name__ == "__main__":
     import argparse
     parser = argparse.ArgumentParser()
     parser.add_argument("-b", "--backend", default="gradio",
-                        choices=["gradio", "qt"], type=str)
+                        choices=["gradio", "qt", "mpl"], type=str)
+    parser.add_argument(
+        "-d", "--debug", action="store_true",
+        help="Debug mode (to tune difficulty and tolerance)"
+    )
     args = parser.parse_args()
-    markdown_description = "# Code to build this app on gradio \n"
+    markdown_description = "# Code to build this app on gradio \n\n"
+    markdown_description += "In local, try using `python app.py --backend qt --debug` to get the best experience with keyboard support aswell \n\n"
+    markdown_description += "Please note that matplobi`--backend mpl` is also functional although it won't look as good\n\n"
     markdown_description += "```python\n"+open(__file__, 'r').read()+"```"
     img = Image.load_image("sample.jpg")
-    captcha_pipe_interactive = interactive_pipeline(
-        gui=args.backend,
-        cache=True,
-        markdown_description=markdown_description
-    )(captcha_pipe)
-    captcha_pipe_interactive(img)
+    main(img, backend=args.backend, debug=args.debug)