Upload modutils.py

modutils.py

@@ -329,18 +329,21 @@ def save_gallery_history(images, files, history_gallery, history_files, progress
     return gr.update(value=output_gallery), gr.update(value=output_files, visible=True)
 
 
-def save_image_history(image: str, gallery, files, model_name: str, progress=gr.Progress(track_tqdm=True)):
-    print(image)
+def save_image_history(image, gallery, files, model_name: str, progress=gr.Progress(track_tqdm=True)):
+    import numpy as np
     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) and not isinstance(image, Image.Image) and not isinstance(image, tuple)): return gr.update(), gr.update()
+    if image is None or (not isinstance(image, str) and not isinstance(image, Image.Image)  and not isinstance(image, np.array) and not isinstance(image, 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)
-    else:
+    elif isinstance(image, Image.Image):
         oldpath = "temp.png"
         image.save(oldpath)
+    elif isinstance(image, np.array):
+        oldpath = "temp.png"
+        Image.fromarray(image).convert('RGBA').save(oldpath)
     newpath = oldpath
     try:
         if newpath != oldpath and oldpath.exists():