Upload app.py

app.py

@@ -307,7 +307,8 @@ def outpaint_image(image, direction, extend_percent, custom_prompt, progress=gr.
 
     if image is None:
         raise gr.Error("Please upload an image first.")
-
+    
+    prompt = custom_prompt or ""
     inpaint.to("cuda")
     blip_model.to("cuda")
     # Move to GPU inside @spaces.GPU — ZeroGPU has allocated GPU here
@@ -321,11 +322,8 @@ def outpaint_image(image, direction, extend_percent, custom_prompt, progress=gr.
 
     progress(0.05, desc="Analyzing image with BLIP...")
 
-    if custom_prompt.strip():
-        blip_caption = custom_prompt.strip()
-    else:
-        blip_caption = get_caption(image)
-    prompt = build_outpaint_prompt(blip_caption)
+    blip_caption = custom_prompt.strip() if custom_prompt.strip() else get_caption(image)
+    prompt = build_outpaint_prompt(blip_caption, image)
 
     base_negative = (
         "blurry, bad quality, watermark, text, "