Update bin_public/app/overwrites.py

bin_public/app/overwrites.py

@@ -6,6 +6,7 @@ import mdtex2html
 import logging
 
 from bin_public.config.presets import *
+from bin_public.app.llama_func import *
 
 
 def compact_text_chunks(self, prompt: Prompt, text_chunks: List[str]) -> List[str]:
@@ -27,18 +28,16 @@ def postprocess(
     Returns:
         List of tuples representing the message and response. Each message and response will be a string of HTML.
     """
-    if y is None:
+    if y is None or y == []:
         return []
-    for i, (message, response) in enumerate(y):
-        y[i] = (
-            # None if message is None else markdown.markdown(message),
-            # None if response is None else markdown.markdown(response),
-            None if message is None else message,
-            None if response is None else mdtex2html.convert(response, extensions=['fenced_code','codehilite','tables']),
-        )
+    user, bot = y[-1]
+    if not detect_converted_mark(user):
+        user = convert_asis(user)
+    if not detect_converted_mark(bot):
+        bot = convert_mdtext(bot)
+    y[-1] = (user, bot)
     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()