import os import gradio as gr from toolbox import get_conf CODE_HIGHLIGHT, ADD_WAIFU, LAYOUT = get_conf("CODE_HIGHLIGHT", "ADD_WAIFU", "LAYOUT") theme_dir = os.path.dirname(__file__) def adjust_theme(): try: set_theme = gr.themes.Soft( primary_hue=gr.themes.Color( c50="#EBFAF2", c100="#CFF3E1", c200="#A8EAC8", c300="#77DEA9", c400="#3FD086", c500="#02C160", c600="#06AE56", c700="#05974E", c800="#057F45", c900="#04673D", c950="#2E5541", name="small_and_beautiful", ), secondary_hue=gr.themes.Color( c50="#576b95", c100="#576b95", c200="#576b95", c300="#576b95", c400="#576b95", c500="#576b95", c600="#576b95", c700="#576b95", c800="#576b95", c900="#576b95", c950="#576b95", ), neutral_hue=gr.themes.Color( name="gray", c50="#f6f7f8", # c100="#f3f4f6", c100="#F2F2F2", c200="#e5e7eb", c300="#d1d5db", c400="#B2B2B2", c500="#808080", c600="#636363", c700="#515151", c800="#393939", # c900="#272727", c900="#2B2B2B", c950="#171717", ), radius_size=gr.themes.sizes.radius_sm, ).set( button_primary_background_fill="*primary_500", button_primary_background_fill_dark="*primary_600", button_primary_background_fill_hover="*primary_400", button_primary_border_color="*primary_500", button_primary_border_color_dark="*primary_600", button_primary_text_color="white", button_primary_text_color_dark="white", button_secondary_background_fill="*neutral_100", button_secondary_background_fill_hover="*neutral_50", button_secondary_background_fill_dark="*neutral_900", button_secondary_text_color="*neutral_800", button_secondary_text_color_dark="white", background_fill_primary="*neutral_50", background_fill_primary_dark="#1F1F1F", block_title_text_color="*primary_500", block_title_background_fill_dark="*primary_900", block_label_background_fill_dark="*primary_900", input_background_fill="#F6F6F6", chatbot_code_background_color="*neutral_950", chatbot_code_background_color_dark="*neutral_950", ) with open(os.path.join(theme_dir, "common.js"), "r", encoding="utf8") as f: js = f"" # 添加一个萌萌的看板娘 if ADD_WAIFU: js += """ """ with open(os.path.join(theme_dir, "green.js"), "r", encoding="utf8") as f: js += f"" 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: set_theme = None print("gradio版本较旧, 不能自定义字体和颜色") return set_theme with open(os.path.join(theme_dir, "green.css"), "r", encoding="utf-8") as f: advanced_css = f.read() with open(os.path.join(theme_dir, "common.css"), "r", encoding="utf-8") as f: advanced_css += f.read()