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

fix processing api

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2026-08-17 18:07:06 +02:00
parent aa8cd9980e
commit e23a70af34
2 changed files with 14 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
CHANGELOG.md
+1
View File
@@ -78,6 +78,7 @@
- metadata: image metadata handle correct image index
- gguf: transformer loader
- scripts: mixture-of-diffusers and mixture-tiling update to use igwn-segments
- api: process
## Update for 2026-08-07
modules/api/process.py
+13 -2
View File
@@ -1,6 +1,7 @@
from threading import Lock
from pydantic import BaseModel, Field # pylint: disable=no-name-in-module
from fastapi.responses import JSONResponse
from fastapi.exceptions import HTTPException
from modules.api.helpers import decode_base64_to_image, encode_pil_to_base64
from modules import errors, shared, postprocessing
from modules.api import models, helpers
@@ -282,7 +283,12 @@ class APIProcess:
reqDict['image'] = helpers.decode_base64_to_image(reqDict['image'])
with self.queue_lock:
result = postprocessing.run_extras(extras_mode=0, image_folder="", input_dir="", output_dir="", video="", save_output=False, script_args=script_args, **reqDict)
return models.ResProcessImage(image=helpers.encode_pil_to_base64(result[0][0]), html_info=result[1])
if len(result) != 4:
raise HTTPException(status_code=500, detail="Invalid processing response")
outputs, _video, info, _params = result
if outputs is None or len(outputs) == 0:
raise HTTPException(status_code=500, detail="Invalid processing response")
return models.ResProcessImage(image=helpers.encode_pil_to_base64(outputs[0]), html_info=info)
def extras_batch_images_api(self, req: models.ReqProcessBatch):
"""Upscale or postprocess a batch of images using the configured upscaler pipeline."""
@@ -291,4 +297,9 @@ class APIProcess:
image_folder = [helpers.decode_base64_to_image(x.data) for x in image_list]
with self.queue_lock:
result = postprocessing.run_extras(extras_mode=1, image_folder=image_folder, image="", input_dir="", output_dir="", video="", save_output=False, script_args=script_args, **reqDict)
return models.ResProcessBatch(images=list(map(helpers.encode_pil_to_base64, result[0])), html_info=result[1])
if len(result) != 4:
raise HTTPException(status_code=500, detail="Invalid processing response")
outputs, _video, info, _params = result
if outputs is None or len(outputs) == 0:
raise HTTPException(status_code=500, detail="Invalid processing response")
return models.ResProcessBatch(images=list(map(helpers.encode_pil_to_base64, outputs)), html_info=info)