Update app.py

app.py

@@ -10,67 +10,23 @@ import importlib
 # do it properly
 WFD_DIR = os.path.dirname(importlib.util.find_spec("wfd").origin)
 
-def fix_theme(theme):
+def set_theme(theme):
+ settings_file = os.path.join(WFD_DIR, "webui", "doki_settings.json")
  try:
- with open(os.path.join(WFD_DIR, "webui/doki_settings.json"), encoding = "utf-8") as fp:
+ with open(settings_file, encoding = "utf-8") as fp:
  doki = json.load(fp)
  
  doki["theme"] = theme
  
- with open(os.path.join(WFD_DIR, "webui/doki_settings.json"), "w", encoding = "utf-8") as fp:
+ with open(settings_file, "w", encoding = "utf-8") as fp:
  json.dump(doki, fp)
  except Exception as e:
  print(e)
 
-def fix_local_file(path):
- # probably should include the tile files in the package. let's get it running first.
- if not os.path.exists(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):
- # fixed by adding files to package_data and set include_package_data=True in setup.py
- target_path = os.path.join(WFD_DIR, path)
- if not os.path.exists(target_path):
- try:
- os.makedirs(os.path.join(WFD_DIR, 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)
- else:
- print("{} is fixed!".format(path))
-
 if __name__ == "__main__":
  print("WFD_DIR =", WFD_DIR)
- # 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")
-
- # fix_theme("KonoSuba: Darkness Light")
  
- from wfd.webui import set_theme, start_demo
+ from wfd.webui import start_demo
  class Arguments:
  colab = False
  link_to_colab = True