From cfad51634b1788d46427c4f419c36f4e4657e7c9 Mon Sep 17 00:00:00 2001 From: Qing Date: Sat, 23 Nov 2024 11:03:22 +0800 Subject: [PATCH] if input image file is not configured, return code 200 --- iopaint/api.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/iopaint/api.py b/iopaint/api.py index aac03f0..b144a3c 100644 --- a/iopaint/api.py +++ b/iopaint/api.py @@ -186,7 +186,7 @@ class Api: file_to_write = Path(file.filename) if not file_to_write.is_file(): return - + origin_image_bytes = file.file.read() with open(self.config.output_dir / file_to_write.name, "wb") as fw: fw.write(origin_image_bytes) @@ -241,7 +241,10 @@ class Api: ) def api_input_image(self) -> FileResponse: - if self.config.input and self.config.input.is_file(): + if self.config.input is None: + raise HTTPException(status_code=200, detail="No input image configured") + + if self.config.input.is_file(): return FileResponse(self.config.input) raise HTTPException(status_code=404, detail="Input image not found")