Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-18 16:54:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

prompt enhance rewrite lists

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2025-03-31 10:53:35 -04:00
parent ff6d5e6e88
commit 5e79cff7a3
1 changed files with 14 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/prompt_enhance.py
+14 -1
View File
@@ -139,8 +139,15 @@ class Script(scripts.Script):
shared.log.debug('Prompt enhance: model unloaded')
def clean(self, response):
response = response.replace('"', '').replace("'", "").replace('“', '').replace('”', '').replace('**', '').replace('\n\n', '\n').replace(' ', ' ')
# remove special characters
response = response.replace('"', '').replace("'", "").replace('“', '').replace('”', '').replace('**', '').replace('\n\n', '\n').replace(' ', ' ').replace('...', '.')
# remove comments between brackets
response = re.sub(r'<.*?>', '', response)
response = re.sub(r'\[.*?\]', '', response)
response = re.sub(r'\/.*?\/', '', response)
# remove llm commentary
removed = ''
if response.startswith('Prompt'):
removed, response = response.split('Prompt', maxsplit=2)
@@ -150,6 +157,12 @@ class Script(scripts.Script):
response, removed = response.split('---', maxsplit=2)
if len(removed) > 0:
debug(f'Prompt enhance: max={self.options.max_delim_index} removed="{removed}"')
# remove bullets and lists
lines = response.splitlines()
filtered = [re.sub(r'^(\s*[-*]|\s*\d+)\s+', '', line) for line in lines]
response = '\n'.join(filtered)
response = response.strip()
return response