Update app.py

app.py

@@ -21,25 +21,25 @@ def resize_image(img, scale):
 
 
 # Canny 邊緣偵測
-def canny_edge1(img):
+def canny_edge1(img,min_val,max_val):
     if img is None:
         return None
     global imageScale
-    img = resize_image(img, imageScale)  # 圖片大小保持
+    img = resize_image(img, imageScale) 
     img_cv = np.array(img)
     img_gray = cv2.cvtColor(img_cv, cv2.COLOR_RGB2GRAY)
-    edges = cv2.Canny(img_gray, 100, 200)
+    edges = cv2.Canny(img_gray, min_val, max_val)
     
     return Image.fromarray(edges)
 
-def canny_edge2(img):
+def canny_edge2(img,min_val,max_val):
     if img is None:
         return None
     global imageScale
-    img = resize_image(img, imageScale) 
+    img = resize_image(img, imageScale)
     img_cv = np.array(img)
     img_gray = cv2.cvtColor(img_cv, cv2.COLOR_RGB2GRAY)
-    edges = cv2.Canny(img_gray, 100, 200)
+    edges = cv2.Canny(img_gray, min_val, max_val)
     edges_colored = cv2.cvtColor(edges, cv2.COLOR_GRAY2BGR)
     edges_colored[:, :, 1:] = 0
     combined = cv2.addWeighted(img_cv, 0.8, edges_colored, 0.8, 0)
@@ -159,6 +159,9 @@ with gr.Blocks() as demo:
         canny_button = gr.Button("Canny 邊緣偵測", elem_id="button1")
         sobel_button = gr.Button("索伯算子 邊緣偵測", elem_id="button2")
         laplacian_button = gr.Button("Laplacian 邊緣偵測", elem_id="button3")
+    with gr.Row():
+        slider_canny1 = gr.Slider(1, 150, step=1, value=100, label="canny threshold 1 (min val)")
+        slider_canny2 = gr.Slider(150, 300, step=1, value=200, label="canny threshold 2 (max val)")
     with gr.Row():
         img_input2 = gr.Image(type="pil", label="範例圖片1", value=default_image2, interactive=False)
         img_input2 = gr.Image(type="pil", label="範例圖片2", value=default_image, interactive=False)
@@ -166,8 +169,8 @@ with gr.Blocks() as demo:
     # 連結功能
     resize_button.click(fn=resize_image, inputs=[img_input, slider], outputs=img_output2)
     resize_button.click(fn=resize_image, inputs=[img_input, slider], outputs=img_output1)
-    canny_button.click(fn=canny_edge1, inputs=[img_input], outputs=img_output1)
-    canny_button.click(fn=canny_edge2, inputs=[img_input], outputs=img_output2)
+    canny_button.click(fn=canny_edge1, inputs=[img_input,slider_canny1,slider_canny2], outputs=img_output1)
+    canny_button.click(fn=canny_edge2, inputs=[img_input,slider_canny1,slider_canny2], outputs=img_output2)
     sobel_button.click(fn=sobel_operators1, inputs=[img_input], outputs=img_output1)
     sobel_button.click(fn=sobel_operators2, inputs=[img_input], outputs=img_output2)
     laplacian_button.click(fn=Laplacian1, inputs=[img_input], outputs=img_output1)