Update app.py

app.py

@@ -2,13 +2,21 @@
 # Copyright @ 2023 wdcqc/aieud project.
 # Open-source under license obtainable in project root (LICENSE.md).
 import argparse
+import os
 import urllib.request
-from wfd.webui import start_demo
 
 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