image_processing_service.py

# -*- coding: utf-8 -*-
"""
Created on Mon Jun 10 09:08:38 2024

@author: faceb
"""

import gradio as gr
import cv2
import numpy as np

def choose_one_methods_and_execute(
    one_img, # 圖片
    method, # 方法
    start_blue, # 去背景
    start_green, # 去背景
    rotation_options # 旋轉圖片
):

    if method == "去背景":
        return background_removal(
            one_img,
            start_blue # 一共有六個界線喔
        )

    elif method == "圖像旋轉":
        return rotate_image(
            one_img,
            rotation_options
        )

def background_removal(one_img, start_blue):

    lower_bound = np.array([10, 10, start_blue]) # RGB
    upper_bound = np.array([200, 100, 100])

    mask = cv2.inRange(one_img, lower_bound, upper_bound)

    masked_img = cv2.bitwise_and(one_img, one_img, mask = mask)

    return gr.update(visible = True), masked_img

def rotate_image(one_img, rotate_image):

    if rotate_image == 90:
        img = cv2.rotate(one_img, cv2.ROTATE_90_CLOCKWISE)
    elif rotate_image == 180:
        img = cv2.rotate(one_img, cv2.ROTATE_180)
    elif rotate_image == 270:
        pass
    else:
        pass
    return gr.update(visible = True), img

with gr.Blocks() as demo:
    # with gr.Tab("影像處理") as image_tab:
    upload_image = gr.Image(
        label = "圖片上傳"
    )
    image_processing_methods = gr.Dropdown( # 之後加入其他功能後，記得要在這裡加入選項
        label = "圖像處理技巧",
        choices = [
            "去背景",
            "圖像旋轉"
        ]
    )

    with gr.Row():
        start_blue = gr.Slider(
            minimum = 0,
            maximum = 255,
            label = "藍色低界線"
        )
        start_red = gr.Slider(
            minimum = 0,
            maximum = 255,
            label = "紅色低界線"
        )
        start_green = gr.Slider(
            minimum = 0,
            maximum = 255,
            label = "綠色低界線"
        )

    with gr.Row():
        end_blue = gr.Slider(
            minimum = 0,
            maximum = 255,
            label = "藍色高界線"
        )
        end_red =gr.Slider(
            minimum = 0,
            maximum = 255,
            label = "紅色高界線"
        )
        end_green = gr.Slider(
            minimum = 0,
            maximum = 255,
            label = "綠色高界線"
        )

    # 圖片旋轉數度等設定
    rotation_options = gr.Dropdown(
        choices = [90, 180, 270],
        max_choices = 1,
        label = "請選擇圖片旋轉度數"
    )

    image_button = gr.Button(
        value = "開始分析"
    )
    display_image = gr.Image(
        label = "結果",
        visible = False
    )

    # with gr.Tab("影片處理") as video_tab:
    #     upload_video = gr.Video(
    #         label = "影片上傳"
    #     )

    image_button.click(
        fn = choose_one_methods_and_execute,
        inputs = [
            upload_image, # 圖片
            image_processing_methods, # 方法
            start_blue,
            start_green,
            rotation_options
        ],
        outputs = [display_image, display_image]
    )

demo.launch(debug = True)