tts-service / tabs /plugins /plugins_core.py
jlopez00's picture
Upload folder using huggingface_hub
bbf5262 verified
import json
import os
import subprocess
import sys
from assets.i18n.i18n import I18nAuto
from tabs.settings.sections.restart import restart_applio
i18n = I18nAuto()
now_dir = os.getcwd()
sys.path.append(now_dir)
plugins_path = os.path.join(now_dir, "tabs", "plugins", "installed")
if not os.path.exists(plugins_path):
os.makedirs(plugins_path)
json_file_path = os.path.join(now_dir, "assets", "config.json")
current_folders = os.listdir(plugins_path)
def get_existing_folders():
if os.path.exists(json_file_path):
with open(json_file_path) as file:
config = json.load(file)
return config["plugins"]
else:
return []
def save_existing_folders(existing_folders):
with open(json_file_path) as file:
config = json.load(file)
config["plugins"] = existing_folders
with open(json_file_path, "w") as file:
json.dump(config, file, indent=2)
def check_new_folders():
existing_folders = get_existing_folders()
new_folders = set(current_folders) - set(existing_folders)
save_existing_folders(current_folders)
if new_folders:
for new_folder in new_folders:
complete_path = os.path.join(plugins_path, new_folder)
print(f"New plugin {new_folder} found, installing it...")
if os.path.exists(os.path.join(complete_path, "requirements.txt")):
if os.name == "nt":
subprocess.run(
[
os.path.join("env", "python.exe"),
"-m",
"pip",
"install",
"-r",
os.path.join(complete_path, "requirements.txt"),
]
)
else:
subprocess.run(
[
"python",
"-m",
"pip",
"install",
"-r",
os.path.join(complete_path, "requirements.txt"),
]
)
else:
print("No requirements.txt file found in the plugin folder.")
print("Plugins checked and installed! Restarting applio to apply the changes.")
restart_applio()