Преглед изворни кода

Add **bold** support in chat mode

oobabooga пре 3 година
родитељ
комит
dfbca86533
1 измењених фајлова са 2 додато и 2 уклоњено
  1. 2 2
      modules/html_generator.py

+ 2 - 2
modules/html_generator.py

@@ -242,8 +242,8 @@ def generate_chat_html(history, name1, name2, character):
 
     for i,_row in enumerate(history[::-1]):
         row = _row.copy()
-        row[0] = re.sub(r"[\\]*\*", r"*", row[0])
-        row[1] = re.sub(r"[\\]*\*", r"*", row[1])
+        row[0] = re.sub(r"(\*\*)([^\*\n]*)(\*\*)", r"<b>\2</b>", row[0])
+        row[1] = re.sub(r"(\*\*)([^\*\n]*)(\*\*)", r"<b>\2</b>", row[1])
         row[0] = re.sub(r"(\*)([^\*\n]*)(\*)", r"<em>\2</em>", row[0])
         row[1] = re.sub(r"(\*)([^\*\n]*)(\*)", r"<em>\2</em>", row[1])
         p = '\n'.join([f"<p>{x}</p>" for x in row[1].split('\n')])