Update app.py

app.py

@@ -136,21 +136,56 @@ def process_image(input_image: Image.Image, fg_threshold: float, mg_threshold: f
         # mask_bg_img
     )
 
+def update_preset(preset: str):
+    presets = {
+        "Preset 1": {
+            "image_url": "https://i.ibb.co/fznz2b2b/hw3-q2.jpg",
+            "fg_threshold": 0.33,
+            "mg_threshold": 0.66
+        },
+        "Preset 2": {
+            "image_url": "https://i.ibb.co/HLZGW7qH/q26.jpg",
+            "fg_threshold": 0.2,
+            "mg_threshold": 0.66
+        }
+    }
+    preset_info = presets[preset]
+    response = requests.get(preset_info["image_url"])
+    image = Image.open(BytesIO(response.content)).convert("RGB")
+    return image, preset_info["fg_threshold"], preset_info["mg_threshold"]
+
+# -----------------------------
+# Gradio Interface Setup
+# -----------------------------
 title = "Blur Effects: Gaussian & Depth-Based Lens Blur with Adjustable Depth Thresholds"
 description = (
-    "Upload an image to apply two distinct effects:\n\n"
+    "Choose a preset image or upload your own image to apply two distinct effects:\n\n"
     "1. A segmentation-based Gaussian blur that blurs the background (using RMBG-2.0).\n"
     "2. A depth-based lens blur effect that simulates realistic lens blur based on depth (using DepthPro).\n\n"
-    "Use the sliders to adjust the foreground and middleground depth thresholds."
+    "The preset selection will automatically set the image and its default foreground/middleground depth thresholds."
 )
 
+# A radio to select between two preset images
+preset_radio = gr.Radio(
+    choices=["Preset 1", "Preset 2"],
+    label="Select Preset Image",
+    value="Preset 1"
+)
+
+# An image component that will be updated based on the preset selection
+preset_img = gr.Image(type="pil", label="Input Image")
+
+# Slider components for depth thresholds (will be updated when a preset is selected)
+fg_slider = gr.Slider(minimum=0, maximum=1, step=0.01, value=0.33, label="Foreground Depth Threshold")
+mg_slider = gr.Slider(minimum=0, maximum=1, step=0.01, value=0.66, label="Middleground Depth Threshold")
+
+# When the preset is changed, update the image and sliders with corresponding defaults.
+preset_radio.change(fn=update_preset, inputs=preset_radio, outputs=[preset_img, fg_slider, mg_slider])
+
+# Build the Gradio interface using the updated components.
 demo = gr.Interface(
     fn=process_image,
-    inputs=[
-        gr.Image(type="pil", label="Input Image", value="https://i.ibb.co/fznz2b2b/hw3-q2.jpg"),
-        gr.Slider(minimum=0, maximum=1, step=0.01, value=0.33, label="Foreground Depth Threshold"),
-        gr.Slider(minimum=0, maximum=1, step=0.01, value=0.66, label="Middleground Depth Threshold")
-    ],
+    inputs=[preset_img, fg_slider, mg_slider],
     outputs=[
         gr.Image(type="pil", label="Segmentation-Based Blur"),
         gr.Image(type="pil", label="Depth Map"),
@@ -163,3 +198,31 @@ demo = gr.Interface(
 
 if __name__ == "__main__":
     demo.launch()
+
+# title = "Blur Effects: Gaussian & Depth-Based Lens Blur with Adjustable Depth Thresholds"
+# description = (
+#     "Upload an image to apply two distinct effects:\n\n"
+#     "1. A segmentation-based Gaussian blur that blurs the background (using RMBG-2.0).\n"
+#     "2. A depth-based lens blur effect that simulates realistic lens blur based on depth (using DepthPro).\n\n"
+#     "Use the sliders to adjust the foreground and middleground depth thresholds."
+# )
+
+# demo = gr.Interface(
+#     fn=process_image,
+#     inputs=[
+#         gr.Image(type="pil", label="Input Image", value="https://i.ibb.co/fznz2b2b/hw3-q2.jpg"),
+#         gr.Slider(minimum=0, maximum=1, step=0.01, value=0.33, label="Foreground Depth Threshold"),
+#         gr.Slider(minimum=0, maximum=1, step=0.01, value=0.66, label="Middleground Depth Threshold")
+#     ],
+#     outputs=[
+#         gr.Image(type="pil", label="Segmentation-Based Blur"),
+#         gr.Image(type="pil", label="Depth Map"),
+#         gr.Image(type="pil", label="Depth-Based Lens Blur")
+#     ],
+#     title=title,
+#     description=description,
+#     allow_flagging="never"
+# )
+
+# if __name__ == "__main__":
+#     demo.launch()