Fix not implemented

app.py

@@ -9,9 +9,10 @@ import base64
 import io
 from PIL import Image
 from functools import wraps
+from copy import deepcopy
 
 
-DEFAULT_TRANSFORM = "Rotate"
+DEFAULT_TRANSFORM = "CoarseDropout"
 
 DEFAULT_IMAGE = "images/doctor.webp"
 DEFAULT_IMAGE_HEIGHT = 400
@@ -58,22 +59,30 @@ for mask in BASE64_DEFAULT_MASKS:
 def run_with_retry(compose):
     @wraps(compose)
     def wrapper(*args, **kwargs):
-        for i in range(4):
+        processors = deepcopy(compose.processors)
+        for _ in range(4):
             try:
-                return compose(*args, **kwargs)
+                result = compose(*args, **kwargs)
+                break
             except NotImplementedError as e:
                 print(f"Caught NotImplementedError: {e}")
                 if "bbox" in str(e):
                     kwargs.pop("bboxes", None)
                     kwargs.pop("category_id", None)
+                    compose.processors.pop("bboxes")
                 if "keypoint" in str(e):
                     kwargs.pop("keypoints", None)
+                    compose.processors.pop("keypoints")
                 if "mask" in str(e):
                     kwargs.pop("mask", None)
+            except Exception as e:
+                compose.processors = processors
+                raise e
+        compose.processors = processors
+        return result
     return wrapper
 
 
-
 def draw_boxes(image, boxes, color=(255, 0, 0), thickness=2) -> np.ndarray:
     """Draw boxes with PIL."""
     pil_image = Image.fromarray(image)
@@ -113,7 +122,15 @@ def draw_not_implemented_image(image):
     """Draw the image with a text. In the middle."""
     pil_image = Image.fromarray(image)
     draw = ImageDraw.Draw(pil_image)
-    draw.text((DEFAULT_IMAGE_WIDTH // 2, DEFAULT_IMAGE_HEIGHT // 2), "Not implemented", fill=(255, 0, 0))
+    # align in the centerm, and make bigger font
+    text = "NOT IMPLEMETED FOR THIS TYPE OF ANNOTATIONS"
+    length = draw.textlength(text)
+    draw.text(
+        (DEFAULT_IMAGE_WIDTH // 2 - length // 2, DEFAULT_IMAGE_HEIGHT // 2),
+        text,
+        fill=(255, 0, 0),
+        align="center",
+    )
     return np.array(pil_image)
 
 
@@ -124,7 +141,9 @@ def get_formatted_signature(function_or_class, indentation=4):
 
     args = []
     for param in signature.parameters.values():
-        if param.default == inspect.Parameter.empty:
+        if param.name == "p":
+            str_param = "p=1.0,"
+        elif param.default == inspect.Parameter.empty:
             str_param = f"{param.name}=,"
         else:
             if isinstance(param.default, str):
@@ -208,28 +227,30 @@ def get_formatted_transform_docs(transform_number):
 with gr.Blocks() as demo:
 
     with gr.Row():
-        with gr.Group():
-            select = gr.Dropdown(
-                label="Select a transformation",
-                choices=transforms_keys,
-                value=DEFAULT_TRANSFORM,
-                type="index",
-                interactive=True,
-            )
-            with gr.Accordion("Documentation", open=False):
-                docs = gr.TextArea(
-                    get_formatted_transform_docs(
-                        transforms_keys.index(DEFAULT_TRANSFORM)
-                    ),
-                    show_label=False,
-                    interactive=False,
+        with gr.Column():
+            with gr.Group():
+                select = gr.Dropdown(
+                    label="Select a transformation",
+                    choices=transforms_keys,
+                    value=DEFAULT_TRANSFORM,
+                    type="index",
+                    interactive=True,
+                )
+                with gr.Accordion("Documentation", open=False):
+                    docs = gr.TextArea(
+                        get_formatted_transform_docs(
+                            transforms_keys.index(DEFAULT_TRANSFORM)
+                        ),
+                        show_label=False,
+                        interactive=False,
+                    )
+                code = gr.Code(
+                    language="python",
+                    value=get_formatted_transform(transforms_keys.index(DEFAULT_TRANSFORM)),
+                    interactive=True,
+                    lines=5,
                 )
-            code = gr.Code(
-                language="python",
-                value=get_formatted_transform(transforms_keys.index(DEFAULT_TRANSFORM)),
-                interactive=True,
-                lines=5,
-            )
+            button = gr.Button("Run")
             #info = gr.Text(interactive=False, label="Image info", value="")
         image = gr.Image(
             value=DEFAULT_IMAGE,
@@ -244,7 +265,6 @@ with gr.Blocks() as demo:
 
     #image.upload(fn=update_image_info, inputs=[image], outputs=[info])
     select.change(fn=get_formatted_transform, inputs=[select], outputs=[code])
-    button = gr.Button("Run")
     button.click(fn=update, inputs=[image, code], outputs=[augmented_image])