Update app.py

app.py

@@ -9,12 +9,13 @@ def apply_color(image, color):
 
 def add_design(image, design, position):
     if design:
+        design = Image.open(design).convert("RGBA")
         image.paste(design, position, design)
     return image
 
 def resize_image(image, size):
     if size:
-        return image.resize(size)
+        return image.resize((size, size))
     return image
 
 def rotate_image(image, angle):
@@ -34,7 +35,8 @@ def add_text(image, text, position, font_size, font_color):
         draw.text(position, text, fill=font_color, font=font)
     return image
 
-def process_image(image, color, design, design_position, text, text_position, font_size, font_color, resize, rotate, crop_coords):
+def process_image(image, color, design, design_position, text, text_position, font_size, font_color, resize, rotate, crop_sketch):
+    crop_coords = get_crop_coords(crop_sketch)
     if color:
         image = apply_color(image, color)
     if design:
@@ -48,6 +50,7 @@ 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_sketch):
     if crop_sketch:
         x_coords = [point[0] for point in crop_sketch]
@@ -60,7 +63,7 @@ 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_sketch: process_image(
-        image, color, design, design_position, text, text_position, font_size, font_color, resize, rotate, get_crop_coords(crop_sketch)
+        image, color, design, design_position, text, text_position, font_size, font_color, resize, rotate, crop_sketch
     ),
     inputs=[
         gr.Image(type="pil"),
@@ -69,14 +72,14 @@ interface = gr.Interface(
         gr.Sketchpad(label="Design Position"),
         gr.Textbox(label="Text"),
         gr.Sketchpad(label="Text Position"),
-        gr.Slider(10, 100, step=1, default=20, label="Font Size"),
+        gr.Slider(10, 100, step=1, value=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.Slider(50, 1000, step=1, value=100, label="Resize"),
+        gr.Slider(0, 360, step=1, value=0, label="Rotate"),
         gr.Sketchpad(label="Crop Area")
     ],
     outputs=gr.Image(type="pil"),
     live=True
 )
 
-interface.launch()
+interface.launch()