Update app.py

app.py

@@ -48,10 +48,10 @@ def process_image(image, color, design, design_position, text, text_position, fo
     if crop_coords:
         image = crop_image(image, crop_coords)
     return image
-def get_crop_coords(crop_points):
-    if crop_points:
-        x_coords = [point[0] for point in crop_points]
-        y_coords = [point[1] for point in crop_points]
+def get_crop_coords(crop_sketch):
+    if crop_sketch:
+        x_coords = [point[0] for point in crop_sketch]
+        y_coords = [point[1] for point in crop_sketch]
         return (min(x_coords), min(y_coords), max(x_coords), max(y_coords))
     return None
 
@@ -59,21 +59,21 @@ design_position = (100, 100)  # Default position for design
 text_position = (50, 50)      # Default position for text
 
 interface = gr.Interface(
-    fn=lambda image, color, design, design_position, text, text_position, font_size, font_color, resize, rotate, crop_points: process_image(
-        image, color, design, design_position, text, text_position, font_size, font_color, resize, rotate, get_crop_coords(crop_points)
+    fn=lambda image, color, design, design_position, text, text_position, font_size, font_color, resize, rotate, crop_sketch: process_image(
+        image, color, design, design_position, text, text_position, font_size, font_color, resize, rotate, get_crop_coords(crop_sketch)
     ),
     inputs=[
         gr.Image(type="pil"),
         gr.ColorPicker(label="Color"),
         gr.File(label="Design"),
-        gr.Image(tool="sketch", label="Design Position"),
+        gr.Sketchpad(label="Design Position", shape=(256, 256)),
         gr.Textbox(label="Text"),
-        gr.Image(tool="sketch", label="Text Position"),
+        gr.Sketchpad(label="Text Position", shape=(256, 256)),
         gr.Slider(10, 100, step=1, default=20, label="Font Size"),
         gr.ColorPicker(label="Font Color"),
         gr.Slider(50, 1000, step=1, default=None, label="Resize"),
         gr.Slider(0, 360, step=1, default=None, label="Rotate"),
-        gr.Image(tool="sketch", label="Crop Area")
+        gr.Sketchpad(label="Crop Area", shape=(256, 256))
     ],
     outputs=gr.Image(type="pil"),
     live=True