app.py

import os
os.system("pip install gradio --upgrade")

import gradio as gr
from pyjosa.josa import Josa
import random
import json

with open('pokemon.json', 'r') as f:
    pokemons = json.load(f)

QUESTION_TEMPLATE = {"question": "다음 포켓몬의 이름은 뭘까요?![]({img_url})", "answer": "{name}"}
with gr.Blocks() as demo:
    quiz_start = gr.State(value=False)
    answer = gr.State(value="")
    chatbot = gr.Chatbot(bubble_full_width=False)
    msg = gr.Textbox()
    clear = gr.ClearButton([msg, chatbot])


    def respond(message, chat_history):
        if not quiz_start.value:
            chosen = random.choice(pokemons)
            name = chosen['name']
            image_path = chosen['image_path']
            answer.value = QUESTION_TEMPLATE['answer'].format(name=name)
            img_url = f"https://huggingface.co/spaces/yoon-gu/pokemon/resolve/main/{image_path}"
            if "퀴즈 시작" == message:
                bot_message = "퀴즈를 시작합니다.\n" + QUESTION_TEMPLATE["question"].format(img_url=img_url)
                quiz_start.value = True
            else:
                bot_message = "퀴즈를 시작하고 싶으시면, **퀴즈 시작**이라고 말씀해주세요."
        else:
            trial = Josa.get_full_string(message, "는")
            if answer.value == message:
                chosen = random.choice(pokemons)
                name = chosen['name']
                image_path = chosen['image_path']
                answer.value = QUESTION_TEMPLATE['answer'].format(name=name)
                img_url = f"https://huggingface.co/spaces/yoon-gu/pokemon/resolve/main/{image_path}"
                bot_message = "정답입니다! 다음 문제입니다." + QUESTION_TEMPLATE["question"].format(img_url=img_url)
            else:
                bot_message = f"{trial} 정답일까요? 다시 한번 생각해보세요."

        chat_history.append((message, bot_message))
        return "", chat_history

    msg.submit(respond, [msg, chatbot], [msg, chatbot])

demo.launch()