nice

app.py

@@ -441,6 +441,7 @@ def process_image(
         prompt = str(get_prompt(prompt, a_prompt))
     negative_prompt = str(n_prompt)
     
+    # Generate the initial room image
     initial_result = pipe(
         prompt=prompt,
         negative_prompt=negative_prompt,
@@ -457,17 +458,10 @@ def process_image(
     
     if furniture_type != "None":
         furniture_mask = generate_furniture_mask(initial_result, furniture_type)
-        furniture_prompt = f"A {furniture_type} in the style of {style_selection}"
+        furniture_prompt = f"{prompt}, with a {furniture_type} in the style of {style_selection}"
         
-        # Apply ControlNet to get the control image for inpainting
-        control_image_inpaint = preprocessor(
-            image=initial_result,
-            image_resolution=image_resolution,
-            detect_resolution=preprocess_resolution,
-        )
-        
-        # Use the regular inpainting pipeline with ControlNet guidance
-        inpainted_image = inpaint_pipe(
+        # Use the inpainting pipeline to add furniture
+        final_result = inpaint_pipe(
             prompt=furniture_prompt,
             image=initial_result,
             mask_image=furniture_mask,
@@ -476,21 +470,10 @@ def process_image(
             guidance_scale=guidance_scale,
             generator=generator,
         ).images[0]
-        
-        # Apply ControlNet guidance on the inpainted result
-        final_result = pipe(
-            prompt=furniture_prompt,
-            negative_prompt=negative_prompt,
-            guidance_scale=guidance_scale,
-            num_images_per_prompt=1,
-            num_inference_steps=num_steps,
-            generator=generator,
-            image=control_image_inpaint,
-            controlnet_conditioning_scale=0.5,  # Adjust this value as needed
-        ).images[0]
     else:
         final_result = initial_result
     
+    # Save and upload results
     timestamp = int(time.time())
     img_path = f"{timestamp}.jpg"
     results_path = f"{timestamp}_out.jpg"