Update app.py

app.py

@@ -37,12 +37,12 @@ def generate_displacement_map(image_a):
     return displacement_map
 
 # Function to warp and fit Image-B onto Image-A
-def fit_and_warp_design(image_a, image_b, design_bbox):
+def fit_and_warp_design(image_a, image_b, top_left_x, top_left_y, bottom_right_x, bottom_right_y):
     displacement_map = generate_displacement_map(image_a)
 
     # Extract bounding box coordinates
-    top_left = (int(design_bbox[0]), int(design_bbox[1]))
-    bottom_right = (int(design_bbox[2]), int(design_bbox[3]))
+    top_left = (int(top_left_x), int(top_left_y))
+    bottom_right = (int(bottom_right_x), int(bottom_right_y))
 
     # Resize the design to fit within the specified bounding box
     design_width = bottom_right[0] - top_left[0]
@@ -74,22 +74,25 @@ def fit_and_warp_design(image_a, image_b, design_bbox):
     return final_image
 
 # Gradio interface function
-def process_images(image_a, image_b, design_bbox):
-    result = fit_and_warp_design(image_a, image_b, design_bbox)
+def process_images(image_a, image_b, top_left_x, top_left_y, bottom_right_x, bottom_right_y):
+    result = fit_and_warp_design(image_a, image_b, top_left_x, top_left_y, bottom_right_x, bottom_right_y)
     return result
 
 # Gradio UI components
 image_input_a = gr.Image(label="Upload Clothing Image", type="pil")
 image_input_b = gr.Image(label="Upload Design Image", type="pil")
-design_bbox_input = gr.Image(label="Adjust Design Position and Size", type="pil")
+top_left_x = gr.Slider(minimum=0, maximum=1000, label="Top-left X Coordinate", default=50)
+top_left_y = gr.Slider(minimum=0, maximum=1000, label="Top-left Y Coordinate", default=100)
+bottom_right_x = gr.Slider(minimum=0, maximum=1000, label="Bottom-right X Coordinate", default=300)
+bottom_right_y = gr.Slider(minimum=0, maximum=1000, label="Bottom-right Y Coordinate", default=400)
 
 # Define the Gradio interface
 iface = gr.Interface(
     fn=process_images,
-    inputs=[image_input_a, image_input_b, design_bbox_input],
+    inputs=[image_input_a, image_input_b, top_left_x, top_left_y, bottom_right_x, bottom_right_y],
     outputs="image",
-    title="Clothing Design Fitting with Drag-and-Drop",
-    description="Upload a clothing image and a design image. Drag and resize the design onto the clothing using the cursor.",
+    title="Clothing Design Fitting with Adjustable Bounding Box",
+    description="Upload a clothing image and a design image. Adjust the design's position and size using sliders to fit the design onto the clothing.",
 )
 
 # Launch the interface