Create loadThemes.py

loadThemes.py

@@ -0,0 +1,122 @@
+import json
+import os
+import importlib
+import gradio as gr
+
+now_dir = os.getcwd()
+
+folder = os.path.dirname(os.path.abspath(__file__))
+folder = os.path.dirname(folder)
+folder = os.path.dirname(folder)
+folder = os.path.join(folder, "assets", "themes")
+config_file = os.path.join(now_dir, "assets", "config.json")
+
+import sys
+
+sys.path.append(folder)
+
+
+def get_class(filename):
+    with open(filename, "r", encoding="utf8") as file:
+        for line_number, line in enumerate(file, start=1):
+            if "class " in line:
+                found = line.split("class ")[1].split(":")[0].split("(")[0].strip()
+                return found
+                break
+    return None
+
+
+def get_list():
+
+    themes_from_files = [
+        os.path.splitext(name)[0]
+        for root, _, files in os.walk(folder, topdown=False)
+        for name in files
+        if name.endswith(".py") and root == folder
+    ]
+
+    json_file_path = os.path.join(folder, "theme_list.json")
+
+    try:
+        with open(json_file_path, "r", encoding="utf8") as json_file:
+            themes_from_url = [item["id"] for item in json.load(json_file)]
+    except FileNotFoundError:
+        themes_from_url = []
+
+    combined_themes = set(themes_from_files + themes_from_url)
+
+    return list(combined_themes)
+
+
+def select_theme(name):
+    selected_file = name + ".py"
+    full_path = os.path.join(folder, selected_file)
+
+    if not os.path.exists(full_path):
+        with open(config_file, "r", encoding="utf8") as json_file:
+            config_data = json.load(json_file)
+
+        config_data["theme"]["file"] = None
+        config_data["theme"]["class"] = name
+
+        with open(config_file, "w", encoding="utf8") as json_file:
+            json.dump(config_data, json_file, indent=2)
+        print(f"Theme {name} successfully selected, restart applio.")
+        gr.Info(f"Theme {name} successfully selected, restart applio.")
+        return
+
+    class_found = get_class(full_path)
+    if class_found:
+        with open(config_file, "r", encoding="utf8") as json_file:
+            config_data = json.load(json_file)
+
+        config_data["theme"]["file"] = selected_file
+        config_data["theme"]["class"] = class_found
+
+        with open(config_file, "w", encoding="utf8") as json_file:
+            json.dump(config_data, json_file, indent=2)
+        print(f"Theme {name} successfully selected, restart applio.")
+        gr.Info(f"Theme {name} successfully selected, restart applio.")
+    else:
+        print(f"Theme {name} was not found.")
+
+
+def read_json():
+    try:
+        with open(config_file, "r", encoding="utf8") as json_file:
+            data = json.load(json_file)
+            selected_file = data["theme"]["file"]
+            class_name = data["theme"]["class"]
+
+            if selected_file is not None and class_name:
+                return class_name
+            elif selected_file == None and class_name:
+                return class_name
+            else:
+                return "ParityError/Interstellar"
+    except Exception as e:
+        print(f"Error reading config.json: {e}")
+        return "ParityError/Interstellar"
+
+
+def load_json():
+    try:
+        with open(config_file, "r", encoding="utf8") as json_file:
+            data = json.load(json_file)
+            selected_file = data["theme"]["file"]
+            class_name = data["theme"]["class"]
+
+            if selected_file is not None and class_name:
+                module = importlib.import_module(selected_file[:-3])
+                obtained_class = getattr(module, class_name)
+                instance = obtained_class()
+                print(f"Theme Loaded: {class_name}")
+                return instance
+            elif selected_file == None and class_name:
+                return class_name
+            else:
+                print("The theme is incorrect.")
+                return None
+    except Exception as e:
+        print(f"Error Loading: {str(e)}")
+        return None