From 625d81f495e57961c0e63fa8b4c40835e0f33b94 Mon Sep 17 00:00:00 2001 From: Brian O'Connor Date: Sun, 9 Apr 2023 21:20:21 -0400 Subject: [PATCH] Update character log logic (#977) * When logs are cleared, save the cleared log over the old log files * Generate a log file when a character is loaded the first time --- modules/chat.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/modules/chat.py b/modules/chat.py index 6a1f7ad1..df39a58b 100644 --- a/modules/chat.py +++ b/modules/chat.py @@ -284,6 +284,9 @@ def clear_chat_log(name1, name2, greeting, mode): if greeting != '': shared.history['internal'] += [['<|BEGIN-VISIBLE-CHAT|>', greeting]] shared.history['visible'] += [['', apply_extensions(greeting, "output")]] + + # Save cleared logs + save_history(timestamp=False) return chat_html_wrapper(shared.history['visible'], name1, name2, mode) @@ -434,9 +437,14 @@ def load_character(character, name1, name2, mode): if Path(f'logs/{shared.character}_persistent.json').exists(): load_history(open(Path(f'logs/{shared.character}_persistent.json'), 'rb').read(), name1, name2) - elif greeting != "": - shared.history['internal'] += [['<|BEGIN-VISIBLE-CHAT|>', greeting]] - shared.history['visible'] += [['', apply_extensions(greeting, "output")]] + else: + # Insert greeting if it exists + if greeting != "": + shared.history['internal'] += [['<|BEGIN-VISIBLE-CHAT|>', greeting]] + shared.history['visible'] += [['', apply_extensions(greeting, "output")]] + + # Create .json log files since they don't already exist + save_history(timestamp=False) return name1, name2, picture, greeting, context, end_of_turn, chat_html_wrapper(shared.history['visible'], name1, name2, mode, reset_cache=True)