mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Save chat history with name/date in filename

Browse source 
closes #50
This commit is contained in:
oobabooga 2023-02-03 09:02:35 -03:00
parent 3af3ffeb90
commit 4e4cd67223
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
server.py
View file

@ -9,6 +9,7 @@ import io
import base64 import base64
import sys import sys
import os import os
from datetime import datetime
from pathlib import Path from pathlib import Path
from PIL import Image from PIL import Image
import copy import copy
@ -563,11 +564,12 @@ if args.chat or args.cai_chat:
return _history return _history
def save_history(): def save_history():
fname = f"{character or ''}{'_' if character else ''}{datetime.now().strftime('%Y%m%d-%H:%M:%S')}.json"
if not Path('logs').exists(): if not Path('logs').exists():
Path('logs').mkdir() Path('logs').mkdir()
with open(Path('logs/conversation.json'), 'w') as f: with open(Path(f'logs/{fname}'), 'w') as f:
f.write(json.dumps({'data': history['internal'], 'data_visible': history['visible']})) f.write(json.dumps({'data': history['internal'], 'data_visible': history['visible']}))
return Path('logs/conversation.json') return Path(f'logs/{fname}')
def upload_history(file, name1, name2): def upload_history(file, name1, name2):
global history global history