src/myapp.py

# -*- coding: utf-8 -*-
from typing import Literal

import argparse
from datetime import datetime
import sys

import gradio as gr
import numpy as np

import src.utils as utils
from src.saliency import SaliencyMap, convertColorMap
from src.reporter import get_current_reporter

PROGRAM_NAME = 'SaliencyMapDemo'
__version__ = utils.get_package_version()
log = get_current_reporter()
    
def jetTab_Selected(image: np.ndarray):
    #print(f"{datetime.now()}#jet")
    saliency = SaliencyMap("SpectralResidual")
    success, saliencyMap = saliency.computeSaliency(image)
    retval = convertColorMap(image, saliencyMap, "jet")
    #print(f"{datetime.now()}#jet")
    
    return retval
    
def hotTab_Selected(image: np.ndarray):
    #print(f"{datetime.now()}#hot")
    saliency = SaliencyMap("SpectralResidual")
    success, saliencyMap = saliency.computeSaliency(image)
    retval = convertColorMap(image, saliencyMap, "turbo")
    #print(f"{datetime.now()}#hot")
    
    return retval

def submit_Clicked(image: np.ndarray, algorithm: str):
    """
    入力画像を元に顕著マップを計算します。

    Parameters:
        image: 入力画像
        str: 顕著性マップのアルゴリズム
    Returns:
        np.ndarray: JET画像
        np.ndarray: HOT画像
    """
    #log.info(f"#submit_Clicked")
    watch = utils.Stopwatch.startNew()
    
    saliency = SaliencyMap(algorithm)
    success, saliencyMap = saliency.computeSaliency(image)
    #log.info(f"#SaliencyMap computeSaliency()")

    if not success:
        return image, image # エラーが発生した場合は入力画像を返します。

    #log.info(f"#jet")        
    jet = convertColorMap(image, saliencyMap, "jet")
    #jet = None
    #log.info(f"#hot")
    hot = convertColorMap(image, saliencyMap, "hot")
    
    saliency = None
    #log.info(f"#submit_Clicked End{watch.stop():.3f}")
    
    return jet, hot

def runApp(args: argparse.Namespace, watch: utils.Stopwatch) -> None:
    """
    アプリの画面を作成し、Gradioサービスを起動します。

    Parameters:
        args: コマンドライン引数
        watch: 起動したスタート時間
    """
    # analytics_enabled=False
    # https://github.com/gradio-app/gradio/issues/4226
    with gr.Blocks(analytics_enabled=False, \
        title=f"{PROGRAM_NAME} {__version__}", \
        head="""
        <meta name="format-detection" content="telephone=no">
        <meta name="robots" content="noindex, nofollow, noarchive">
        <meta name="referrer" content="no-referrer" />
        """) as demo:
    	
        gr.Markdown(
        """
        # Saliency Map demo.
        """)
        with gr.Accordion("取り扱い説明書", open=False):
            gr.Markdown(
            """
            1. inputタブで画像を選択します。
            2. Submitボタンを押します。
               ※画像は外部送信していません。ローカルで処理が完結します。
            3. 結果は、JETタブとHOTタブに表示します。  
            """)

        algorithmType = gr.Radio(["SpectralResidual", "FineGrained"], label="Saliency", value="SpectralResidual", interactive=True)

        submit_button = gr.Button("submit")

        with gr.Row():
            with gr.Tab("input", id="input"):

                image_input = gr.Image(sources = ["upload", "clipboard"], interactive=True)
            with gr.Tab("overlay(JET)"):
                image_overlay_jet = gr.Image(interactive=False)
                #tab_jet.select(jetTab_Selected, inputs=[image_input], outputs=image_overlay_jet)
            with gr.Tab("overlay(HOT)"):
                image_overlay_hot = gr.Image(interactive=False)
                #tab_hot.select(hotTab_Selected, inputs=[image_input], outputs=image_overlay_hot, api_name=False)
        
        submit_button.click(submit_Clicked, inputs=[image_input, algorithmType], outputs=[image_overlay_jet, image_overlay_hot])

        gr.Markdown(
        f"""
        Python {sys.version}  
        App {__version__}  
        """)
        
        demo.queue(default_concurrency_limit=5)
        
        log.info(f"#アプリ起動完了({watch.stop():.3f}s)")
        # https://www.gradio.app/docs/gradio/blocks#blocks-launch
        
        demo.launch(
            max_file_size=args.max_file_size,
            server_port=args.server_port,
            inbrowser=args.inbrowser,
            share=args.share,
        )