import os import gradio as gr from toolbox import get_conf, ProxyNetworkActivate 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"): print("正在下载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: print("下载Gradio主题时出现异常。") return set_theme def adjust_theme(): try: set_theme = gr.themes.ThemeClass() with ProxyNetworkActivate("Download_Gradio_Theme"): print("正在下载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: print("下载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"", f"{js}".encode("utf8")) res.init_headers() return res gr.routes.templates.TemplateResponse = ( gradio_new_template_fn # override gradio template ) except Exception: set_theme = None print("gradio版本较旧, 不能自定义字体和颜色。") return set_theme with open(os.path.join(theme_dir, "common.css"), "r", encoding="utf-8") as f: advanced_css = f.read()