added some error handling

This commit is contained in:
2023-06-22 09:24:13 -04:00
parent ddfe682499
commit a40cc83882

View File

@@ -41,7 +41,6 @@ layout = [
window = sg.Window("Model Folder", layout)
while True:
event, values = window.read()
print(event)
if event == sg.WIN_CLOSED or event == "Cancel":
break
if event == "model_folder":
@@ -58,11 +57,17 @@ while True:
if event == "models":
model_path = values["models"]
preview_image, civitai_info = get_extra_filenames(model_path)
if preview_image:
img = Image.open(preview_image)
img.thumbnail((512, 512))
bio = io.BytesIO()
img.save(bio, format="PNG")
window["preview_image"].update(data=bio.getvalue())
else:
window["preview_image"].update(data=None)
if civitai_info:
window["civitai_info"].update(get_civitai_text(civitai_info))
else:
window["civitai_info"].update(data=None)
window.close()