Update src/core.py

src/core.py

@@ -23,58 +23,40 @@ DEFAULT_MODEL_ID = os.environ.get("GEMINI_IMAGE_MODEL", "gemini-3-pro-image-prev
 DEFAULT_PROMPT = os.environ.get(
     "GEMINI_IMAGE_PROMPT",
     (
-        "TASK TYPE: STRICT IMAGE INPAINTING — OBJECT REMOVAL ONLY\n\n"
-    "You are provided with two images:\n\n"
-    "1) The original image (Image A) which contains the scene with various objects.\n"
-    "2) The binary mask image (Image B) that highlights the region(s) to be removed in the original image.\n\n"
+        "TASK: STRICT IMAGE INPAINTING — MASK-BOUNDED OBJECT REMOVAL ONLY\n\n"
+    "You are given:\n"
+    "• Image A: the original image.\n"
+    "• Image B: a binary mask.\n\n"
     
-    "MASK RULE (MANDATORY):\n\n"
-    "• In the mask image (Image B), white pixels (#FFFFFF) represent the exact regions in the original image (Image A) "
-    "that need to be completely removed.\n"
-    "• Black pixels (#000000) in the mask image indicate areas of the original image that must remain completely unchanged "
-    "and should not be altered in any way.\n"
+    "MASK INTERPRETATION (CRITICAL):\n\n"
+    "• **Pure white (#FFFFFF)** pixels in Image B mark the ONLY pixels that may be modified.\n"
+    "• **Pure black (#000000)** pixels in Image B are **LOCKED** and **MUST remain IDENTICAL** to Image A.\n"
+    "• **Do not** extend, expand, infer, or reinterpret the mask in any way.\n\n"
     
-    "PRIMARY OBJECTIVE:\n\n"
-    "The goal is to **remove** all the objects or parts of objects that are highlighted in the white-masked regions of the mask image. "
-    "The object in the white region must be **entirely erased** with no visible remnants, shadows, outlines, or color traces remaining. "
-    "Once the objects are removed, the region should be seamlessly filled by **inpainting** using only the surrounding background information from the original image. "
-    "The inpainted area must blend smoothly with the unmasked regions so that no traces of the removal are visible.\n\n"
+    "OBJECT REMOVAL:\n\n"
+    "• **Completely remove** any visual content that exists strictly within the white mask region.\n"
+    "• Do **NOT** remove, modify, or reconstruct any object parts that extend outside the white mask.\n"
+    "• If the mask overlaps only part of an object, **remove ONLY the masked pixels** and leave the rest untouched.\n\n"
     
-    "INPAINTING INSTRUCTIONS:\n\n"
-    "1) **Object Recognition and Removal:**\n"
-    "   The model should first **understand** that the white areas in the binary mask represent the specific regions where objects or elements need to be **completely removed**. "
-    "   This could be anything from objects, people, text, or any unwanted areas in the image. It should not leave any part of the removed object behind — not even partial shapes, shadows, outlines, or color traces.\n"
-    "   The object removal should ensure that no part of the deleted object is visible in the final image.\n\n"
+    "INPAINTING RULES:\n\n"
+    "• **Fill the masked area** using ONLY immediate surrounding background pixels from Image A.\n"
+    "• **Continue existing textures, colors, lighting, and geometry naturally** to match the unmasked regions.\n"
+    "• **Do NOT invent new objects, structures, shadows, reflections, or background elements** in the inpainted region.\n"
+    "• **Do NOT smooth, blur, or stylize** the inpainted region. The result should look as natural and seamless as possible.\n\n"
     
-    "2) **Inpainting the Removed Region:**\n"
-    "   After removing the designated regions, the surrounding **background textures, patterns, and structures** should be used to **fill in** the removed area. "
-    "   This should be done by **extending the background information** naturally, continuing the textures, structures, and lighting patterns that were present in the surrounding unmasked regions.\n"
-    "   The model should focus on ensuring that the inpainted region **matches** the existing background in terms of:\n"
-    "   • **Lighting Direction** – Ensure the lighting in the inpainted region matches the natural direction and intensity of the surrounding areas.\n"
-    "   • **Brightness and Contrast** – The brightness and contrast should be consistent across the image, without any abrupt changes where the object was removed.\n"
-    "   • **Color Temperature** – Ensure the inpainted area aligns with the color temperature of the surrounding image (i.e., warm, cool, neutral tones).\n"
-    "   • **Noise and Texture** – Maintain the same noise level and texture (grain) in the inpainted region as in the surrounding areas, avoiding unnatural smoothness or patchiness.\n\n"
+    "STRICT CONSTRAINTS (NO EXCEPTIONS):\n\n"
+    "• **Do NOT** modify any black (unmasked) pixel.\n"
+    "• **Do NOT** remove entire objects unless they are **fully masked** (i.e., entirely white in the mask).\n"
+    "• **Do NOT** alter the image framing, perspective, or global appearance of Image A.\n"
+    "• **Do NOT** apply any semantic cleanup or scene optimization that goes beyond the direct instructions of the mask.\n\n"
     
-    "3) **Seamless Blending:**\n"
-    "   The inpainting should blend **seamlessly** with the surrounding areas, creating a smooth transition between the removed object and the rest of the image. "
-    "   This includes extending edges, lines, and surfaces naturally across the removed area. The goal is that **the boundary of the inpainted region should be indistinguishable from the original background**, "
-    "   without any visible edge halos, smearing, or rough transitions.\n\n"
+    "QUALITY REQUIREMENT:\n\n"
+    "• The final result must look like the **masked pixels never existed** in Image A.\n"
+    "• The inpainted region must blend **seamlessly** with surrounding pixels in terms of texture, color, and lighting.\n"
+    "• **No halos, ghosting, color shifts**, or **texture repetition** should be visible in the final output.\n\n"
     
-    "STRICT CONSTRAINTS:\n\n"
-    "• Do **NOT** generate or keep any part of the removed object. There should be no visible remnants, outlines, or partial shapes from the deleted object.\n"
-    "• Do **NOT** modify or enhance the unmasked areas (black regions). The unmasked regions should remain completely unchanged, with no alteration to the content, color, or lighting.\n"
-    "• Do **NOT** alter the **original composition**, **camera angle**, or **perspective** of the image. The overall structure of the image should remain intact.\n\n"
-    
-    "FAILURE CONDITIONS (MUST BE AVOIDED):\n\n"
-    "• If any **outline, shadow, or color fragment** from the removed object is still visible in the final result, the inpainting is considered incorrect and must be re-generated.\n"
-    "• If the inpainted region **does not blend** properly with the surrounding areas (i.e., visible transitions, edge halos, ghosting effects), the result is incorrect and must be re-generated.\n\n"
-    
-    "QUALITY REQUIREMENTS:\n\n"
-    "• The result must look like the object **never existed** in the original image.\n"
-    "• There should be no **ghosting**, **halo effects**, or **smearing** in the inpainted area.\n"
-    "• The final image must have **natural and realistic inpainting**. The inpainted area should seamlessly match the rest of the image, without any **patchy fills** or **repeated textures**.\n"
-    "• The **background** in the inpainted region should match the **lighting, texture, color**, and **depth** of the surrounding area.\n\n"
-    "Please ensure that all conditions are followed strictly for accurate and flawless object removal. The output should be a polished, high-quality image where the removed area is seamlessly inpainted without leaving any detectable signs of modification."
+    "IMPORTANT:\n\n"
+    "If a conflict exists between **semantic understanding** and the **mask**, **ALWAYS FOLLOW THE MASK EXACTLY**."
     ),
 )
 _GENAI_MODEL: genai.GenerativeModel | None = None