Upload modutils.py

modutils.py

@@ -337,7 +337,6 @@ def save_image_history(image, gallery, files, model_name: str, progress=gr.Progr
         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):
@@ -346,14 +345,15 @@ def save_image_history(image, gallery, files, model_name: str, progress=gr.Progr
         elif isinstance(image, np.ndarray):
             oldpath = "temp.png"
             Image.fromarray(image).convert('RGBA').save(oldpath)
-        if newpath != oldpath and oldpath.exists():
+        newpath = oldpath
+        if oldpath.exists():
             shutil.copy(oldpath.resolve(), Path(filename).resolve())
             newpath = Path(filename).resolve()
+        files.insert(0, str(newpath))
+        gallery.insert(0, (str(newpath), str(filename)))
     except Exception as e:
         print(e)
     finally: 
-        files.insert(0, str(newpath))
-        gallery.insert(0, (str(newpath), str(filename)))
         return gr.update(value=gallery), gr.update(value=files, visible=True)