Add try-catch block for error handling
parent
428664a2a4
commit
7e37356d2f
|
|
@ -2,6 +2,7 @@ from scripts.iib.parsers.comfyui import ComfyUIParser
|
|||
from scripts.iib.parsers.sd_webui import SdWebUIParser
|
||||
from scripts.iib.parsers.fooocus import FooocusParser
|
||||
from scripts.iib.parsers.model import ImageGenerationInfo
|
||||
from scripts.iib.logger import logger
|
||||
from PIL import Image
|
||||
|
||||
|
||||
|
|
@ -10,5 +11,10 @@ def parse_image_info(image_path: str) -> ImageGenerationInfo:
|
|||
with Image.open(image_path) as img:
|
||||
for parser in parsers:
|
||||
if parser.test(img, image_path):
|
||||
return parser.parse(img, image_path)
|
||||
try:
|
||||
return parser.parse(img, image_path)
|
||||
except Exception as e:
|
||||
logger.error(e, stack_info=True)
|
||||
print(e)
|
||||
return ImageGenerationInfo()
|
||||
raise Exception("matched parser is not found")
|
||||
|
|
|
|||
Loading…
Reference in New Issue