Update src/core.py

src/core.py

@@ -22,56 +22,16 @@ log = logging.getLogger(__name__)
 DEFAULT_MODEL_ID = os.environ.get("GEMINI_IMAGE_MODEL", "gemini-3-pro-image-preview")
 DEFAULT_PROMPT = os.environ.get(
     "GEMINI_IMAGE_PROMPT",
-    ("TASK: STRICT IMAGE INPAINTING — MASK-BOUNDED OBJECT REMOVAL ONLY\n\n"
-    "You are given:\n"
-    "1) **Image A**: The original image, which contains various elements and objects.\n"
-    "2) **Image B**: A binary mask that highlights the areas to be removed from the original image.\n\n"
-    
-    "MASK INTERPRETATION (CRITICAL):\n\n"
-    "• **White pixels (#FFFFFF)** in Image B represent **the exact regions** in Image A that need to be removed. Only these regions can be modified or deleted.\n"
-    "• **Black pixels (#000000)** in Image B must **remain completely unchanged** in Image A. Do **NOT** modify or generate anything in these black areas.\n"
-    "• **Do NOT** modify or alter any other parts of Image A that are outside the white-masked region. Do not paint, blur, or alter any pixels that are not covered by the white mask.\n"
-    "• **Do NOT** attempt to remove or alter anything in the unmasked regions, even if the object in the unmasked area looks similar to the object inside the mask.\n"
-    "• **The white-masked region is the **only** area to be modified. The rest of the image must **remain intact**. No other object or area should be painted or modified.\n"
-    "\n"
-    
-    "STEP 1: OBJECT REMOVAL ANALYSIS\n\n"
-    "• **Analyze the mask carefully** and **identify the exact areas** to be removed. Only focus on the white-masked region for removal.\n"
-    "• The model must recognize what needs to be removed based on the white mask and **do nothing to the black (unmasked) regions**.\n"
-    "• Do **NOT** modify, erase, or paint anything outside the white-masked region. Only remove what is explicitly covered by the white mask.\n\n"
-    
-    "STEP 2: OBJECT REMOVAL\n\n"
-    "• **Completely remove** everything inside the white-masked region in Image A. This includes:\n"
-    "    - Erasing the object(s) entirely with **no visible remnants**, including shadows, outlines, or color traces.\n"
-    "    - If the mask overlaps only part of an object, **only remove the pixels covered by the white mask**, leaving the rest of the object untouched.\n"
-    "• The goal is to **clear the masked region** entirely, **removing all visual content** from it.\n\n"
-    
-    "IMPORTANT NOTE ABOUT INPAINTING:\n\n"
-    "• **The inpainted region must NOT be filled with pure white (#FFFFFF)**. The model should **never** generate a blank, white, or featureless region.\n"
-    "• The white areas should be filled with **the natural surrounding background** of Image A. This means extending **existing textures, colors, lighting, and geometry naturally**, without introducing any artificial white filling.\n"
-    "• **Do not paint or alter any other region** in the image, including the unmasked areas. **Only the white-masked area** should be modified, and no other content should be painted or changed.\n\n"
-    
-    "INPAINTING RULES:\n\n"
-    "• After the object is removed, **fill the masked area** using **only the immediate surrounding pixels** from Image A.\n"
-    "• **Do not invent new objects, structures, or details**. Only extend or copy background textures, lighting, and geometry from the unmasked areas.\n"
-    "• The lighting, contrast, brightness, color temperature, and texture in the inpainted region must **perfectly match** the surrounding areas.\n"
-    "• **Do not add** shadows, reflections, or new structures to the inpainted region.\n"
-    "• The inpainted region should seamlessly blend with the surrounding pixels. It should appear as though the object was **never there**.\n\n"
-    
-    "STRICT CONSTRAINTS (NO EXCEPTIONS):\n\n"
-    "• **Do NOT modify any black (unmasked) pixels**. The unmasked areas must remain **exactly** as they are in Image A.\n"
-    "• **Do NOT remove entire objects** unless they are **fully and completely masked**. Only the regions inside the white mask should be modified.\n"
-    "• **Do NOT alter the original image's **framing**, **perspective**, or **composition**. Only the regions inside the white mask should be affected.\n"
-    "• **Do NOT apply any semantic cleanup** or scene optimization. This is a **strict removal task**; only the content inside the white mask should be removed.\n"
-    
-    "QUALITY REQUIREMENTS:\n\n"
-    "• The final result should look as though the object **never existed** in the original image.\n"
-    "• The inpainted region must blend **seamlessly** with surrounding pixels.\n"
-    "• **No visible artifacts**, such as ghosting, halos, color shifts, or texture repetition, should be present.\n"
-    "• The transition between the inpainted region and the unmasked areas must be **natural** and **imperceptible**.\n\n"
-    
-    "IMPORTANT:\n\n"
-    "If there is a conflict between **semantic understanding** and the **mask**, **ALWAYS follow the mask exactly**. The mask is the **absolute guide**, and no interpretation should deviate from the mask’s instructions.\n"
+    ("Remove ONLY the white areas shown in the mask. Keep everything else EXACTLY as it is.\n\n"
+    "CRITICAL RULES:\n"
+    "1. White pixels in mask = REMOVE (inpaint with background)\n"
+    "2. Black pixels in mask = KEEP UNCHANGED\n"
+    "3. Do NOT remove similar objects outside the mask\n"
+    "4. Do NOT recreate or add new objects\n"
+    "5. Only modify pixels where mask is white\n"
+    "6. Fill removed areas with surrounding background texture\n"
+    "7. Match lighting, colors, and texture perfectly\n\n"
+    "Follow the mask EXACTLY. Do not interpret semantically."
     ),
 )
 _GENAI_MODEL: genai.GenerativeModel | None = None