Upload 3 files

modutils.py

@@ -300,9 +300,6 @@ def save_images(images: list[Image.Image], metadatas: list[str]):
 
 def save_gallery_images(images, model_name="", progress=gr.Progress(track_tqdm=True)):
     progress(0, desc="Updating gallery...")
-    #dt_now = datetime.now(timezone(timedelta(hours=9)))
-    #dt_now = datetime.now(FILENAME_TIMEZONE)
-    #basename = dt_now.strftime('%Y%m%d_%H%M%S_')
     basename = f"{model_name.split('/')[-1]}_{datetime.now(FILENAME_TIMEZONE).strftime('%Y%m%d_%H%M%S')}_"
     if not images: return images, gr.update()
     output_images = []
@@ -332,9 +329,9 @@ 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)):
+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 not image: return gr.update(), gr.update()
+    if image is None or not isinstance(image, tuple): return gr.update(), gr.update()
     if not gallery: gallery = []
     if not files: files = []
     filename = f"{basename}.png"