mirror of
https://github.com/KoboldAI/KoboldAI-Client.git
synced 2025-06-05 21:59:24 +02:00
Add the ability to insert the story prompt into the art guide.
The art guide now optionally supports the sequence <|> If this exists in the art guide, <|> is replaced by the story summary. Otherwise, the art guide is appended to the summary as before. Update the default art guide to place the medium before the summary. Also update the art guide tooltip to include the new default.
This commit is contained in:
18
aiserver.py
18
aiserver.py
@@ -9492,7 +9492,7 @@ def UI_2_generate_image_from_story(data):
|
||||
koboldai_vars.generating_image = True
|
||||
eventlet.sleep(0)
|
||||
|
||||
art_guide = '{}'.format(koboldai_vars.img_gen_art_guide)
|
||||
art_guide = str(koboldai_vars.img_gen_art_guide)
|
||||
|
||||
if 'action_id' in data and (int(data['action_id']) in koboldai_vars.actions.actions or int(data['action_id']) == -1):
|
||||
action_id = int(data['action_id'])
|
||||
@@ -9526,13 +9526,21 @@ def UI_2_generate_image_from_story(data):
|
||||
|
||||
max_length = args.max_summary_length - len(koboldai_vars.summary_tokenizer.encode(art_guide))
|
||||
keys = [summarize(text, max_length=max_length)]
|
||||
logger.debug("Text from summarizer: {}".format(keys[0]))
|
||||
|
||||
prompt = ", ".join(keys)
|
||||
logger.debug("Text from summarizer: {}".format(prompt))
|
||||
|
||||
if art_guide:
|
||||
if '<|>' in art_guide:
|
||||
full_prompt = art_guide.replace('<|>', prompt)
|
||||
else:
|
||||
full_prompt = f"{prompt}, {art_guide}"
|
||||
else:
|
||||
full_prompt = prompt
|
||||
|
||||
generate_story_image(
|
||||
", ".join([part for part in [prompt, art_guide] if part]),
|
||||
full_prompt,
|
||||
file_prefix=f"action_{action_id}",
|
||||
display_prompt=prompt,
|
||||
display_prompt=full_prompt,
|
||||
log_data={"actionId": action_id},
|
||||
)
|
||||
|
||||
|
Reference in New Issue
Block a user