Update app.py

app.py

@@ -14,7 +14,7 @@ IMAGE_DATASET_DIR = Path("image_dataset") / f"train-{uuid4()}"
 IMAGE_DATASET_DIR.mkdir(parents=True, exist_ok=True)
 IMAGE_JSONL_PATH = IMAGE_DATASET_DIR / "metadata.jsonl"
 
-# Read Hugging Face token from secret (HF Spaces automatically sets it)
+# Read Hugging Face token from secret
 HF_TOKEN = os.environ.get("HF_TOKEN")
 if not HF_TOKEN:
     raise ValueError("HF_TOKEN not found! Please set it in HF Spaces Secrets.")
@@ -25,18 +25,18 @@ scheduler = CommitScheduler(
     repo_type="dataset",
     folder_path=IMAGE_DATASET_DIR,
     path_in_repo=IMAGE_DATASET_DIR.name,
-    token=HF_TOKEN,  # pass the token here
+    token=HF_TOKEN,  # use the token from secret
 )
 
 # ---------------- Save uploaded image ----------------
 def save_uploaded_image(user_id: str, image: Image.Image) -> str:
     """
-    Save the uploaded image to local dataset folder and log metadata.
+    Save the uploaded image to local dataset folder, log metadata, and push to HF Hub.
     """
     image_path = IMAGE_DATASET_DIR / f"{uuid4()}.png"
 
     with scheduler.lock:
-        # Save the image
+        # Save the image locally
         image.save(image_path)
 
         # Save metadata to JSONL
@@ -48,7 +48,10 @@ def save_uploaded_image(user_id: str, image: Image.Image) -> str:
             }, f)
             f.write("\n")
 
-    return f"Image uploaded successfully for user {user_id}!"
+        # Automatically commit & push to HF Hub
+        scheduler.commit(message=f"Add image {image_path.name} for user {user_id}")
+
+    return f"Image uploaded and pushed to repo successfully for user {user_id}!"
 
 # ---------------- Gradio UI ----------------
 def get_demo():