Update app.py

app.py

@@ -5,19 +5,27 @@ import argparse
 import os
 import urllib.request
 
-if __name__ == "__main__":
-    try:
-        os.makedirs("/home/user/.local/lib/python3.8/site-packages/wfd/mpqapi/")
-    except Exception as e:
-        print(e)
-    # 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")
+def fix_file(path):
+    # I think this is because I did the setup.py wrong, but pip install git+repo works in colab so idk how to really fix
+    target_path = "/home/user/.local/lib/python3.8/site-packages/wfd/{}".format(path)
+    if not os.path.exists(target_path):
+        try:
+            os.makedirs("/home/user/.local/lib/python3.8/site-packages/wfd/{}".format(os.path.dirname(path)))
+        except Exception as e:
+            print(e)
+        try:
+            urllib.request.urlretrieve(
+                "https://github.com/wdcqc/WaveFunctionDiffusion/raw/remaster/wfd/{}".format(path),
+                target_path
+            )
+        except Exception as e:
+            print(e)
 
-    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!!!")
+if __name__ == "__main__":
+    fix_file("mpqapi/libstorm.so")
+    fix_file("webui/templates/tempura.css")
+    fix_file("webui/templates/background.css")
+    fix_file("webui/doki_settings.json")
     
     from wfd.webui import start_demo
     class Arguments: