File size: 2,186 Bytes
f8946c1
15f14f5
 
f8946c1
47289f8
 
15f14f5
f8946c1
47289f8
971ac20
 
47289f8
 
 
 
 
 
971ac20
 
 
47289f8
f8946c1
 
 
47289f8
 
 
 
 
 
 
f8946c1
 
d0703ef
 
 
47289f8
15f14f5
 
47289f8
f8946c1
 
47289f8
f8946c1
 
47289f8
 
 
 
 
f8946c1
 
47289f8
 
f8946c1
 
47289f8
 
f8946c1
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
import logging
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"):
        logging.info("正在下载Gradio主题,请稍等。")
        if THEME.startswith("Huggingface-"):
            THEME = THEME.lstrip("Huggingface-")
        if THEME.startswith("huggingface-"):
            THEME = THEME.lstrip("huggingface-")
        set_theme = set_theme.from_hub(THEME.lower())
    return set_theme


def adjust_theme():
    try:
        set_theme = gr.themes.ThemeClass()
        with ProxyNetworkActivate("Download_Gradio_Theme"):
            logging.info("正在下载Gradio主题,请稍等。")
            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())

        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
        from toolbox import trimmed_format_exc

        logging.error("gradio版本较旧, 不能自定义字体和颜色:", trimmed_format_exc())
    return set_theme


with open(os.path.join(theme_dir, "common.css"), "r", encoding="utf-8") as f:
    advanced_css = f.read()