From f1739dd1846cc3cd8b2766e9de8c177158f128eb Mon Sep 17 00:00:00 2001
From: Henk <henk@henk.tech>
Date: Fri, 13 Jan 2023 13:04:54 +0100
Subject: [PATCH] Chatmode Regex

---
 aiserver.py |  6 ++++--
 utils.py    | 18 ++++++++++++++++++
 2 files changed, 22 insertions(+), 2 deletions(-)

diff --git a/aiserver.py b/aiserver.py
index 5415e06d..cab01077 100644
--- a/aiserver.py
+++ b/aiserver.py
@@ -6475,7 +6475,7 @@ def getnewcontent(txt):
 #==================================================================#
 def applyinputformatting(txt):
     # Add sentence spacing
-    if(koboldai_vars.frmtadsnsp):
+    if(koboldai_vars.frmtadsnsp and not koboldai_vars.chatmode):
         txt = utils.addsentencespacing(txt, koboldai_vars)
  
     return txt
@@ -6511,7 +6511,9 @@ def applyoutputformatting(txt, no_sentence_trimming=False, no_single_line=False)
 	# Single Line Mode
     if(koboldai_vars.singleline and not no_single_line):
         txt = utils.singlelineprocessing(txt, koboldai_vars)
-    
+ 	# Chat Mode Trimming
+    if(koboldai_vars.chatmode):
+        txt = utils.chatmodeprocessing(txt, koboldai_vars)   
     for sub in koboldai_vars.substitutions:
         if not sub["enabled"]:
             continue
diff --git a/utils.py b/utils.py
index b8baa786..22fc05a4 100644
--- a/utils.py
+++ b/utils.py
@@ -147,6 +147,24 @@ def singlelineprocessing(txt, koboldai_vars):
         txt = txt + "\n"
     return txt
 
+def chatmodeprocessing(txt, koboldai_vars):
+    chatregex = re.compile(r'%s:[.|\n|\W|\w]*'%koboldai_vars.chatname)
+    txt = chatregex.sub('', txt)
+    if(len(koboldai_vars.actions) > 0):
+        if(len(koboldai_vars.actions[-1]) > 0):
+            action = koboldai_vars.actions[-1]
+            lastchar = action[-1] if len(action) else ""
+        else:
+            # Last action is blank, this should never happen, but
+            # since it did let's bail out.
+            return txt
+    else:
+        action = koboldai_vars.prompt
+        lastchar = action[-1] if len(action) else ""
+    if(lastchar != "\n"):
+        txt = txt + "\n"
+    return txt
+
 #==================================================================#
 #  Cleans string for use in file name
 #==================================================================#