Update app.py

app.py

@@ -34,8 +34,8 @@ def sort_centroids_clockwise(centroids: np.ndarray):
 
 def infer(image_input, dst_height: str, dst_width: str):
     if isinstance(image_input, dict):
-        image_in = image_input["image"]
-        mask_in = image_input["mask"]
+        image_in = np.array(image_input['composite'])
+        mask_in = np.array(image_input['layers'][0]) if image_input['layers'] else np.zeros_like(image_in)
     else:
         image_in = image_input
         mask_in = np.zeros_like(image_in)
@@ -77,7 +77,7 @@ def infer(image_input, dst_height: str, dst_width: str):
 
 description = """In this space you can warp an image using perspective transform with the Kornia library as seen in [this tutorial](https://kornia.github.io/tutorials/#category=Homography).
 1. Upload an image or use the example provided
-2. Set 4 points into the image with your cursor, which define the area to warp
+2. Set 4 points into the image using the brush tool, which define the area to warp
 3. Set a desired output size (or go with the default)
 4. Click Submit to run the demo
 """
@@ -86,19 +86,38 @@ description = """In this space you can warp an image using perspective transform
 example_image = Image.open("bruce.png")
 example_image_np = np.array(example_image)
 
-inputs = [
-    gr.Image(type="numpy", label="Input Image", tool="sketch"),
-    gr.Textbox(label="Destination Height", value="64"),
-    gr.Textbox(label="Destination Width", value="128"),
-]
-outputs = gr.Plot(label="Output")
+with gr.Blocks() as demo:
+    gr.Markdown("# Homography Warping")
+    gr.Markdown(description)
+    
+    with gr.Row():
+        image_input = gr.ImageEditor(
+            type="numpy",
+            label="Input Image",
+            brush=gr.Brush(colors=["#ff0000"], size=5),
+            height=400,
+            width=600
+        )
+        output_plot = gr.Plot(label="Output")
+    
+    with gr.Row():
+        dst_height = gr.Textbox(label="Destination Height", value="64")
+        dst_width = gr.Textbox(label="Destination Width", value="128")
+    
+    submit_button = gr.Button("Submit")
+    submit_button.click(
+        fn=infer,
+        inputs=[image_input, dst_height, dst_width],
+        outputs=output_plot
+    )
+    
+    gr.Examples(
+        examples=[[example_image_np, "64", "128"]],
+        inputs=[image_input, dst_height, dst_width],
+        outputs=output_plot,
+        fn=infer,
+        cache_examples=True
+    )
 
-gr.Interface(
-    fn=infer,
-    inputs=inputs,
-    outputs=outputs,
-    title="Homography Warping",
-    description=description,
-    examples=[[example_image_np, "64", "128"]],
-    cache_examples=True
-).launch()
+if __name__ == "__main__":
+    demo.launch()