import time

import gradio as gr
import modelscope_studio.components.base as ms

messages = {
    'en': {
        "hello": "Hello"
    },
    'en-US': {
        "hello": "Hello"
    },
    'zh-CN': {
        "hello": "你好"
    },
    "ja-JP": {
        "hello": "こんにちは"
    },
    "ko-KR": {
        "hello": "안녕하세요"
    }
}

default_lang = "en"


def mount(e: gr.EventData, _state):
    lang = e._data["language"]
    if lang in messages:
        _state["current_lang"] = lang
    yield 'Switch Language...', _state
    time.sleep(2)
    yield messages[_state["current_lang"]]["hello"], _state


with gr.Blocks() as demo:
    with ms.Application() as app:
        state = gr.State({"current_lang": default_lang})
        markdown = gr.Markdown(value=messages[default_lang]["hello"])

        app.mount(fn=mount, inputs=[state], outputs=[markdown, state])

if __name__ == "__main__":
    demo.queue().launch()