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: