app.py

# -*- 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)