added some error handling
This commit is contained in:
@@ -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)
|
||||
img = Image.open(preview_image)
|
||||
img.thumbnail((512, 512))
|
||||
bio = io.BytesIO()
|
||||
img.save(bio, format="PNG")
|
||||
window["preview_image"].update(data=bio.getvalue())
|
||||
window["civitai_info"].update(get_civitai_text(civitai_info))
|
||||
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()
|
||||
|
||||
Reference in New Issue
Block a user