# -*- coding: utf-8 -*-
# Copyright @ 2023 wdcqc/aieud project.
# Open-source under license obtainable in project root (LICENSE.md).
import argparse
import os
import urllib.request

if __name__ == "__main__":
    os.makedirs("/home/user/.local/lib/python3.8/site-packages/wfd/mpqapi/")
    # try to fix the CDLL problem
    urllib.request.urlretrieve("https://github.com/wdcqc/WaveFunctionDiffusion/raw/remaster/wfd/mpqapi/libstorm.so", "/home/user/.local/lib/python3.8/site-packages/wfd/mpqapi/libstorm.so")
    urllib.request.urlretrieve("https://github.com/wdcqc/WaveFunctionDiffusion/raw/remaster/wfd/mpqapi/libstorm.so", "libstorm.so")

    if os.path.exists("/home/user/.local/lib/python3.8/site-packages/wfd/mpqapi/libstorm.so"):
        print("libstorm EXISTS!!!")
    else:
        print("libstorm DOES NOT EXIST!!!")
    
    from wfd.webui import start_demo
    class Arguments:
        colab = False
        link_to_colab = True
    args = Arguments()
    start_demo(args)