complete pipeline

app.py

@@ -1,26 +1,77 @@
-import gradio as gr
-import os, shutil
+import gradio as gr, os, shutil, glob, datetime, getpass
+from huggingface_hub import upload_file
 
-UPLOAD_DIR = "uploads"              # permanent storage
-os.makedirs(UPLOAD_DIR, exist_ok=True)
+# ---------- CONFIG -------------------------------------------------
+DATASET_REPO   = "your‑name/my_dataset"        # change me
+PENDING_DIR    = "pending"
+APPROVED_DIR   = "approved"
+ADMIN_KEY_ENV  = os.getenv("ADMIN_KEY")        # optional
+HF_TOKEN       = os.getenv("HF_TOKEN")
 
-def save_file(temp_path: str):      # temp_path is *already* a str
-    if not temp_path:
+for d in (PENDING_DIR, APPROVED_DIR):
+    os.makedirs(d, exist_ok=True)
+# -------------------------------------------------------------------
+
+# -------- UPLOAD HANDLER (public) ----------------------------------
+def handle_upload(tmp_path: str):
+    if not tmp_path:
         return "❌ No file received."
+    fname = os.path.basename(tmp_path)
+    dest  = os.path.join(PENDING_DIR, fname)
+    shutil.copy(tmp_path, dest)
+    return f"✅ `{fname}` uploaded for review!"
+
+# -------- LIST PENDING FILES --------------------------------------
+def list_pending():
+    return sorted(glob.glob(f"{PENDING_DIR}/*"))
+
+# -------- ADMIN: APPROVE & PUSH -----------------------------------
+def approve_files(password, selected_files):
+    if ADMIN_KEY_ENV and password != ADMIN_KEY_ENV:
+        return "🔒 Wrong admin key!", None
+
+    if not selected_files:
+        return "⚠️ Nothing selected.", None
 
-    filename   = os.path.basename(temp_path)
-    dest_path  = os.path.join(UPLOAD_DIR, filename)
-    shutil.copy(temp_path, dest_path)      # copy from tmp → uploads/
-    return f"✅ `{filename}` saved!"
+    log = []
+    for file_path in selected_files:
+        filename = os.path.basename(file_path)
+        # 1) push to dataset repo
+        upload_file(
+            path_or_fileobj=file_path,
+            path_in_repo=f"images/{filename}",
+            repo_id=DATASET_REPO,
+            repo_type="dataset",
+            token=HF_TOKEN,
+        )
+        # 2) move to approved/
+        shutil.move(file_path, os.path.join(APPROVED_DIR, filename))
+        log.append(f"✅ {filename}")
+    return "\n".join(log), list_pending()
 
+# -------------------- UI ------------------------------------------
 with gr.Blocks() as demo:
-    gr.Markdown("## 🖼️ Anonymous image upload (no login required)")
-    uploader = gr.File(
-        label="Upload image",
-        file_types=["image"],        # or [".png", ".jpg", ".jpeg"]
-        type="filepath"              # <‑‑ CRUCIAL
+    gr.Markdown("## 🖼️ Anonymous Image Upload")
+    with gr.Row():
+        up_comp  = gr.File(type="filepath", file_types=["image"])
+        up_msg   = gr.Textbox(label="Status")
+    up_comp.upload(handle_upload, up_comp, up_msg)
+
+    # ----- review panel -----
+    gr.Markdown("## 🔍 Admin review panel")
+    admin_pass  = gr.Textbox(label="Admin key", type="password")
+    gallery     = gr.Gallery(label="Pending images").style(grid=4)
+    approve_btn = gr.Button("Approve & publish")
+    review_log  = gr.Textbox(label="Log / Result")
+
+    # initialise gallery
+    gallery.value = list_pending()
+
+    # approve action
+    approve_btn.click(
+        fn=approve_files,
+        inputs=[admin_pass, gallery],
+        outputs=[review_log, gallery],
     )
-    status = gr.Textbox(label="Status")
-    uploader.upload(save_file, uploader, status)   # use .upload for clarity
 
 demo.queue().launch()