Update app.py

app.py

@@ -5,7 +5,22 @@ import argparse
 import os
 import urllib.request
 
-def fix_file(path):
+def fix_local_file(path):
+    # probably should include the tile files in the package
+    if not os.path.exists(target_path):
+        try:
+            os.makedirs(os.path.dirname(path))
+        except Exception as e:
+            print(e)
+        try:
+            urllib.request.urlretrieve(
+                "https://github.com/wdcqc/WaveFunctionDiffusion/raw/remaster/{}".format(path),
+                path
+            )
+        except Exception as e:
+            print(e)
+
+def fix_package_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):
@@ -20,18 +35,20 @@ def fix_file(path):
             )
         except Exception as e:
             print(e)
+    else:
+        print("{} is fixed!".format(path))
 
 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")
-    fix_file("tile_data/wfc/platform_32x32.npz")
-    fix_file("tile_data/platform.cv5.bin")
-    fix_file("tile_data/platform.png")
-    fix_file("tile_data/platform_mapping.npz")
-    fix_file("tile_data/platform_v2.npz")
-    fix_file("default/base.chk")
+    fix_package_file("mpqapi/libstorm.so")
+    fix_package_file("webui/templates/tempura.css")
+    fix_package_file("webui/templates/background.css")
+    fix_package_file("webui/doki_settings.json")
+    fix_local_file("tile_data/platform.cv5.bin")
+    fix_local_file("tile_data/platform.png")
+    fix_local_file("tile_data/platform_mapping.npz")
+    fix_local_file("tile_data/platform_v2.npz")
+    fix_local_file("tile_data/wfc/platform_32x32.npz")
+    fix_local_file("default/base.chk")
     
     from wfd.webui import start_demo
     class Arguments: