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

Merge pull request #351 from one-some/ui2-imggen-horde-polish

Browse Source 
Image error handling
This commit is contained in:
ebolam
2022-12-20 07:28:04 -05:00
committed by GitHub
parent 2c47982f3b 149ec6b49d
commit ca9397f670
1 changed files with 14 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
aiserver.py
View File

@@ -9572,6 +9572,8 @@ def generate_story_image(
# This function is a wrapper around generate_image() that integrates the # This function is a wrapper around generate_image() that integrates the
# result with the story (read: puts it in the corner of the screen). # result with the story (read: puts it in the corner of the screen).
log_data = log_data or {}
if not display_prompt: if not display_prompt:
display_prompt = prompt display_prompt = prompt
koboldai_vars.picture_prompt = display_prompt koboldai_vars.picture_prompt = display_prompt
@@ -9596,7 +9598,10 @@ def generate_story_image(
log_image_generation(prompt, display_prompt, file_name, generation_type, log_data) log_image_generation(prompt, display_prompt, file_name, generation_type, log_data)
#let's also add this data to the action so we know where the latest picture is at #let's also add this data to the action so we know where the latest picture is at
logger.info("setting picture filename") logger.info("setting picture filename")
try:
koboldai_vars.actions.set_picture(int(log_data['actionId']), file_name, prompt) koboldai_vars.actions.set_picture(int(log_data['actionId']), file_name, prompt)
except KeyError:
pass
logger.debug("Time to Generate Image {}".format(time.time()-start_time)) logger.debug("Time to Generate Image {}".format(time.time()-start_time))
@@ -9689,6 +9694,14 @@ def text2img_horde(prompt: str) -> Optional[Image.Image]:
id_req = requests.post("https://stablehorde.net/api/v2/generate/async", json=final_submit_dict, headers=cluster_headers) id_req = requests.post("https://stablehorde.net/api/v2/generate/async", json=final_submit_dict, headers=cluster_headers)
if not id_req.ok: if not id_req.ok:
if id_req.status_code == 403:
show_error_notification(
"Stable Horde failure",
"Stable Horde is currently not accepting anonymous requuests. " \
"Try again in a few minutes or register for priority access at https://stablehorde.net",
do_log=True
)
return None
logger.error(f"HTTP {id_req.status_code}, expected OK-ish") logger.error(f"HTTP {id_req.status_code}, expected OK-ish")
logger.error(id_req.text) logger.error(id_req.text)
logger.error(f"Response headers: {id_req.headers}") logger.error(f"Response headers: {id_req.headers}")