attempt 2

debug.py

@@ -5,6 +5,7 @@ import uuid
 import time
 from huggingface_hub import CommitScheduler, HfApi
 from PIL import Image
+import numpy as np
 
 APP_VERSION = "0_3"
 HF_DATASET_REPO = "LPX55/upscaler_logs"  # Change to your dataset repo
@@ -14,7 +15,6 @@ LOG_DIR = "logs_" + APP_VERSION
 IMAGE_DIR = os.path.join(LOG_DIR, "upscaler")
 
 LOG_FILE = os.path.join(LOG_DIR, f"{int(time.time())}-logs.csv")
-os.makedirs(IMAGE_DIR, exist_ok=True)
 
 
 scheduler = CommitScheduler(
@@ -24,10 +24,18 @@ scheduler = CommitScheduler(
     every=5,
     private=True,
     token=HF_TOKEN,
-    allow_patterns=["*.csv", "images/*.png", "images/*.webp", "images/*.jpg", "images/*.jpeg"],
     path_in_repo="v" + APP_VERSION
 
 )
+def save_image(image_id: str, image_array: np.ndarray) -> None:
+    os.makedirs(IMAGE_DIR, exist_ok=True)
+    image_path = IMAGE_DIR / f"{image_id}.png"
+    with scheduler.lock:
+        try:
+            Image.fromarray(image_array).save(image_path)
+            print(f"Saved image: {image_path}")
+        except Exception as e:
+            print(f"Error saving image: {str(e)}")
 
 def log_params(
     prompt, scale, steps, controlnet_conditioning_scale, guidance_scale, seed, guidance_end,
@@ -37,8 +45,8 @@ def log_params(
     after_id = str(uuid.uuid4()) + "_after.png"
     before_path = os.path.join(IMAGE_DIR, before_id)
     after_path = os.path.join(IMAGE_DIR, after_id)
-    before_image.save(before_path)
-    after_image.save(after_path)
+    save_image(before_id, before_image)
+    save_image(after_id, after_image)
 
     is_new = not os.path.exists(LOG_FILE)
     with open(LOG_FILE, "a", newline='') as f: