Update app.py

app.py

@@ -151,14 +151,26 @@ def util_ui():
                 items = gr.JSON(value=None, visible=False)
                 png2main = gr.Button('参数发送到文生图')
     return page, png2main, items, info, image
-
-def ui():
+    
+def load_javascript():
     head = ''
     with open('jsReloader.js', 'r', encoding='utf8') as j:
         head += f'<script>{j.read()}</script>\n'
     for f in sorted(os.listdir('./tagcomplete/javascript')):
         head += f'<script type="text/javascript" src="file=tagcomplete/javascript/{f}"></script>\n'
-    with gr.Blocks(title="NAI Client", head=head, theme=gr.themes.Base()) as website:
+
+    share = gr.routes.templates.TemplateResponse
+    def template_response(*args, **kwargs):
+        res = share(*args, **kwargs)
+        res.body = res.body.replace(b'</body>', f'{head}</body>'.encode("utf8"))
+        res.init_headers()
+        return res
+
+    gr.routes.templates.TemplateResponse = template_response
+    
+def ui():
+    load_javascript()
+    with gr.Blocks(title="NAI Client", theme=gr.themes.Base()) as website:
         with gr.Tabs():
             with gr.TabItem("文生图", elem_id="client_ui_main"):
                 _, paras = main_ui()