service/view/__init__.py

import gradio as gr
import os
from service.controllers import inference_tag2text

# 테스트 이미지 세팅
examples = ["./service/examples/" + i for i in os.listdir("./service/examples")]


def start_app():
    with gr.Blocks(title="유사사례 찾기") as app:
        with gr.Box():
            input_img = gr.Image(type="pil", label="사고 상황을 사진으로 입력", shape=(200, 200))
            gr.Examples(examples, inputs=[input_img], label="사진 예시")
            input_emphasis_word = gr.Text(label="강조하고 싶은 단어 입력")
            with gr.Row():
                find_btn = gr.Button("유사한 사고 사례를 찾습니다")
        with gr.Box():
            output_url = gr.Text(label="사고 상황과 유사한 영상 URL")
            output_tags = gr.Text(label="사고 상황 태그")
            output_caption = gr.Text(label="사고 상황 캡션")
        find_btn.click(
            fn=inference_tag2text,
            inputs=[input_img, input_emphasis_word],
            outputs=[output_url, output_tags, output_caption],
            api_name="greet",
        )

        return app