Update bin_public/app/overwrites.py

bin_public/app/overwrites.py

@@ -4,7 +4,7 @@ from llama_index import Prompt
 from typing import List, Tuple
 import mdtex2html
 
-from llama_func import *
+from bin_public.config.presets import *
 
 
 def compact_text_chunks(self, prompt: Prompt, text_chunks: List[str]) -> List[str]:
@@ -36,3 +36,21 @@ def postprocess(
             None if response is None else mdtex2html.convert(response, extensions=['fenced_code','codehilite','tables']),
         )
     return y
+
+
+with open(customJS_Path, "r", encoding="utf-8") as f, open(Kelpy_Codos_Path, "r", encoding="utf-8") as f2:
+    customJS = f.read()
+    kelpyCodos = f2.read()
+
+def reload_javascript():
+    print("Reloading javascript...")
+    js = f'<script>{customJS}</script><script>{kelpyCodos}</script>'
+    def template_response(*args, **kwargs):
+        res = GradioTemplateResponseOriginal(*args, **kwargs)
+        res.body = res.body.replace(b'</html>', f'{js}</html>'.encode("utf8"))
+        res.init_headers()
+        return res
+
+    gr.routes.templates.TemplateResponse = template_response
+
+GradioTemplateResponseOriginal = gr.routes.templates.TemplateResponse