optimize UI and user experience

app/config/config.py

@@ -1,6 +1,6 @@
 import os
 import socket
-import tomli
+import toml
 from loguru import logger
 
 root_dir = os.path.dirname(os.path.dirname(os.path.dirname(os.path.realpath(__file__))))
@@ -16,17 +16,17 @@ if not os.path.isfile(config_file):
 logger.info(f"load config from file: {config_file}")
 
 try:
-    with open(config_file, mode="rb") as fp:
-        _cfg = tomli.load(fp)
+    _cfg = toml.load(config_file)
 except Exception as e:
     logger.warning(f"load config failed: {str(e)}, try to load as utf-8-sig")
     with open(config_file, mode="r", encoding='utf-8-sig') as fp:
         _cfg_content = fp.read()
-        _cfg = tomli.loads(_cfg_content)
+        _cfg = toml.loads(_cfg_content)
 
 app = _cfg.get("app", {})
 whisper = _cfg.get("whisper", {})
 pexels = _cfg.get("pexels", {})
+ui = _cfg.get("ui", {})
 
 hostname = socket.gethostname()
 
@@ -47,9 +47,18 @@ ffmpeg_path = app.get("ffmpeg_path", "")
 if ffmpeg_path and os.path.isfile(ffmpeg_path):
     os.environ["IMAGEIO_FFMPEG_EXE"] = ffmpeg_path
 
+
 # __cfg = {
 #     "hostname": hostname,
 #     "listen_host": listen_host,
 #     "listen_port": listen_port,
 # }
 # logger.info(__cfg)
+
+
+def save_config():
+    with open(config_file, "w", encoding="utf-8") as f:
+        _cfg["app"] = app
+        _cfg["whisper"] = whisper
+        _cfg["pexels"] = pexels
+        f.write(toml.dumps(_cfg))

app/services/material.py

@@ -11,13 +11,14 @@ from app.models.schema import VideoAspect, VideoConcatMode, MaterialInfo
 from app.utils import utils
 
 requested_count = 0
-pexels_api_keys = config.app.get("pexels_api_keys")
-if not pexels_api_keys:
-    raise ValueError(
-        f"\n\n##### pexels_api_keys is not set #####\n\nPlease set it in the config.toml file: {config.config_file}\n\n{utils.to_json(config.app)}")
 
 
 def round_robin_api_key():
+    pexels_api_keys = config.app.get("pexels_api_keys")
+    if not pexels_api_keys:
+        raise ValueError(
+            f"\n\n##### pexels_api_keys is not set #####\n\nPlease set it in the config.toml file: {config.config_file}\n\n{utils.to_json(config.app)}")
+
     # if only one key is provided, return it
     if isinstance(pexels_api_keys, str):
         return pexels_api_keys

app/utils/utils.py

@@ -1,3 +1,4 @@
+import locale
 import os
 import platform
 import threading
@@ -174,3 +175,25 @@ def split_string_by_punctuations(s):
 def md5(text):
     import hashlib
     return hashlib.md5(text.encode('utf-8')).hexdigest()
+
+
+def get_system_locale():
+    try:
+        loc = locale.getdefaultlocale()
+        # zh_CN, zh_TW return zh
+        # en_US, en_GB return en
+        language_code = loc[0].split("_")[0]
+        return language_code
+    except Exception as e:
+        return "en"
+
+
+def load_locales(i18n_dir):
+    _locales = {}
+    for root, dirs, files in os.walk(i18n_dir):
+        for file in files:
+            if file.endswith(".json"):
+                lang = file.split(".")[0]
+                with open(os.path.join(root, file), "r", encoding="utf-8") as f:
+                    _locales[lang] = json.loads(f.read())
+    return _locales