diff --git a/themes/contrast.py b/themes/contrast.py
index d407d92..bf48808 100644
--- a/themes/contrast.py
+++ b/themes/contrast.py
@@ -1,6 +1,8 @@
+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():
@@ -57,7 +59,7 @@ def adjust_theme():
button_cancel_text_color_dark="white",
)
- with open('themes/common.js', 'r', encoding='utf8') as f:
+ with open(os.path.join(theme_dir, 'common.js'), 'r', encoding='utf8') as f:
js = f""
# 添加一个萌萌的看板娘
@@ -67,7 +69,9 @@ def adjust_theme():
"""
- gradio_original_template_fn = gr.routes.templates.TemplateResponse
+ 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'