Update app.py

app.py

@@ -28,7 +28,7 @@ def add_polygon_vertex(x, y):
         if len(vertices) > 2 and np.linalg.norm(np.array(vertices[-1]) - np.array(vertices[0])) < 10:
             vertices[-1] = vertices[0]  # Close the polygon
             polygon_drawn = True
-    return vertices
+    return tshirt_img, design_img, design_pos
 
 def fit_design():
     global design_img, tshirt_img, vertices, polygon_drawn
@@ -55,13 +55,12 @@ def rotate_design(angle):
     return tshirt_img, design_img, design_pos
 
 def display_image(x=0, y=0, angle=0, image_type=None, image=None):
-    global vertices, design_pos
     if image_type == "tshirt":
         return load_tshirt(image)
     elif image_type == "design":
         return load_design(image)
     elif image_type == "polygon":
-        add_polygon_vertex(x, y)
+        return add_polygon_vertex(x, y)
     elif image_type == "fit":
         return fit_design()
     elif image_type == "rotate":
@@ -79,12 +78,12 @@ with gr.Blocks() as demo:
     design_output = gr.Image(label="Output Design on T-Shirt")
     
     # Upload options for T-shirt and design images
-    tshirt_image.upload(display_image, [x_coord, y_coord, rotate_slider, gr.State(value="tshirt"), tshirt_image], tshirt_output)
-    design_image.upload(display_image, [x_coord, y_coord, rotate_slider, gr.State(value="design"), design_image], [tshirt_output, design_output])
+    tshirt_image.upload(lambda image: display_image(image_type="tshirt", image=image), tshirt_image)
+    design_image.upload(lambda image: display_image(image_type="design", image=image), [tshirt_output, design_output])
 
     # Controls for adding polygon vertices, fitting the design, and rotating
-    gr.Button("Add Polygon Vertex").click(display_image, [x_coord, y_coord, rotate_slider, gr.State(value="polygon")], [tshirt_output, design_output])
-    gr.Button("Fit Design").click(display_image, [x_coord, y_coord, rotate_slider, gr.State(value="fit")], [tshirt_output, design_output])
-    rotate_slider.change(display_image, [x_coord, y_coord, rotate_slider, gr.State(value="rotate")], [tshirt_output, design_output])
+    gr.Button("Add Polygon Vertex").click(lambda x, y: display_image(x=x, y=y, image_type="polygon"), [x_coord, y_coord], [tshirt_output, design_output])
+    gr.Button("Fit Design").click(lambda: display_image(image_type="fit"), None, [tshirt_output, design_output])
+    rotate_slider.change(lambda angle: display_image(angle=angle, image_type="rotate"), rotate_slider, [tshirt_output, design_output])
 
 demo.launch(share=True)