examples

demo.py

@@ -645,6 +645,28 @@ def latent_animation_to_grayscale(latent_animation):
     return temp_file
 
 
+# Add function to load view-specific mask
+def load_view_mask(view):
+    mask_path = f"assets/{view.lower()}_seg.png"
+    try:
+        mask_image = Image.open(mask_path).convert("L")
+        mask_image = mask_image.resize((400, 400), Image.Resampling.LANCZOS)
+        # Make it binary (0 or 255)
+        mask_image = ImageOps.autocontrast(mask_image, cutoff=0)
+        mask_array = np.array(mask_image)
+
+        # Create the editor value structure
+        editor_value = {
+            "background": np.zeros((400, 400), dtype=np.uint8),  # Black background
+            "layers": [mask_array],  # The mask as an editable layer
+            "composite": mask_array,  # The composite image
+        }
+        return editor_value
+    except Exception as e:
+        print(f"Error loading mask for view {view}: {e}")
+        return None
+
+
 def create_demo():
     # Define the theme and layout
     with gr.Blocks(theme=gr.themes.Soft()) as demo:
@@ -686,7 +708,7 @@ def create_demo():
 
                         # Load the default mask image if it exists
                         try:
-                            mask_image = Image.open("assets/seg.png").convert("L")
+                            mask_image = Image.open("assets/a4c_seg.png").convert("L")
                             mask_image = mask_image.resize(
                                 (400, 400), Image.Resampling.LANCZOS
                             )
@@ -758,7 +780,8 @@ def create_demo():
                     type="numpy",
                     height=400,
                     width=400,
-                    # show_label=False,
+                    show_download_button=True,
+                    interactive=False,  # Cannot be uploaded/edited
                 )
 
                 # Decode button (initially disabled)
@@ -774,7 +797,8 @@ def create_demo():
                     type="numpy",
                     height=400,
                     width=400,
-                    # show_label=False,
+                    show_download_button=True,
+                    interactive=False,  # Cannot be uploaded/edited
                 )
 
             # Column 2: Privacy Filter
@@ -869,6 +893,13 @@ def create_demo():
         latent_animation_state = gr.State(None)
 
         # Event handlers
+        class_selection.change(
+            fn=load_view_mask,
+            inputs=[class_selection],
+            outputs=[mask_input],
+            queue=False,
+        )
+
         generate_btn.click(
             fn=generate_latent_image,
             inputs=[mask_input, class_selection, sampling_steps],
@@ -954,6 +985,121 @@ def create_demo():
             queue=True,
         )
 
+        # Add examples
+        gr.Examples(
+            examples=[
+                # Example 1: A4C view
+                [
+                    # Inputs
+                    {
+                        "background": np.zeros((400, 400), dtype=np.uint8),
+                        "layers": [
+                            np.array(
+                                Image.open("assets/a4c_seg.png")
+                                .convert("L")
+                                .resize((400, 400))
+                            )
+                        ],
+                        "composite": np.array(
+                            Image.open("assets/a4c_seg.png")
+                            .convert("L")
+                            .resize((400, 400))
+                        ),
+                    },
+                    "A4C",  # view
+                    100,  # sampling steps
+                    65,  # EF slider
+                    100,  # animation steps
+                    1.0,  # cfg scale
+                    # Pre-computed outputs
+                    Image.open("assets/examples/a4c_latent.png"),  # latent image
+                    Image.open("assets/examples/a4c_decoded.png"),  # decoded image
+                    "✅ **Success:** Generated image passed privacy check.",  # privacy status
+                    Image.open("assets/examples/a4c_filtered.png"),  # filtered latent
+                    "assets/examples/a4c_latent.mp4",  # latent animation
+                    "assets/examples/a4c_decoded.mp4",  # decoded animation
+                ],
+                # Example 2: PSAX view
+                [
+                    # Inputs
+                    {
+                        "background": np.zeros((400, 400), dtype=np.uint8),
+                        "layers": [
+                            np.array(
+                                Image.open("assets/psax_seg.png")
+                                .convert("L")
+                                .resize((400, 400))
+                            )
+                        ],
+                        "composite": np.array(
+                            Image.open("assets/psax_seg.png")
+                            .convert("L")
+                            .resize((400, 400))
+                        ),
+                    },
+                    "PSAX",  # view
+                    150,  # sampling steps
+                    75,  # EF slider
+                    150,  # animation steps
+                    2.0,  # cfg scale
+                    # Pre-computed outputs
+                    Image.open("assets/examples/psax_latent.png"),  # latent image
+                    Image.open("assets/examples/psax_decoded.png"),  # decoded image
+                    "✅ **Success:** Generated image passed privacy check.",  # privacy status
+                    Image.open("assets/examples/psax_filtered.png"),  # filtered latent
+                    "assets/examples/psax_latent.mp4",  # latent animation
+                    "assets/examples/psax_decoded.mp4",  # decoded animation
+                ],
+                # Example 3: PLAX view
+                [
+                    # Inputs
+                    {
+                        "background": np.zeros((400, 400), dtype=np.uint8),
+                        "layers": [
+                            np.array(
+                                Image.open("assets/plax_seg.png")
+                                .convert("L")
+                                .resize((400, 400))
+                            )
+                        ],
+                        "composite": np.array(
+                            Image.open("assets/plax_seg.png")
+                            .convert("L")
+                            .resize((400, 400))
+                        ),
+                    },
+                    "PLAX",  # view
+                    200,  # sampling steps
+                    55,  # EF slider
+                    200,  # animation steps
+                    3.0,  # cfg scale
+                    # Pre-computed outputs
+                    Image.open("assets/examples/plax_latent.png"),  # latent image
+                    Image.open("assets/examples/plax_decoded.png"),  # decoded image
+                    "✅ **Success:** Generated image passed privacy check.",  # privacy status
+                    Image.open("assets/examples/plax_filtered.png"),  # filtered latent
+                    "assets/examples/plax_latent.mp4",  # latent animation
+                    "assets/examples/plax_decoded.mp4",  # decoded animation
+                ],
+            ],
+            inputs=[
+                mask_input,
+                class_selection,
+                sampling_steps,
+                ef_slider,
+                animation_steps,
+                cfg_slider,
+                latent_image_display,
+                decoded_image_display,
+                privacy_status,
+                filtered_latent_display,
+                latent_animation_display,
+                decoded_animation_display,
+            ],
+            label="Example Configurations",
+            examples_per_page=3,
+        )
+
     return demo