Add image upload support to bulk JSON import feature

app.py

@@ -1196,14 +1196,22 @@ with gr.Blocks(
 
                     with gr.Tab("Bulk Import (JSON)"):
                         gr.Markdown(
-                            "Paste a JSON array like: `[{\"input\": \"...\", \"output\": \"...\"}]`",
+                            "Paste a JSON array like: `[{\"input\": \"...\", \"output\": \"...\"}]`<br>"
+                            "**Images**: Upload images below and reference them by index (0-based) in JSON using `\"image_index\": 0`, "
+                            "or include base64 directly using `\"image\": \"data:image/...\"`",
                             elem_classes=["small-note"]
                         )
                         bulk_json = gr.Textbox(
                             show_label=False,
-                            placeholder='[{"input": "...", "output": "..."}]',
+                            placeholder='[{"input": "...", "output": "...", "image_index": 0}]',
                             lines=6
                         )
+                        bulk_images = gr.File(
+                            label="Upload Images (Optional)",
+                            file_count="multiple",
+                            file_types=["image"],
+                            height=100
+                        )
                         btn_import = gr.Button(
                             "Import JSON",
                             elem_classes=["btn-secondary"]
@@ -1418,8 +1426,8 @@ with gr.Blocks(
     )
     
     # Bulk Import
-    def import_bulk_json(json_text, current_dataset):
-        """Import examples from JSON with comprehensive error handling."""
+    def import_bulk_json(json_text, current_dataset, uploaded_images):
+        """Import examples from JSON with comprehensive error handling and image support."""
         try:
             # Validate inputs
             if not json_text or not json_text.strip():
@@ -1441,6 +1449,34 @@ with gr.Blocks(
             if len(data) == 0:
                 raise gr.Error("JSON array is empty. Add at least one example object.")
             
+            # Process uploaded images into base64 format
+            image_list = []
+            if uploaded_images:
+                for img_file in uploaded_images:
+                    try:
+                        if img_file is None:
+                            continue
+                        # Handle different file input formats
+                        if isinstance(img_file, str):
+                            # File path
+                            if os.path.exists(img_file):
+                                img_b64 = gradio_image_to_base64(img_file)
+                                if img_b64:
+                                    image_list.append(img_b64)
+                        elif hasattr(img_file, 'name'):
+                            # File object with name attribute
+                            img_b64 = gradio_image_to_base64(img_file.name)
+                            if img_b64:
+                                image_list.append(img_b64)
+                        else:
+                            # Try to process as image directly
+                            img_b64 = gradio_image_to_base64(img_file)
+                            if img_b64:
+                                image_list.append(img_b64)
+                    except Exception as e:
+                        logger.warning(f"Failed to process uploaded image: {str(e)}")
+                        continue
+            
             # Validate and import items
             imported_count = 0
             errors = []
@@ -1466,12 +1502,31 @@ with gr.Blocks(
                         errors.append(f"Item {i+1}: 'input' and 'output' cannot be empty")
                         continue
                     
+                    # Handle image - check for image_index first, then direct image field
+                    img_b64 = None
+                    if "image_index" in item:
+                        # Reference uploaded image by index
+                        img_idx = item["image_index"]
+                        if not isinstance(img_idx, int):
+                            errors.append(f"Item {i+1}: 'image_index' must be an integer")
+                            continue
+                        if img_idx < 0 or img_idx >= len(image_list):
+                            errors.append(f"Item {i+1}: 'image_index' {img_idx} is out of range (0-{len(image_list)-1})")
+                            continue
+                        img_b64 = image_list[img_idx]
+                    elif "image" in item:
+                        # Direct base64 image in JSON
+                        img_b64 = item["image"]
+                        if img_b64 and not isinstance(img_b64, str):
+                            errors.append(f"Item {i+1}: 'image' must be a base64 string")
+                            continue
+                    
                     # Add valid item
                     current_dataset.append({
                         "input": input_val.strip(),
                         "output": output_val.strip(),
-                        "image": item.get("image"),  # Optional
-                        "image_preview": "🖼️ Image" if item.get("image") else "-"
+                        "image": img_b64,  # Optional - can be None
+                        "image_preview": "🖼️ Image" if img_b64 else "-"
                     })
                     imported_count += 1
                     
@@ -1508,7 +1563,7 @@ with gr.Blocks(
     
     btn_import.click(
         import_bulk_json,
-        inputs=[bulk_json, dataset_state],
+        inputs=[bulk_json, dataset_state, bulk_images],
         outputs=[dataset_state, bulk_json]
     ).then(
         safe_update_table,