rixty
/
KoboldAI-Client
mirror of https://github.com/KoboldAI/KoboldAI-Client.git
1
1
Fork 0
Code Issues Projects Releases Wiki Activity

Merge pull request #27 from UWUplus/escape 

HTML escape user prompt.
This commit is contained in:
KoboldAI Dev 2021-06-02 15:40:17 -04:00 committed by GitHub
parent 1e95f7e1a5 e9b62cd3ae
commit 6a9587b61d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
aiserver.py
View File

@ -1091,7 +1091,7 @@ def applyoutputformatting(txt):
# Sends the current story content to the Game Screen # Sends the current story content to the Game Screen
#==================================================================# #==================================================================#
def refresh_story(): def refresh_story():
text_parts = ['<chunk n="0" id="n0">', vars.prompt, '</chunk>'] text_parts = ['<chunk n="0" id="n0">', html.escape(vars.prompt), '</chunk>']
for idx, item in enumerate(vars.actions, start=1): for idx, item in enumerate(vars.actions, start=1):
text_parts.extend(('<chunk n="', str(idx), '" id="n', str(idx), '">', html.escape(item), '</chunk>')) text_parts.extend(('<chunk n="', str(idx), '" id="n', str(idx), '">', html.escape(item), '</chunk>'))
emit('from_server', {'cmd': 'updatescreen', 'data': formatforhtml(''.join(text_parts))}) emit('from_server', {'cmd': 'updatescreen', 'data': formatforhtml(''.join(text_parts))})