initiate captcha example

README.md

@@ -1,6 +1,6 @@
 ---
-title: Interactive Pipe Captcha
-emoji: 📊
+title: Interactive Pipe Captcha Demo
+emoji: 🔒
 colorFrom: indigo
 colorTo: green
 sdk: gradio

app.py

@@ -0,0 +1,86 @@
+from interactive_pipe import interactive_pipeline, interactive, Image
+import numpy as np
+
+# Helper functions
+# ----------------
+
+
+def flip_image(img, flip=True, mirror=True):
+    img = img[::-1] if flip else img
+    img = img[:, ::-1] if mirror else img
+    return img
+
+
+def get_crop(img, pos_x, pos_y, crop_size=0.1):
+    c_size = int(crop_size * img.shape[0])
+    crop_x = (int(pos_x * img.shape[1]), int((pos_x) * img.shape[1]) + c_size)
+    crop_y = (int(pos_y * img.shape[0]), int((pos_y) * img.shape[0]) + c_size)
+    return crop_x, crop_y
+
+# Processing blocks
+# -----------------
+
+
+@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)
+    context["puzzle_flip_mirror"] = np.random.choice([True, False], 2)
+
+
+def create_puzzle(img, context: dict = {}):
+    out = img.copy()
+    x_gt, y_gt, intensity = context["puzzle"]
+    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
+    crop = flip_image(crop, flip=flip_gt, mirror=mirror_gt)
+    return out, crop
+
+
+@interactive(
+    flip=(True, "Flip Image"),
+    mirror=(True, "Mirror Image"),
+)
+def flip_mirror_piece(piece: np.ndarray, flip=True, mirror=True):
+    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"),
+)
+def place_puzzle(puzzle, piece, pos_x: float = 0.5, pos_y: float = 0.5):
+    out = puzzle.copy()
+    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
+
+
+# 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]
+
+
+if __name__ == "__main__":
+    import argparse
+    parser = argparse.ArgumentParser()
+    parser.add_argument("-b", "--backend", default="gradio",
+                        choices=["gradio", "qt"], type=str)
+    args = parser.parse_args()
+    markdown_description = "# Code to build this app on gradio \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)

requirements.txt

@@ -0,0 +1 @@
+interactive-pipe>=0.8.3