a01ca93362
* logging sys to loguru: stage 1 complete * import loguru: stage 2 * logging -> loguru: stage 3 * support o1-preview and o1-mini * logging -> loguru stage 4 * update social helper * logging -> loguru: final stage * fix: console output * update translation matrix * fix: loguru argument error with proxy enabled (#1977) * relax llama index version * remove comment * Added some modules to support openrouter (#1975) * Added some modules for supporting openrouter model Added some modules for supporting openrouter model * Update config.py * Update .gitignore * Update bridge_openrouter.py * Not changed actually * Refactor logging in bridge_openrouter.py --------- Co-authored-by: binary-husky <qingxu.fu@outlook.com> * remove logging extra --------- Co-authored-by: Steven Moder <java20131114@gmail.com> Co-authored-by: Ren Lifei <2602264455@qq.com>
64 lines
2.3 KiB
Python
64 lines
2.3 KiB
Python
import os
|
||
import gradio as gr
|
||
from toolbox import get_conf, ProxyNetworkActivate
|
||
from loguru import logger
|
||
|
||
CODE_HIGHLIGHT, ADD_WAIFU, LAYOUT = get_conf("CODE_HIGHLIGHT", "ADD_WAIFU", "LAYOUT")
|
||
theme_dir = os.path.dirname(__file__)
|
||
|
||
|
||
def dynamic_set_theme(THEME):
|
||
set_theme = gr.themes.ThemeClass()
|
||
with ProxyNetworkActivate("Download_Gradio_Theme"):
|
||
logger.info("正在下载Gradio主题，请稍等。")
|
||
try:
|
||
if THEME.startswith("Huggingface-"):
|
||
THEME = THEME.lstrip("Huggingface-")
|
||
if THEME.startswith("huggingface-"):
|
||
THEME = THEME.lstrip("huggingface-")
|
||
set_theme = set_theme.from_hub(THEME.lower())
|
||
except:
|
||
logger.error("下载Gradio主题时出现异常。")
|
||
return set_theme
|
||
|
||
|
||
def adjust_theme():
|
||
try:
|
||
set_theme = gr.themes.ThemeClass()
|
||
with ProxyNetworkActivate("Download_Gradio_Theme"):
|
||
logger.info("正在下载Gradio主题，请稍等。")
|
||
try:
|
||
THEME = get_conf("THEME")
|
||
if THEME.startswith("Huggingface-"):
|
||
THEME = THEME.lstrip("Huggingface-")
|
||
if THEME.startswith("huggingface-"):
|
||
THEME = THEME.lstrip("huggingface-")
|
||
set_theme = set_theme.from_hub(THEME.lower())
|
||
except:
|
||
logger.error("下载Gradio主题时出现异常。")
|
||
|
||
from themes.common import get_common_html_javascript_code
|
||
js = get_common_html_javascript_code()
|
||
|
||
if not hasattr(gr, "RawTemplateResponse"):
|
||
gr.RawTemplateResponse = gr.routes.templates.TemplateResponse
|
||
gradio_original_template_fn = gr.RawTemplateResponse
|
||
|
||
def gradio_new_template_fn(*args, **kwargs):
|
||
res = gradio_original_template_fn(*args, **kwargs)
|
||
res.body = res.body.replace(b"</html>", f"{js}</html>".encode("utf8"))
|
||
res.init_headers()
|
||
return res
|
||
|
||
gr.routes.templates.TemplateResponse = (
|
||
gradio_new_template_fn # override gradio template
|
||
)
|
||
except Exception:
|
||
set_theme = None
|
||
logger.error("gradio版本较旧, 不能自定义字体和颜色。")
|
||
return set_theme
|
||
|
||
|
||
with open(os.path.join(theme_dir, "common.css"), "r", encoding="utf-8") as f:
|
||
advanced_css = f.read()
|