Upload 8 files

app.py

@@ -75,47 +75,32 @@ def image_record(filename):
     )
 
 
-def gallery_items(db=None):
-    """Return real PIL images for Gradio Gallery so saved files are visibly rendered."""
-    if db is None:
-        db = load_db()
+def gallery_items(db):
+    """Return Gradio Gallery items using absolute paths plus captions."""
     items = []
     for item in db.get("images", []):
         path = Path(item.get("path", ""))
-        try:
-            if path.exists() and path.is_file():
-                with Image.open(path) as im:
-                    items.append((im.convert("RGB").copy(), item.get("filename", path.name)))
-        except Exception:
-            continue
+        if path.is_file():
+            items.append((str(path), item.get("filename", path.name)))
     return items
 
 
-def uploaded_path(file):
-    """Get the actual temporary upload path across Gradio 4/5 FileData formats."""
-    if file is None:
-        return None
+def normalize_uploaded_file(file):
+    """Get a real local path from Gradio filepath/FileData variants."""
     if isinstance(file, (str, os.PathLike)):
         return Path(file)
     if isinstance(file, dict):
-        for key in ("path", "name"):
-            value = file.get(key)
-            if value:
-                candidate = Path(str(value))
-                if candidate.exists():
-                    return candidate
-        return None
-    for attr in ("path", "name"):
-        value = getattr(file, attr, None)
-        if value:
-            candidate = Path(str(value))
-            if candidate.exists():
-                return candidate
-    return None
+        value = file.get("path") or file.get("name") or file.get("url")
+        if value and not str(value).startswith(("http://", "https://")):
+            return Path(str(value))
+    value = getattr(file, "path", None) or getattr(file, "name", None)
+    if value:
+        return Path(str(value))
+    raise ValueError(f"Unsupported upload object: {type(file).__name__}")
 
 
 def upload_training_images(files):
-    """Copy uploaded files into dataset storage and immediately refresh every image UI."""
+    """Copy uploaded files into persistent dataset storage and register them."""
     DATA.mkdir(parents=True, exist_ok=True)
     IMAGE_DIR.mkdir(parents=True, exist_ok=True)
     db = load_db()
@@ -138,9 +123,9 @@ def upload_training_images(files):
 
     for file in files:
         try:
-            src = uploaded_path(file)
-            if src is None or not src.is_file():
-                raise FileNotFoundError("Could not resolve the uploaded image's temporary file path")
+            src = normalize_uploaded_file(file)
+            if not src.exists() or not src.is_file():
+                raise FileNotFoundError(f"Uploaded file not found: {src}")
 
             ext = src.suffix.lower() or ".jpg"
             if ext not in {".jpg", ".jpeg", ".png", ".webp", ".bmp"}:
@@ -808,7 +793,7 @@ with gr.Blocks(title="Anything Detector", js=ANNOTATOR_JS) as demo:
         )
         upload_status = gr.Markdown()
         saved_images = gr.Gallery(
-            value=gallery_items(),
+            value=gallery_items(load_db()),
             label="Saved training images",
             columns=5,
             rows=2,
@@ -970,4 +955,5 @@ if __name__ == "__main__":
         server_port=7860,
         ssr_mode=False,
         max_file_size="100mb",
+        allowed_paths=[str(DATA), str(IMAGE_DIR)],
     )