Update app.py

app.py

@@ -16,7 +16,14 @@ def get_image_ids(batch_id: str) -> list[str]:
     response = requests.get(f"{IIIF_URL}/arkis!{batch_id}/manifest")
     response.raise_for_status()
     response = response.json()
-    image_ids = [item["id"].split("!")[1][:14] for item in response["items"]]
+    image_ids = []
+    for item in response.get("items", []):
+        id_parts = item["id"].split("!")
+        if len(id_parts) > 1:
+            image_id = id_parts[1][:14]
+            image_ids.append(image_id)
+        else:
+            logging.warning(f"Unexpected id format: {item['id']}")
     logging.info(f"Found {len(image_ids)} images in batch {batch_id}")
     return image_ids
 
@@ -52,6 +59,8 @@ def download_image_by_image_id(image_id: str):
 def download_batch_images(batch_id: str, workers: int = 2, progress=None):
     logging.info(f"Starting download for batch {batch_id}")
     image_ids = get_image_ids(batch_id)
+    if not image_ids:
+        raise ValueError("No images found to download.")
     total_images = len(image_ids)
 
     if progress:
@@ -89,10 +98,10 @@ def gradio_interface(batch_id_input, progress=gr.Progress()):
     batch_id = batch_id_input.strip()
     try:
         zip_file = download_batch_images(batch_id, progress=progress)
-        return zip_file  # Return the zip file for download
+        return zip_file, ""  # No error message
     except Exception as e:
         logging.error(f"Error processing batch: {e}")
-        return str(e)
+        return None, f"Error: {str(e)}"  # Return error message
 
 with gr.Blocks() as app:
     gr.Markdown("# Batch Image Downloader")
@@ -103,11 +112,12 @@ with gr.Blocks() as app:
             download_button = gr.Button("Download Images")
         with gr.Column():
             output_file = gr.File(label="Download Zip File")
+            error_message = gr.Textbox(label="Error Message", interactive=False)
 
     download_button.click(
         gradio_interface,
         inputs=[batch_id_input],
-        outputs=[output_file]
+        outputs=[output_file, error_message]
     )
 
 app.queue()