Spaces:
Running
Running
File size: 4,054 Bytes
55adc26 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 |
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
|