Upload 9 files

app.py

@@ -75,8 +75,47 @@ 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()
+    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
+    return items
+
+
+def uploaded_path(file):
+    """Get the actual temporary upload path across Gradio 4/5 FileData formats."""
+    if file is None:
+        return None
+    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
+
+
 def upload_training_images(files):
-    """Copy uploaded files into persistent dataset storage and register them."""
+    """Copy uploaded files into dataset storage and immediately refresh every image UI."""
     DATA.mkdir(parents=True, exist_ok=True)
     IMAGE_DIR.mkdir(parents=True, exist_ok=True)
     db = load_db()
@@ -85,7 +124,7 @@ def upload_training_images(files):
         return (
             gr.update(choices=choices, value=choices[-1] if choices else None),
             "⚠️ No images selected.",
-            [x["path"] for x in db["images"] if Path(x.get("path", "")).exists()],
+            gallery_items(db),
             gr.update(choices=choices, value=choices[-1] if choices else None),
             annotation_html(choices[-1] if choices else None),
         )
@@ -99,14 +138,9 @@ def upload_training_images(files):
 
     for file in files:
         try:
-            # Gradio normally supplies a filepath string; also support objects
-            # exposing a .name attribute.
-            raw = getattr(file, "name", file)
-            if not raw:
-                raise ValueError("Upload has no file path")
-            src = Path(str(raw))
-            if not src.exists() or not src.is_file():
-                raise FileNotFoundError(f"Uploaded file not found: {src}")
+            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")
 
             ext = src.suffix.lower() or ".jpg"
             if ext not in {".jpg", ".jpeg", ".png", ".webp", ".bmp"}:
@@ -150,7 +184,7 @@ def upload_training_images(files):
     else:
         msg = "⚠️ No images were saved."
     latest = choices[-1] if choices else None
-    gallery_value = [x["path"] for x in db["images"] if Path(x.get("path", "")).exists()]
+    gallery_value = gallery_items(db)
     annotator_value = (
         f'<div class="annotator-shell" data-filename="{html.escape(latest or "", quote=True)}">'
         f'{annotation_html(latest)}</div>'
@@ -774,7 +808,7 @@ with gr.Blocks(title="Anything Detector", js=ANNOTATOR_JS) as demo:
         )
         upload_status = gr.Markdown()
         saved_images = gr.Gallery(
-            value=[x["path"] for x in load_db()["images"] if Path(x.get("path", "")).exists()],
+            value=gallery_items(),
             label="Saved training images",
             columns=5,
             rows=2,