diff --git a/modules/chat.py b/modules/chat.py index 356119a2..84c60273 100644 --- a/modules/chat.py +++ b/modules/chat.py @@ -159,7 +159,7 @@ def cai_chatbot_wrapper(text, tokens, do_sample, max_new_tokens, temperature, to yield generate_chat_html(_history, name1, name2, shared.character) def regenerate_wrapper(text, tokens, do_sample, max_new_tokens, temperature, top_p, typical_p, repetition_penalty, top_k, min_length, no_repeat_ngram_size, num_beams, penalty_alpha, length_penalty, early_stopping, name1, name2, context, check, chat_prompt_size, picture=None): - if shared.character is not None and len(shared.history['visible']) == 1: + if shared.character != 'None' and len(shared.history['visible']) == 1: if shared.args.cai_chat: yield generate_chat_html(shared.history['visible'], name1, name2, shared.character) else: @@ -265,10 +265,11 @@ def tokenize_dialogue(dialogue, name1, name2): return _history def save_history(timestamp=True): + prefix = '' if shared.character == 'None' else f"{shared.character}_" if timestamp: - fname = f"{shared.character or ''}{'_' if shared.character else ''}{datetime.now().strftime('%Y%m%d-%H%M%S')}.json" + fname = f"{prefix}{datetime.now().strftime('%Y%m%d-%H%M%S')}.json" else: - fname = f"{shared.character or ''}{'_' if shared.character else ''}persistent.json" + fname = f"{prefix}persistent.json" if not Path('logs').exists(): Path('logs').mkdir() with open(Path(f'logs/{fname}'), 'w') as f: diff --git a/server.py b/server.py index ee5c9372..b5418bfe 100644 --- a/server.py +++ b/server.py @@ -186,10 +186,9 @@ else: default_text = shared.settings['prompt'] if shared.args.chat or shared.args.cai_chat: - if Path(f'logs/persistent.json').exists(): - chat.load_history(open(Path(f'logs/persistent.json'), 'rb').read(), shared.settings[f'name1{suffix}'], shared.settings[f'name2{suffix}']) - with gr.Blocks(css=ui.css+ui.chat_css, analytics_enabled=False) as interface: + if Path(f'logs/persistent.json').exists(): + chat.load_history(open(Path(f'logs/persistent.json'), 'rb').read(), shared.settings[f'name1{suffix}'], shared.settings[f'name2{suffix}']) if shared.args.cai_chat: display = gr.HTML(value=generate_chat_html(shared.history['visible'], shared.settings[f'name1{suffix}'], shared.settings[f'name2{suffix}'], shared.character)) else: