| 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 |
|
|
