Update app.py

app.py

@@ -5,11 +5,10 @@ import argparse
 import os
 import urllib.request
 import json
-import tqdm
+import importlib
 
-# the tqdm package is chosen with a dice, completely random.
-SITE_PACKAGES_DIR = os.path.dirname(os.path.dirname(tqdm.__file__))
-WFD_DIR = os.path.join(SITE_PACKAGES_DIR, "wfd")
+# do it properly
+WFD_DIR = os.path.dirname(importlib.util.find_spec("wfd").origin)
 
 def fix_theme(theme):
     try:
@@ -57,24 +56,26 @@ def fix_package_file(path):
         print("{} is fixed!".format(path))
 
 if __name__ == "__main__":
-    print("SITE_PACKAGES_DIR =", SITE_PACKAGES_DIR)
-    print("WFD_DIR =", WFD_DIR, "and it", "exists" if os.path.exists(WFD_DIR) else "does not exist")
-    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")
+    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")
+    # fix_theme("KonoSuba: Darkness Light")
     
-    from wfd.webui import start_demo
+    from wfd.webui import set_theme, start_demo
     class Arguments:
         colab = False
         link_to_colab = True
     args = Arguments()
+
+    set_theme("KonoSuba: Darkness Light")
+    
     start_demo(args)