rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Random Story Implemented

Browse Source
This commit is contained in:
ebolam
2022-08-14 21:15:44 -04:00
parent 43c8fd7e25
commit c3a79002c4
5 changed files with 69 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
aiserver.py
View File

@@ -6320,7 +6320,12 @@ def UI_2_Set_Selected_Text(data):
@socketio.on('Use Option Text')
def UI_2_Set_Selected_Text(data):
print("Using Option Text: {}".format(data))
koboldai_vars.actions.use_option(int(data['option']), action_step=int(data['chunk']))
if koboldai_vars.prompt == "":
print(koboldai_vars.actions.get_current_options())
koboldai_vars.prompt = koboldai_vars.actions.get_current_options()[int(data['option'])]['text']
koboldai_vars.actions.clear_unused_options()
else:
koboldai_vars.actions.use_option(int(data['option']), action_step=int(data['chunk']))
#==================================================================#
@@ -6328,10 +6333,18 @@ def UI_2_Set_Selected_Text(data):
#==================================================================#
@socketio.on('submit')
def UI_2_submit(data):
koboldai_vars.actions.clear_unused_options()
koboldai_vars.lua_koboldbridge.feedback = None
koboldai_vars.recentrng = koboldai_vars.recentrngm = None
actionsubmit(data['data'], actionmode=koboldai_vars.actionmode)
print(data)
if not koboldai_vars.noai and data['theme'] != "":
memory = koboldai_vars.memory
koboldai_vars.memory = "{}\n\nYou generate the following {} story concept :".format(koboldai_vars.memory, data['theme'])
koboldai_vars.lua_koboldbridge.feedback = None
actionsubmit("", force_submit=True, force_prompt_gen=True)
koboldai_vars.memory = memory
else:
koboldai_vars.actions.clear_unused_options()
koboldai_vars.lua_koboldbridge.feedback = None
koboldai_vars.recentrng = koboldai_vars.recentrngm = None
actionsubmit(data['data'], actionmode=koboldai_vars.actionmode)
#==================================================================#
# Event triggered when user clicks the submit button