Update app.py

app.py

@@ -1,5 +1,4 @@
 import gradio as gr
-from skimage import color
 import cv2
 import numpy as np
 
@@ -12,39 +11,30 @@ def edge_detection(image, threshold1, threshold2):
 # 圖像分割函式
 def image_segmentation(image, compactness):
     from skimage.segmentation import slic
+    from skimage import color
     segments = slic(image, compactness=compactness, n_segments=200)
     return color.label2rgb(segments, image, kind='avg')
 
-# 色彩濾鏡函式
-def color_filter_with_rgb(image, r_min, r_max, g_min, g_max, b_min, b_max):
+# 顏色範圍調整函式
+def adjust_color(image, r_min, r_max, g_min, g_max, b_min, b_max):
     # 確保圖片格式為 RGB
     if image.shape[-1] == 4:  # 如果圖片有 alpha 通道
         image = cv2.cvtColor(image, cv2.COLOR_RGBA2RGB)
     
-    # 分離 R、G、B 通道
-    r_channel = image[:, :, 0]
-    g_channel = image[:, :, 1]
-    b_channel = image[:, :, 2]
+    # 調整 R、G、B 通道的範圍
+    image = image.astype(np.float32)  # 將圖像轉為浮點數進行處理
+    image[:, :, 0] = np.clip(image[:, :, 0], r_min, r_max)  # 調整 R 通道
+    image[:, :, 1] = np.clip(image[:, :, 1], g_min, g_max)  # 調整 G 通道
+    image[:, :, 2] = np.clip(image[:, :, 2], b_min, b_max)  # 調整 B 通道
+    image = image.astype(np.uint8)  # 轉回整數型圖像
     
-    # 建立遮罩，篩選符合 RGB 範圍的像素
-    mask = (
-        (r_channel >= r_min) & (r_channel <= r_max) &
-        (g_channel >= g_min) & (g_channel <= g_max) &
-        (b_channel >= b_min) & (b_channel <= b_max)
-    )
-    
-    # 創建結果圖片，保留符合範圍的顏色，其他設為灰階
-    result = image.copy()
-    gray_image = cv2.cvtColor(image, cv2.COLOR_RGB2GRAY)
-    result[~mask] = np.stack([gray_image] * 3, axis=-1)[~mask]
-    
-    return result
+    return image
 
 # 主應用程式
 def app():
     with gr.Blocks() as demo:
         gr.Markdown("# 影像處理功能展示")
-        gr.Markdown("本應用程式展示了使用 Scikit-Image 與 OpenCV 實現的基本影像處理功能。")
+        gr.Markdown("本應用程式展示了使用 OpenCV 實現的影像處理功能，包括動態調整 RGB 範圍的功能。")
         
         with gr.Tab("邊緣檢測"):
             with gr.Row():
@@ -73,10 +63,10 @@ def app():
                 outputs=seg_result
             )
         
-        with gr.Tab("色彩濾鏡 (手動調整 RGB)"):
+        with gr.Tab("顏色範圍調整"):
             with gr.Row():
                 input_image = gr.Image(label="輸入圖片")
-                filter_result = gr.Image(label="濾鏡結果")
+                adjusted_result = gr.Image(label="調整後的圖片")
             with gr.Row():
                 r_min = gr.Slider(0, 255, value=0, step=1, label="R 最小值")
                 r_max = gr.Slider(0, 255, value=255, step=1, label="R 最大值")
@@ -84,11 +74,11 @@ def app():
                 g_max = gr.Slider(0, 255, value=255, step=1, label="G 最大值")
                 b_min = gr.Slider(0, 255, value=0, step=1, label="B 最小值")
                 b_max = gr.Slider(0, 255, value=255, step=1, label="B 最大值")
-            filter_button = gr.Button("應用濾鏡")
-            filter_button.click(
-                fn=color_filter_with_rgb,
+            adjust_button = gr.Button("調整顏色範圍")
+            adjust_button.click(
+                fn=adjust_color,
                 inputs=[input_image, r_min, r_max, g_min, g_max, b_min, b_max],
-                outputs=filter_result
+                outputs=adjusted_result
             )
     
     return demo