mirror sticker

app.py

@@ -170,13 +170,43 @@ def generate_3d_anaglyph_dynamic(person_img, side_by_side_bg, shift, scale_facto
     final_rgb = cv2.cvtColor(final_bgr, cv2.COLOR_BGR2RGB)
     
     # Create a full-image binary mask from the sticker's alpha channel.
-    # region_coords = (x_offset, y_offset, w, h)
     x, y, w, h = region_coords
     sticker_mask = (scaled_sticker[..., 3] > 128).astype(np.uint8)  # 1 inside sticker, 0 outside
     full_mask = np.zeros(final_rgb.shape[:2], dtype=np.uint8)
     full_mask[y:y+h, x:x+w] = sticker_mask
-    # Now full_mask==1 where the sticker is and 0 elsewhere.
+    return final_rgb, full_mask
+
+def generate_3d_anaglyph_dynamic_mirrored(person_img, side_by_side_bg, shift, scale_factor, x_frac, y_frac):
+    """
+    Generate a 3D anaglyph image with the sticker mirrored horizontally.
+    The procedure is identical to generate_3d_anaglyph_dynamic except that the sticker is flipped.
+    Returns:
+      - final_rgb: The final composite (RGB).
+      - full_mask: The binary sticker mask.
+    """
+    sticker = generate_sticker(person_img)
+    if sticker is None:
+        error_img = side_by_side_bg.copy()
+        cv2.putText(error_img, "No person detected.", (30, 30),
+                    cv2.FONT_HERSHEY_SIMPLEX, 1, (255, 0, 0), 2)
+        return error_img, None
+
+    # Mirror the sticker horizontally.
+    mirrored_sticker = cv2.flip(sticker, 1)
+    scaled_sticker = scale_sticker(mirrored_sticker, scale_factor)
+    left_bg, right_bg = split_side_by_side(side_by_side_bg)
+    
+    left_overlay, region_coords = overlay_sticker_dynamic(left_bg.copy(), scaled_sticker, shift=-shift, x_frac=x_frac, y_frac=y_frac)
+    right_overlay, _ = overlay_sticker_dynamic(right_bg.copy(), scaled_sticker, shift=shift, x_frac=x_frac, y_frac=y_frac)
     
+    anaglyph = create_anaglyph(left_overlay, right_overlay)
+    final_bgr = anaglyph
+    final_rgb = cv2.cvtColor(final_bgr, cv2.COLOR_BGR2RGB)
+    
+    x, y, w, h = region_coords
+    sticker_mask = (scaled_sticker[..., 3] > 128).astype(np.uint8)
+    full_mask = np.zeros(final_rgb.shape[:2], dtype=np.uint8)
+    full_mask[y:y+h, x:x+w] = sticker_mask
     return final_rgb, full_mask
 
 def adjust_bg_brightness(anaglyph_rgb, sticker_mask, brightness_factor):
@@ -218,10 +248,11 @@ Adjust:
 
 After the 3D image is generated, use the Background Brightness slider to adjust the brightness
 of the background only (the sticker remains unchanged).
+You can also mirror the sticker using the 'Mirror Sticker' button.
 """
 
 with gr.Blocks() as demo:
-    gr.Markdown(f"# 3D Anaglyph Generator with Background Brightness Adjustment\n\n{description}")
+    gr.Markdown(f"# 3D Anaglyph Generator with Background Brightness and Mirror Sticker Option\n\n{description}")
     
     with gr.Row():
         person_image = gr.Image(type="numpy", label="Upload Person Image")
@@ -257,5 +288,20 @@ with gr.Blocks() as demo:
         inputs=[anaglyph_state, mask_state, brightness_slider],
         outputs=output_image
     )
-
+    
+    mirror_btn = gr.Button("Mirror Sticker")
+    # When mirror_btn is clicked, regenerate the composite with the sticker mirrored.
+    mirror_btn.click(
+        generate_3d_anaglyph_dynamic_mirrored,
+        inputs=[person_image, side_by_side_bg, shift_slider, scale_slider, offset_x_slider, offset_y_slider],
+        outputs=[output_image, mask_state]
+    ).then(
+        lambda x: x,
+        inputs=output_image,
+        outputs=anaglyph_state
+    )
+    
+    # Place brightness slider and mirror button together.
+    gr.Markdown("Adjust the background brightness using the slider above, or mirror the sticker with the button below.")
+    
 demo.launch()