diff --git a/koboldcpp.py b/koboldcpp.py
index fb6282833..91f0b490c 100644
--- a/koboldcpp.py
+++ b/koboldcpp.py
@@ -2555,16 +2555,19 @@ class KcppServerRequestHandler(http.server.SimpleHTTPRequestHandler):
genbtnval = (parsed_dict['generate'][0] if 'generate' in parsed_dict else "")
gencommand = (genbtnval=="Generate" or genbtnval=="Send")
chatmode = int(parsed_dict['chatmode'][0]) if 'chatmode' in parsed_dict else 0
+ human_name = str(parsed_dict['human_name'][0]) if 'human_name' in parsed_dict else "User"
+ bot_name = str(parsed_dict['bot_name'][0]) if 'bot_name' in parsed_dict else "Assistant"
stops = []
prefix = ""
if chatmode:
ban_eos_token = False
+ prompt = prompt.replace("1HdNl1","\n")
if chatmsg:
- prompt += f"\nUser: {chatmsg}\nAssistant:"
+ prompt += f"\n{human_name}: {chatmsg}\n{bot_name}:"
else:
gencommand = False
- stops = ["\nUser:","\nAssistant:"]
- prefix = "[This is a chat conversation log between User and Assistant.]\n"
+ stops = [f"\n{human_name}:",f"\n{bot_name}:"]
+ prefix = f"[This is a chat conversation log between {human_name} and {bot_name}.]\n"
if modelbusy.locked():
status = "Model is currently busy, try again later."
@@ -2601,12 +2604,17 @@ class KcppServerRequestHandler(http.server.SimpleHTTPRequestHandler):
self.end_headers(content_type='text/html')
return
- bodycontent = f'''{"Chat Mode" if chatmode else "Story Mode"}
'''
+ bodycontent = f'''{"Chat Mode" if chatmode else "Story Mode"}
'''
if chatmode:
- tmp = prompt.strip().replace("\n","
")
- bodycontent += f'''
{"No History Yet. Talk to the AI." if prompt=="" else tmp}
+ oldconvo = prompt.strip().replace(f"{human_name}:",f"{human_name}:").replace(f"{bot_name}:",f"{bot_name}:").replace("\n","{newconvo if prompt=="" else oldconvo}
+ - (Be patient)''' else: bodycontent += f'''