Upload 9 files

app.py

@@ -82,7 +82,13 @@ def upload_training_images(files):
     db = load_db()
     if not files:
         choices = [x["filename"] for x in db["images"]]
-        return gr.update(choices=choices), "⚠️ No images selected."
+        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()],
+            gr.update(choices=choices, value=choices[-1] if choices else None),
+            annotation_html(choices[-1] if choices else None),
+        )
 
     if not isinstance(files, list):
         files = [files]
@@ -143,7 +149,20 @@ def upload_training_images(files):
         msg = f"✅ Saved {saved} image(s) to {location}."
     else:
         msg = "⚠️ No images were saved."
-    return gr.update(choices=choices, value=choices[-1] if choices else None), msg
+    latest = choices[-1] if choices else None
+    gallery_value = [x["path"] for x in db["images"] if Path(x.get("path", "")).exists()]
+    annotator_value = (
+        f'<div class="annotator-shell" data-filename="{html.escape(latest or "", quote=True)}">'
+        f'{annotation_html(latest)}</div>'
+        if latest else annotation_html(None)
+    )
+    return (
+        gr.update(choices=choices, value=latest),
+        msg,
+        gallery_value,
+        gr.update(choices=choices, value=latest),
+        annotator_value,
+    )
 
 
 def make_data_url(path):
@@ -739,6 +758,7 @@ with gr.Blocks(title="Anything Detector", js=ANNOTATOR_JS) as demo:
         files = gr.File(
             file_count="multiple",
             file_types=["image"],
+            type="filepath",
             label="Training images",
         )
         save_images = gr.Button(
@@ -753,13 +773,17 @@ with gr.Blocks(title="Anything Detector", js=ANNOTATOR_JS) as demo:
             label="Training image",
         )
         upload_status = gr.Markdown()
-
-        save_images.click(
-            upload_training_images,
-            inputs=files,
-            outputs=[image_selector, upload_status],
+        saved_images = gr.Gallery(
+            value=[x["path"] for x in load_db()["images"] if Path(x.get("path", "")).exists()],
+            label="Saved training images",
+            columns=5,
+            rows=2,
+            height="auto",
+            object_fit="contain",
+            show_label=True,
         )
 
+
     with gr.Tab("2 · Draw boxes"):
         gr.Markdown(
             """
@@ -777,11 +801,11 @@ This is a native HTML5 canvas — there is no `gradio_image_annotation` package.
 """
         )
 
+        _initial_images = load_db()["images"]
+        _initial_filename = _initial_images[0]["filename"] if _initial_images else None
         ann_selector = gr.Dropdown(
-            choices=[
-                x["filename"]
-                for x in load_db()["images"]
-            ],
+            choices=[x["filename"] for x in _initial_images],
+            value=_initial_filename,
             label="Image to annotate",
         )
 
@@ -829,6 +853,12 @@ This is a native HTML5 canvas — there is no `gradio_image_annotation` package.
             outputs=annotator_html,
         )
 
+        save_images.click(
+            upload_training_images,
+            inputs=files,
+            outputs=[image_selector, upload_status, saved_images, ann_selector, annotator_html],
+        )
+
         annotation_payload.change(
             save_annotation_payload,
             inputs=annotation_payload,