Update app.py

app.py

@@ -1,9 +1,12 @@
+from concurrent.futures import ThreadPoolExecutor
 import os
 import requests
 import shutil
 import gradio as gr
 from zipfile import ZipFile
 import logging
+from typing import IO
+
 
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
@@ -58,6 +61,28 @@ def download_image_by_image_id(image_id: str):
     dest = os.path.join(batch_id, image_id + ".jpg")
     download_image(url, dest)
 
+def rest_download_batch_images(batch_id: str) -> str:
+
+    image_ids = get_image_ids(batch_id)
+
+
+    def track_download(image_id):
+        download_image_by_image_id(image_id)
+
+    with ThreadPoolExecutor() as executor:
+        for _, image_id in enumerate(image_ids):
+            executor.submit(track_download, image_id)
+
+    zip_filename = f"{batch_id}.zip"
+    with ZipFile(zip_filename, 'w') as zipf:
+        for image_id in image_ids:
+            img_path = os.path.join(batch_id, f"{image_id}.jpg")
+            if os.path.exists(img_path):
+                zipf.write(img_path, arcname=os.path.basename(img_path))
+
+    return zip_filename
+
+
 def download_batch_images(batch_id: str, progress=None):
     logging.info(f"Starting download for batch {batch_id}")
 
@@ -97,22 +122,54 @@ def gradio_interface(batch_id_input, progress=gr.Progress()):
     except Exception as e:
         logging.error(f"Error processing batch: {e}")
         raise gr.Error(f"Error: {str(e)}")
+    
+def rest_gradio_interface(batch_id_input :str ) -> IO[bytes]:
+    try:
+        zip_file = rest_download_batch_images(batch_id_input)
+        return zip_file 
+    except Exception as e:
+        logging.error(f"Error processing batch: {e}")
+        raise gr.Error(f"Error: {str(e)}")
+
 
 with gr.Blocks() as app:
-    gr.Markdown("# Batch Image Downloader")
+    gr.Markdown("# IIIF Downloader")
+
+    with gr.Tab("Download Batch"):
+
+        with gr.Row():
+            with gr.Column():
+                batch_id_input = gr.Textbox(label="Batch ID", placeholder="Enter batch ID.")
+                download_button = gr.Button("Download Images")
+
+            with gr.Column():
+                output_file = gr.File(label="Download Zip File")
+
+        download_button.click(
+            gradio_interface,
+            inputs=[batch_id_input],
+            outputs=[output_file]
+        )
+
+        download_button.click(
+            rest_gradio_interface,
+            api_name="iiif_rest_download" ,
+            inputs=[batch_id_input],
+            outputs=[output_file]
+        )
+
+    with gr.Tab("Multiple Batches"):
+        gr.Markdown("WIP")
+        gr.Markdown("Make it possible to download batches to a huggingface account so it can be used through fastapi")
+        gr.Markdown("Will uses threading")
+
+        pass
+
 
-    with gr.Row():
-        with gr.Column():
-            batch_id_input = gr.Textbox(label="Batch ID", placeholder="Enter batch ID.")
-            download_button = gr.Button("Download Images")
-        with gr.Column():
-            output_file = gr.File(label="Download Zip File")
+    with gr.Tab("How to use"):
+        gr.Markdown("WIP, instructional video")
+        pass
 
-    download_button.click(
-        gradio_interface,
-        inputs=[batch_id_input],
-        outputs=[output_file]
-    )
 
 app.queue()
-app.launch()
+app.launch()