drc97/sd_viewer
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

added some error handling

Browse Source
This commit is contained in:
Daniel Cormier
2023-06-22 09:24:13 -04:00
parent ddfe682499
commit a40cc83882
1 changed files with 12 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
src/program.py
View File

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