diff --git a/modules/chat.py b/modules/chat.py index 061177d2..1a43cf3d 100644 --- a/modules/chat.py +++ b/modules/chat.py @@ -33,12 +33,14 @@ def generate_chat_prompt(user_input, max_new_tokens, name1, name2, context, chat i = len(shared.history['internal'])-1 while i >= 0 and len(encode(''.join(rows), max_new_tokens)[0]) < max_length: rows.insert(1, f"{name2}: {shared.history['internal'][i][1].strip()}\n") - if not (shared.history['internal'][i][0] == '<|BEGIN-VISIBLE-CHAT|>'): - rows.insert(1, f"{name1}: {shared.history['internal'][i][0].strip()}\n") + prev_user_input = shared.history['internal'][i][0] + if len(prev_user_input) > 0 and prev_user_input != '<|BEGIN-VISIBLE-CHAT|>': + rows.insert(1, f"{name1}: {prev_user_input.strip()}\n") i -= 1 if not impersonate: - rows.append(f"{name1}: {user_input}\n") + if len(user_input) > 0: + rows.append(f"{name1}: {user_input}\n") rows.append(apply_extensions(f"{name2}:", "bot_prefix")) limit = 3 else: diff --git a/modules/html_generator.py b/modules/html_generator.py index 940d5486..ff18c913 100644 --- a/modules/html_generator.py +++ b/modules/html_generator.py @@ -119,13 +119,13 @@ def load_html_image(paths): def generate_chat_html(history, name1, name2, character): output = f'
' - + img_bot = load_html_image([f"characters/{character}.{ext}" for ext in ['png', 'jpg', 'jpeg']] + ["img_bot.png","img_bot.jpg","img_bot.jpeg"]) img_me = load_html_image(["img_me.png", "img_me.jpg", "img_me.jpeg"]) for i,_row in enumerate(history[::-1]): row = [convert_to_markdown(entry) for entry in _row] - + output += f"""
@@ -142,22 +142,24 @@ def generate_chat_html(history, name1, name2, character):
""" - if not (i == len(history)-1 and len(row[0]) == 0): - output += f""" -
-
- {img_me} -
-
-
- {name1} -
-
- {row[0]} -
-
+ if len(row[0]) == 0: # don't display empty user messages + continue + + output += f""" +
+
+ {img_me} +
+
+
+ {name1}
- """ +
+ {row[0]} +
+
+
+ """ output += "
" return output