|
@@ -5,9 +5,8 @@ import importlib.metadata
|
|
|
import pkgutil
|
|
|
import chromadb
|
|
|
from chromadb import Settings
|
|
|
-from base64 import b64encode
|
|
|
from bs4 import BeautifulSoup
|
|
|
-from typing import TypeVar, Generic, Union
|
|
|
+from typing import TypeVar, Generic
|
|
|
from pydantic import BaseModel
|
|
|
from typing import Optional
|
|
|
|
|
@@ -19,7 +18,6 @@ import markdown
|
|
|
import requests
|
|
|
import shutil
|
|
|
|
|
|
-from secrets import token_bytes
|
|
|
from constants import ERROR_MESSAGES
|
|
|
|
|
|
####################################
|
|
@@ -768,12 +766,14 @@ class BannerModel(BaseModel):
|
|
|
dismissible: bool
|
|
|
timestamp: int
|
|
|
|
|
|
+try:
|
|
|
+ banners = json.loads(os.environ.get("WEBUI_BANNERS", "[]"))
|
|
|
+ banners = [BannerModel(**banner) for banner in banners]
|
|
|
+except Exception as e:
|
|
|
+ print(f"Error loading WEBUI_BANNERS: {e}")
|
|
|
+ banners = []
|
|
|
|
|
|
-WEBUI_BANNERS = PersistentConfig(
|
|
|
- "WEBUI_BANNERS",
|
|
|
- "ui.banners",
|
|
|
- [BannerModel(**banner) for banner in json.loads("[]")],
|
|
|
-)
|
|
|
+WEBUI_BANNERS = PersistentConfig("WEBUI_BANNERS", "ui.banners", banners)
|
|
|
|
|
|
|
|
|
SHOW_ADMIN_DETAILS = PersistentConfig(
|