123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115 |
- 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"<script>{f.read()}</script>"
- # 添加一个萌萌的看板娘
- if ADD_WAIFU:
- js += """
- <script src="file=docs/waifu_plugin/jquery.min.js"></script>
- <script src="file=docs/waifu_plugin/jquery-ui.min.js"></script>
- <script src="file=docs/waifu_plugin/autoload.js"></script>
- """
- with open(os.path.join(theme_dir, "green.js"), "r", encoding="utf8") as f:
- js += f"<script>{f.read()}</script>"
- 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:
- 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()
|