Add HSV Color Filter tab and functionality to image processing app

app.py

@@ -77,7 +77,7 @@ def apply_edge_detection(image, min_val, max_val, operation, kernel_size):
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
     gr.Markdown("# OpenCV Image Processing with Gradio - Add Noise, Remove Noise, Morphological Operations and Edge Detection")
 
-    tab_names = ["Add Noise", "Remove Noise", "Morphological Operations", "Edge Detection"]
+    tab_names = ["Add Noise", "Remove Noise", "Morphological Operations", "Edge Detection", "HSV Color Filter"]
 
     # ---- ADD NOISE TAB ----
     with gr.Tab("Add Noise"):
@@ -165,6 +165,46 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         edge_operation.change(fn=on_edge_operation_change, inputs=edge_operation, outputs=[min_val_slider, max_val_slider, kernel_size_slider])
         apply_edge_button.click(fn=apply_edge_detection, inputs=[edge_img_input, min_val_slider, max_val_slider, edge_operation, kernel_size_slider], outputs=edge_img_output)
 
+    # ---- HSV COLOR FILTER TAB ----
+    with gr.Tab("HSV Color Filter"):
+        with gr.Row():
+            hsv_img_input = gr.Image(label="Input Image", type="pil")
+            hsv_img_output = gr.Image(label="Filtered Output", type="pil")
+        with gr.Row():
+            with gr.Column():
+                h_min = gr.Slider(0, 179, value=0, label="Hue Min")
+                s_min = gr.Slider(0, 255, value=0, label="Saturation Min")
+                v_min = gr.Slider(0, 255, value=0, label="Value Min")
+            with gr.Column():
+                h_max = gr.Slider(0, 179, value=179, label="Hue Max")
+                s_max = gr.Slider(0, 255, value=255, label="Saturation Max")
+                v_max = gr.Slider(0, 255, value=255, label="Value Max")
+        show_mask = gr.Checkbox(label="Show Binary Mask Only", value=False)
+        apply_hsv_button = gr.Button("Apply HSV Filter")
+
+        def apply_hsv_filter(image, hmin, hmax, smin, smax, vmin, vmax, show_mask_flag):
+            if image is None:
+                return None
+            img = np.array(image.convert('RGB'))
+            hsv = cv2.cvtColor(img, cv2.COLOR_RGB2HSV)
+            # Ensure min <= max for each channel
+            if hmin > hmax: hmin, hmax = hmax, hmin
+            if smin > smax: smin, smax = smax, smin
+            if vmin > vmax: vmin, vmax = vmax, vmin
+            lower = np.array([hmin, smin, vmin], dtype=np.uint8)
+            upper = np.array([hmax, smax, vmax], dtype=np.uint8)
+            mask = cv2.inRange(hsv, lower, upper)
+            if show_mask_flag:
+                return Image.fromarray(mask)
+            result = cv2.bitwise_and(img, img, mask=mask)
+            return Image.fromarray(result)
+
+        apply_hsv_button.click(
+            fn=apply_hsv_filter,
+            inputs=[hsv_img_input, h_min, h_max, s_min, s_max, v_min, v_max, show_mask],
+            outputs=hsv_img_output
+        )
+
     gr.Markdown("### To transfer the image to another tab for further processing, select the source and destination tabs and click the Transfer Image button.")
     
     # ---- DYNAMIC TRANSFER BUTTON ----
@@ -173,7 +213,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         destination_tab_dropdown = gr.Dropdown(tab_names, label="Transfer To Tab")
         transfer_image_button = gr.Button("Transfer Image")
 
-    def dynamic_image_transfer(add_noise_input, add_noise_output, denoise_input, denoise_output, morph_input, morph_output, edge_input, edge_output, source, destination):
+    def dynamic_image_transfer(add_noise_input, add_noise_output, denoise_input, denoise_output, morph_input, morph_output, edge_input, edge_output, hsv_input, hsv_output, source, destination):
         image_to_send = None
         if source == "Add Noise":
             image_to_send = add_noise_output if add_noise_output else add_noise_input
@@ -183,20 +223,23 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             image_to_send = morph_output if morph_output else morph_input
         elif source == "Edge Detection":
             image_to_send = edge_output if edge_output else edge_input
+        elif source == "HSV Color Filter":
+            image_to_send = hsv_output if hsv_output else hsv_input
 
         updates = {
             "Add Noise": gr.update(value=image_to_send) if destination == "Add Noise" else gr.update(),
             "Remove Noise": gr.update(value=image_to_send) if destination == "Remove Noise" else gr.update(),
             "Morphological Operations": gr.update(value=image_to_send) if destination == "Morphological Operations" else gr.update(),
             "Edge Detection": gr.update(value=image_to_send) if destination == "Edge Detection" else gr.update(),
+            "HSV Color Filter": gr.update(value=image_to_send) if destination == "HSV Color Filter" else gr.update(),
         }
 
-        return [updates.get("Add Noise"), updates.get("Remove Noise"), updates.get("Morphological Operations"), updates.get("Edge Detection")]
+        return [updates.get("Add Noise"), updates.get("Remove Noise"), updates.get("Morphological Operations"), updates.get("Edge Detection"), updates.get("HSV Color Filter")]
 
     transfer_image_button.click(
         fn=dynamic_image_transfer,
-        inputs=[img_input, img_output, denoise_img_input, denoise_img_output, morph_img_input, morph_img_output, edge_img_input, edge_img_output, source_tab_dropdown, destination_tab_dropdown],
-        outputs=[img_input, denoise_img_input, morph_img_input, edge_img_input]
+        inputs=[img_input, img_output, denoise_img_input, denoise_img_output, morph_img_input, morph_img_output, edge_img_input, edge_img_output, hsv_img_input, hsv_img_output, source_tab_dropdown, destination_tab_dropdown],
+        outputs=[img_input, denoise_img_input, morph_img_input, edge_img_input, hsv_img_input]
     )
 
     gr.Markdown("""