print prompt

app.py

@@ -453,25 +453,24 @@ def process_image(
     ).images[0]
     
     # Randomly choose whether to add furniture and which type
-    furniture_types = ["None", "sofa", "table", "chair"]
+    furniture_types = ["sofa", "table", "chair", "dresser", "bookshelf", "desk", "coffee table"]
     furniture_type = random.choice(furniture_types)
     
-    if furniture_type != "None":
-        furniture_mask = generate_furniture_mask(initial_result, furniture_type)
-        furniture_prompt = f"{prompt}, with a {furniture_type} in the style of {style_selection}"
-        
-        # Use the inpainting pipeline to add furniture
-        final_result = inpaint_pipe(
-            prompt=furniture_prompt,
-            image=initial_result,
-            mask_image=furniture_mask,
-            negative_prompt=negative_prompt,
-            num_inference_steps=num_steps,
-            guidance_scale=guidance_scale,
-            generator=generator,
-        ).images[0]
-    else:
-        final_result = initial_result
+    
+    furniture_mask = generate_furniture_mask(initial_result, furniture_type)
+    furniture_prompt = f"{prompt}, with a {furniture_type} in the style of {style_selection}"
+    print(furniture_prompt)
+    
+    # Use the inpainting pipeline to add furniture
+    final_result = inpaint_pipe(
+        prompt=furniture_prompt,
+        image=initial_result,
+        mask_image=furniture_mask,
+        negative_prompt=negative_prompt,
+        num_inference_steps=num_steps,
+        guidance_scale=guidance_scale,
+        generator=generator,
+    ).images[0]
     
     # Save and upload results
     timestamp = int(time.time())