Fix clear history

This commit is contained in:
oobabooga 2023-02-26 23:58:04 -03:00
parent 7a776ccf87
commit f24b6e78a3

View File

@ -216,11 +216,16 @@ def clear_html():
def clear_chat_log(name1, name2): def clear_chat_log(name1, name2):
if shared.character != 'None': if shared.character != 'None':
found = False
for i in range(len(shared.history['internal'])): for i in range(len(shared.history['internal'])):
if '<|BEGIN-VISIBLE-CHAT|>' in shared.history['internal'][i][0]: if '<|BEGIN-VISIBLE-CHAT|>' in shared.history['internal'][i][0]:
shared.history['visible'] = [['', apply_extensions(shared.history['internal'][i][1], "output")]] shared.history['visible'] = [['', apply_extensions(shared.history['internal'][i][1], "output")]]
shared.history['internal'] = shared.history['internal'][:i+1] shared.history['internal'] = shared.history['internal'][:i+1]
found = True
break break
if not found:
shared.history['visible'] = []
shared.history['internal'] = []
else: else:
shared.history['internal'] = [] shared.history['internal'] = []
shared.history['visible'] = [] shared.history['visible'] = []