Update conversation.py

conversation.py

@@ -203,12 +203,29 @@ class Conversation:
                 #     f'<img src="data:image/png;base64,{img_b64_str}" alt="{alt_str}" />'
                 # )
                 img_str_list.append(img_str)
-
+            
+            if ('\[' in msg["content"] and '\]' in msg["content"]) or ('\(' in msg["content"] and '\)' in msg["content"]):
+                # 行内公式 or 行间公式
+                content = msg["content"].replace('\[', '$$').replace('\]', '$$').replace('\(', '$$').replace('\)', '$$')
+                content = content.split('$$')
+                for i in range(len(content)):
+                    if i % 2:
+                        content[i] = content[i].strip()
+                content = '$$'.join(content)
+                print('content:', content)
+                # content = (
+                #     r"<span>" + content + r"</span>"
+                #     r"<script type='text/javascript'>"
+                #     r"MathJax.typesetPromise();"
+                #     r"</script>"
+                # )
+            else:
+                content = msg["content"]
             if msg["role"] == self.USER:
-                msg_str = " ".join(img_str_list) + msg["content"]
+                msg_str = " ".join(img_str_list) + content
                 ret.append([msg_str, None])
             else:
-                msg_str = msg["content"] + " ".join(img_str_list)
+                msg_str = content + " ".join(img_str_list)
                 ret[-1][-1] = msg_str
         return ret
 
@@ -304,3 +321,4 @@ class Conversation:
         return filename
 
 
+