Upload modutils.py

modutils.py

@@ -331,21 +331,21 @@ def save_gallery_history(images, files, history_gallery, history_files, progress
 
 
 def save_image_history(image, gallery, files, model_name: str, progress=gr.Progress(track_tqdm=True)):
-    basename = f"{model_name.split('/')[-1]}_{datetime.now(FILENAME_TIMEZONE).strftime('%Y%m%d_%H%M%S')}"
-    if image is None or not isinstance(image, (str, Image.Image, np.ndarray, tuple)): return gr.update(), gr.update()
     if not gallery: gallery = []
     if not files: files = []
-    filename = f"{basename}.png"
-    if isinstance(image, tuple): image = image[0]
-    if isinstance(image, str): oldpath = Path(image)
-    elif isinstance(image, Image.Image):
-        oldpath = "temp.png"
-        image.save(oldpath)
-    elif isinstance(image, np.ndarray):
-        oldpath = "temp.png"
-        Image.fromarray(image).convert('RGBA').save(oldpath)
-    newpath = oldpath
     try:
+        basename = f"{model_name.split('/')[-1]}_{datetime.now(FILENAME_TIMEZONE).strftime('%Y%m%d_%H%M%S')}"
+        if image is None or not isinstance(image, (str, Image.Image, np.ndarray, tuple)): return gr.update(), gr.update()
+        filename = f"{basename}.png"
+        newpath = oldpath
+        if isinstance(image, tuple): image = image[0]
+        if isinstance(image, str): oldpath = Path(image)
+        elif isinstance(image, Image.Image):
+            oldpath = "temp.png"
+            image.save(oldpath)
+        elif isinstance(image, np.ndarray):
+            oldpath = "temp.png"
+            Image.fromarray(image).convert('RGBA').save(oldpath)
         if newpath != oldpath and oldpath.exists():
             shutil.copy(oldpath.resolve(), Path(filename).resolve())
             newpath = Path(filename).resolve()