Update app.py

app.py

@@ -12,12 +12,8 @@ print(f"Running on device: {device}")
 
 # Load models
 segmenter = pipeline("mask-generation", model="facebook/sam-vit-huge")
-upscaler = StableDiffusionUpscalePipeline.from_pretrained(
-    "stabilityai/stable-diffusion-x4-upscaler"  # Bỏ torch_dtype=torch.float16
-).to(device)
-inpaint = StableDiffusionInpaintPipeline.from_pretrained(
-    "runwayml/stable-diffusion-inpainting"  # Bỏ torch_dtype=torch.float16
-).to(device)
+upscaler = StableDiffusionUpscalePipeline.from_pretrained("stabilityai/stable-diffusion-x4-upscaler").to(device)
+inpaint = StableDiffusionInpaintPipeline.from_pretrained("runwayml/stable-diffusion-inpainting").to(device)
 
 # Helper functions
 def upscale_image(image):
@@ -56,14 +52,14 @@ def process_image(base_image, paste_images_input, points_input, objects_input):
     points = [list(map(int, p.split(","))) for p in points_input.split(";")]
     objects = objects_input.split(";")
 
-    if len(points) != len(objects) or len(points) > len(paste_images):
-        raise ValueError("Number of points, objects, and paste images must match!")
+    if len(points) != len(objects) or len(points) != len(paste_images):
+        raise ValueError(f"Number of points ({len(points)}), objects ({len(objects)}), and paste images ({len(paste_images)}) must match!")
 
     upscaled_base = upscale_image(base_image)
     masks = segment_image(upscaled_base, points)
     refined_masks = [refine_mask(mask) for mask in masks]
     inverted_masks = [invert_mask(mask) for mask in refined_masks]
-    seamless_pastes = [make_seamless(img) for img in paste_images[:len(points)]]
+    seamless_pastes = [make_seamless(img) for img in paste_images]
     pasted_image = paste_by_mask(upscaled_base, seamless_pastes, refined_masks)
 
     combined_prompt = ", ".join([f"modern {obj}" for obj in objects]) + " in high quality interior, 2025 trends"
@@ -83,16 +79,24 @@ interface = gr.Interface(
     fn=process_image,
     inputs=[
         gr.Image(type="pil", label="Base Image (Interior)"),
-        gr.File(file_count="multiple", label="Paste Images (New designs)"),
-        gr.Textbox(label="Points (x,y; separated by ';')", value="500,500;600,600"),  # Thay default bằng value
-        gr.Textbox(label="Objects (separated by ';')", value="counter;chair")  # Thay default bằng value
+        gr.File(file_count="multiple", label="Paste Images (One per object)"),
+        gr.Textbox(
+            label="Points (x,y; separated by ';')",
+            value="500,500;600,600",
+            placeholder="e.g., '500,500;600,600' (one point per object)"
+        ),
+        gr.Textbox(
+            label="Objects (separated by ';')",
+            value="counter;chair",
+            placeholder="e.g., 'counter;chair' (one object per image)"
+        )
     ],
     outputs=[
         gr.Image(label="Intermediate Image (After Pasting)"),
         gr.Image(label="Final Enhanced Interior")
     ],
     title="Interior Design Enhancer",
-    description="Upload a base image and new designs. Specify points and objects to replace (e.g., '500,500;600,600' and 'counter;chair')."
+    description="Upload a base image and new designs (one image per object). Specify points (x,y) and object names, separated by ';'. Ensure the number of points, objects, and images match (e.g., 2 points, 2 objects, 2 images)."
 )
 
 interface.launch(share=True)