Update app.py

app.py

@@ -16,30 +16,27 @@ def image_segmentation(image, compactness):
     return color.label2rgb(segments, image, kind='avg')
 
 # 色彩濾鏡函式
-def color_filter(image, color_choice):
-    # 將圖像轉換為 HSV 格式
-    hsv_image = cv2.cvtColor(image, cv2.COLOR_RGB2HSV)
+def color_filter_with_rgb(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)
     
-    # 定義顏色範圍（H: 色調，S: 飽和度，V: 明度）
-    if color_choice == "紅色":
-        lower_bound = np.array([0, 50, 50])  # 紅色範圍
-        upper_bound = np.array([10, 255, 255])
-    elif color_choice == "綠色":
-        lower_bound = np.array([40, 50, 50])  # 綠色範圍
-        upper_bound = np.array([80, 255, 255])
-    elif color_choice == "藍色":
-        lower_bound = np.array([100, 50, 50])  # 藍色範圍
-        upper_bound = np.array([140, 255, 255])
-    else:
-        return "未支援的顏色範圍選擇"
+    # 分離 R、G、B 通道
+    r_channel = image[:, :, 0]
+    g_channel = image[:, :, 1]
+    b_channel = image[:, :, 2]
     
-    # 創建遮罩，過濾選定的顏色範圍
-    mask = cv2.inRange(hsv_image, lower_bound, upper_bound)
+    # 建立遮罩，篩選符合 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 == 0] = np.stack([gray_image] * 3, axis=-1)[mask == 0]
+    result[~mask] = np.stack([gray_image] * 3, axis=-1)[~mask]
     
     return result
 
@@ -76,21 +73,33 @@ def app():
                 outputs=seg_result
             )
         
-        with gr.Tab("色彩濾鏡"):
+        with gr.Blocks() as demo:
+        gr.Markdown("# 影像處理功能展示")
+        gr.Markdown("本應用程式展示了使用 OpenCV 實現的影像處理功能，包括手動調整 RGB 範圍的色彩濾鏡功能。")
+        
+        with gr.Tab("色彩濾鏡 (手動調整 RGB)"):
             with gr.Row():
                 input_image = gr.Image(label="輸入圖片")
                 filter_result = gr.Image(label="濾鏡結果")
             with gr.Row():
-                color_choice = gr.Radio(["紅色", "綠色", "藍色"], label="選擇顏色", value="紅色")
+                r_min = gr.Slider(0, 255, value=0, step=1, label="R 最小值")
+                r_max = gr.Slider(0, 255, value=255, step=1, label="R 最大值")
+                g_min = gr.Slider(0, 255, value=0, step=1, label="G 最小值")
+                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, 
-                inputs=[input_image, color_choice], 
+                fn=color_filter_with_rgb,
+                inputs=[input_image, r_min, r_max, g_min, g_max, b_min, b_max],
                 outputs=filter_result
             )
     
     return demo
 
+    
+    return demo
+
 # 啟動應用程式
 if __name__ == "__main__":
     app().launch()