# from llamafactory.webui.interface import create_ui, create_web_demo

import os

from llamafactory.webui.interface import create_ui

def main():
    gradio_ipv6 = os.getenv("GRADIO_IPV6", "0").lower() in ["true", "1"]
    gradio_share = os.getenv("GRADIO_SHARE", "0").lower() in ["true", "1"]
    server_name = os.getenv("GRADIO_SERVER_NAME", "[::]" if gradio_ipv6 else "0.0.0.0")
    create_ui().queue().launch(share=gradio_share, server_name=server_name, server_port=6006, inbrowser=True)

if __name__ == "__main__":
    main()