add check

app.py

@@ -24,49 +24,93 @@ def get_crop(img, pos_x, pos_y, crop_size=0.1):
 @interactive(seed=(45, [0, 100], "Puzzle seed"))
 def generate_random_puzzle(seed: int = 45, context: dict = {}):
     np.random.seed(seed)
-    pos_x, pos_y, intensity = np.random.uniform(0.2, 0.8, 3)
-    context["puzzle"] = (pos_x, pos_y, intensity)
+    pos_x, pos_y = np.random.uniform(0.2, 0.8, 2)
+    context["puzzle_pos"] = (pos_x, pos_y)
     context["puzzle_flip_mirror"] = np.random.choice([True, False], 2)
 
 
-def create_puzzle(img, context: dict = {}):
+def create_puzzle(img, intensity: float = 0.4, context: dict = {}):
     out = img.copy()
-    x_gt, y_gt, intensity = context["puzzle"]
+    x_gt, y_gt = context["puzzle_pos"]
     flip_gt, mirror_gt = context["puzzle_flip_mirror"]
     cs_x, cs_y = get_crop(img, x_gt, y_gt)
     crop = img[cs_y[0]:cs_y[1], cs_x[0]:cs_x[1], ...]
-    out[cs_y[0]:cs_y[1], cs_x[0]:cs_x[1]] = intensity*0.4*crop
+    out[cs_y[0]:cs_y[1], cs_x[0]:cs_x[1]] = intensity*crop
     crop = flip_image(crop, flip=flip_gt, mirror=mirror_gt)
     return out, crop
 
 
 @interactive(
-    flip=(True, "Flip Image"),
-    mirror=(True, "Mirror Image"),
+    flip=(False, "Flip Image"),
+    mirror=(False, "Mirror Image"),
 )
-def flip_mirror_piece(piece: np.ndarray, flip=True, mirror=True):
+def flip_mirror_piece(
+    piece: np.ndarray,
+    flip: bool = False,
+    mirror: bool = False,
+    context: dict = {}
+) -> np.ndarray:
+    context["user_flip_mirror"] = (flip, mirror)
     return flip_image(piece.copy(), flip=flip, mirror=mirror)
 
 
 @interactive(
-    pos_x=(0.5, [0.1, 0.9], "Position X"),
-    pos_y=(0.5, [0.1, 0.9], "Position Y"),
+    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, piece, pos_x: float = 0.5, pos_y: float = 0.5):
+def place_puzzle(
+    puzzle: np.ndarray,
+    piece: np.ndarray,
+    pos_x: float = 0.5,
+    pos_y: float = 0.5,
+    context: dict = {}
+) -> np.ndarray:
     out = puzzle.copy()
+    context["user_pos"] = (pos_x, pos_y)
     cp_x, cp_y = get_crop(img, pos_x, pos_y)
     out[cp_y[0]:cp_y[1], cp_x[0]:cp_x[1]] = piece
     return out
 
 
+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 = {}):
+    x_gt, y_gt = context["puzzle_pos"]
+    flip_gt, mirror_gt = context["puzzle_flip_mirror"]
+    x, y = context["user_pos"]
+    flip, mirror = context["user_flip_mirror"]
+    check_pos = np.allclose([x_gt, y_gt], [x, y],
+                            atol=TOLERANCES.get(tolerance, 0.01))
+    check_flip_mirror = (flip_gt == flip) and (mirror_gt == mirror)
+    success = check_pos and check_flip_mirror
+    context["success"] = success
+
+
+def show_feedback(context: dict = {}):
+    success = context["success"]
+    flat_array = np.ones((256, 256, 3))
+    if success:
+        return flat_array*np.array([0., 1., 0.])[None, None, :]
+    else:
+        return flat_array*np.array([1., 0., 0.])[None, None, :]
+
 # pipeline definition
 # -------------------
+
+
 def captcha_pipe(inp):
     generate_random_puzzle()
     puzzle, puzzle_piece = create_puzzle(inp)
     puzzle_piece = flip_mirror_piece(puzzle_piece)
     puzzle = place_puzzle(puzzle, puzzle_piece)
-    return [puzzle_piece, puzzle]
+    check_puzzle()
+    validity = show_feedback()
+    return [puzzle_piece, puzzle, validity]
 
 
 if __name__ == "__main__":