extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-05-01 06:44:56 +02:00
Code Issues Packages Projects Releases Wiki Activity

Compatibility mode for API calls made without the streaming flag

Browse Source
This commit is contained in:
cmdr2 2022-09-22 22:55:57 +05:30
parent df73be495e
commit b934a6b6e9
1 changed files with 12 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
ui/server.py
View File

@ -114,16 +114,24 @@ def image(req : ImageRequest):
r.use_face_correction = req.use_face_correction r.use_face_correction = req.use_face_correction
r.show_only_filtered_image = req.show_only_filtered_image r.show_only_filtered_image = req.show_only_filtered_image
r.stream_progress_updates = req.stream_progress_updates r.stream_progress_updates = True # the underlying implementation only supports streaming
r.stream_image_progress = req.stream_image_progress r.stream_image_progress = req.stream_image_progress
try: try:
if not req.stream_progress_updates:
r.stream_image_progress = False
res = runtime.mk_img(r) res = runtime.mk_img(r)
if r.stream_progress_updates: if req.stream_progress_updates:
return StreamingResponse(res, media_type='application/json') return StreamingResponse(res, media_type='application/json')
else: else: # compatibility mode: buffer the streaming responses, and return the last one
return res.json() last_result = None
for result in res:
last_result = result
return json.loads(last_result)
except Exception as e: except Exception as e:
print(traceback.format_exc()) print(traceback.format_exc())
return HTTPException(status_code=500, detail=str(e)) return HTTPException(status_code=500, detail=str(e))