Upload modutils.py

modutils.py

@@ -331,11 +331,15 @@ 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): return gr.update(), gr.update()
+    if image is None or (not isinstance(image, str) and not isinstance(image, Image.Image) and not isinstance(image, tuple)): return gr.update(), gr.update()
     if not gallery: gallery = []
     if not files: files = []
     filename = f"{basename}.png"
-    oldpath = Path(image)
+    if isinstance(image, tuple): image = image[0]
+    if isinstance(image, str): oldpath = Path(image)
+    else:
+        oldpath = "temp.png"
+        image.save(oldpath)
     newpath = oldpath
     try:
         if newpath != oldpath and oldpath.exists():