add text as input

app.py

@@ -46,18 +46,20 @@ def move_circle(img: np.ndarray, x: int = 100, y: int = 100) -> np.ndarray:
     return out
 
 
-def remove_caps(txt_in: str = "TUTU") -> str:
-    return txt_in.lower()
+def remove_caps(txt_prefix: str, txt_in: str = "TUTU") -> str:
+    return txt_prefix + "\n" + txt_in.lower()
+
+
 # Pipeline definition
 # -------------------
 
 
-def pipe(inp: np.ndarray) -> Tuple[np.ndarray, np.ndarray]:
+def pipe(inp: np.ndarray, txt_prefix) -> Tuple[np.ndarray, np.ndarray]:
     out = add_extra_caption(inp)
     out = processing_block(out)
     out = draw_text(out)
     out = move_circle(out)
-    txt_out = remove_caps()
+    txt_out = remove_caps(txt_prefix)
     return inp, out, txt_out
 
 # Add interactivity to the processing blocks
@@ -102,7 +104,7 @@ def launch(img: np.ndarray, backend: str = "gradio", markdown_description: str =
         cache=True,
         markdown_description=markdown_description
     )(pipe)
-    pipe_interactive(img)
+    pipe_interactive(img, "REMOVE CAPS! ")
 
 
 if __name__ == "__main__":