add text prompting

app.py

@@ -1,4 +1,5 @@
-from interactive_pipe import interactive_pipeline, interactive, Image
+from interactive_pipe import interactive_pipeline, interactive, Image, TextPrompt
+import cv2
 from typing import Tuple
 import numpy as np
 
@@ -14,11 +15,45 @@ def processing_block(
     return amplify*inp
 
 
+def draw_text(img: np.ndarray, free_text: str = "Hello", preset_text: str = "choice") -> np.ndarray:
+    for txt, pos in [(free_text, (10, 50)), (preset_text, (10, 100))]:
+        out = cv2.putText(
+            img,
+            txt,
+            pos,
+            cv2.FONT_HERSHEY_SIMPLEX,
+            1,
+            (255, 255, 255), 2
+        )
+    return out
+
+
+def add_extra_caption(img: np.ndarray, caption: str = "Extra caption") -> np.ndarray:
+    out = cv2.putText(
+        img,
+        caption,
+        (10, 200),
+        cv2.FONT_HERSHEY_SIMPLEX,
+        1,
+        (255, 0, 255), 2
+    )
+    return out
+
+
+def move_circle(img: np.ndarray, x: int = 100, y: int = 100) -> np.ndarray:
+    out = img.copy()
+    cv2.circle(out, (x, y), 10, (255, 0, 0), -1)
+    return out
+
 # Pipeline definition
 # -------------------
 
+
 def pipe(inp: np.ndarray) -> Tuple[np.ndarray, np.ndarray]:
-    out = processing_block(inp)
+    out = add_extra_caption(inp)
+    out = processing_block(out)
+    out = draw_text(out)
+    out = move_circle(out)
     return inp, out
 
 # Add interactivity to the processing blocks
@@ -26,6 +61,24 @@ def pipe(inp: np.ndarray) -> Tuple[np.ndarray, np.ndarray]:
 
 
 def add_controls() -> None:
+
+    # 2 ways to add prompts:
+    # 1 = free_text=TextPrompt("DEFAULT TEXT", name="Caption custom")
+    # 2 = free_text=("DEFAULT TEXT", None, "Write Free text")
+    interactive(
+        caption=TextPrompt("Created with interactive pipe",
+                           name="Caption custom")
+    )(add_extra_caption)
+    interactive(
+        free_text=("Hello world!", None, "Write Free text"),
+        preset_text=("Hello", ["Hello", "World", "Goodbye"])
+    )(draw_text)
+    # Keyboards shortcuts: up, down, left, right -> only supported when using the Qt/MPL backend
+    # Replaced by sliders when using the Gradio backend
+    interactive(
+        x=(100, [0, 200], "X", ["left", "right"]),
+        y=(100, [0, 200], "Y", ["up", "down"])
+    )(move_circle)
     interactive(
         amplify=(0.5, [0., 1.], "Amplify")
     )(processing_block)

requirements.txt

@@ -1 +1 @@
-interactive-pipe>=0.8.3
+interactive-pipe>=0.8.4