initial upload attempt

initial upload attempt

system/gradio_pages/themes/loadThemes.py

@@ -0,0 +1,114 @@
+import os
+import json
+import sys
+import importlib
+
+# Hard-coded paths
+this_folder = os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "..", ".."))
+config_file = os.path.join(this_folder, "confignew.json")
+gradio_pages_folder = os.path.join(this_folder, "system", "gradio_pages", "themes")
+theme_list_file = os.path.join(gradio_pages_folder, "theme_list.json")
+themes_file_path = os.path.join(gradio_pages_folder, "themes.py")
+
+# Add gradio_pages folder to the system path
+sys.path.append(gradio_pages_folder)
+
+def get_class(filename):
+    with open(filename, "r", encoding="utf8") as file:
+        for line in file:
+            if "class " in line:
+                return line.split("class ")[1].split(":")[0].split("(")[0].strip()
+    return None
+
+
+def get_list():
+    themes_from_files = [
+        os.path.splitext(name)[0]
+        for name in os.listdir(gradio_pages_folder)
+        if name.endswith(".py")
+    ]
+    try:
+        with open(theme_list_file, "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 = sorted(set(themes_from_files + themes_from_url))
+    return list(combined_themes)
+
+
+def select_theme(name):
+    selected_file = name + ".py"
+    full_path = os.path.join(gradio_pages_folder, selected_file)
+
+    try:
+        with open(config_file, "r", encoding="utf8") as json_file:
+            config_data = json.load(json_file)
+
+        if not os.path.exists(full_path):
+            # If the theme is from theme_list.json, update only the class field
+            config_data["theme"]["file"] = None
+            config_data["theme"]["class"] = name
+            message = f"Theme {name} successfully selected, restart AllTalk."
+        else:
+            # If the theme file exists locally, update both file and class fields
+            class_found = get_class(full_path)
+            if class_found:
+                config_data["theme"]["file"] = selected_file
+                config_data["theme"]["class"] = class_found
+                message = f"Theme {name} successfully selected, restart AllTalk."
+            else:
+                message = f"Theme {name} was not found."
+
+        with open(config_file, "w", encoding="utf8") as json_file:
+            json.dump(config_data, json_file, indent=2)
+        
+        print(message)
+        return message
+
+    except Exception as e:
+        print(f"[AllTalk TTS] Error selecting theme {name}: {str(e)}")
+        return message
+
+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 and class_name:
+                return class_name
+            elif not selected_file and class_name:
+                return class_name
+            else:
+                return "gradio/base"
+    except Exception as e:
+        print(f"[AllTalk TTS] Error reading config.json: {e}")
+        return "gradio/base"
+
+
+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 and class_name:
+                module_name = os.path.splitext(selected_file)[0]
+                module_spec = importlib.util.spec_from_file_location(module_name, themes_file_path)
+                module = importlib.util.module_from_spec(module_spec)
+                module_spec.loader.exec_module(module)
+                obtained_class = getattr(module, class_name)
+                instance = obtained_class()
+                print(f"Theme Loaded: {class_name}")
+                return instance
+            elif not selected_file 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