import os

UI_VARIANT = os.getenv("MEGUMIN_UI", "default").strip().lower()

if UI_VARIANT == "messenger":
    from app_gradio_messenger import CUSTOM_CSS
    from app_gradio_messenger import MESSENGER_ALLOWED_PATHS
    from app_gradio_messenger import demo
else:
    from app_gradio import CUSTOM_CSS
    from app_gradio import demo


if __name__ == "__main__":
    launch_kwargs = {"css": CUSTOM_CSS, "ssr_mode": False}
    if UI_VARIANT == "messenger":
        launch_kwargs["allowed_paths"] = MESSENGER_ALLOWED_PATHS
    demo.launch(**launch_kwargs)