Update app.py

app.py

@@ -46,10 +46,17 @@ def download_batch_images(batch_id: str, workers: int = 2, progress=None):
 
     if progress:
         progress(0, desc=f"Starting download for {batch_id}...")
-    
+
+    def track_download(image_id):
+        download_image_by_image_id(image_id)
+        if progress:
+            # Update progress after each image
+            current_progress = image_ids.index(image_id) / total_images
+            progress(current_progress, desc=f"Downloading {image_id}...")
+
     with ThreadPoolExecutor(max_workers=workers) as executor:
         for image_id in image_ids:
-            executor.submit(download_image_by_image_id, image_id)
+            executor.submit(track_download, image_id)
 
     # Zip the folder with downloaded images
     zip_filename = f"{batch_id}.zip"
@@ -60,9 +67,10 @@ def download_batch_images(batch_id: str, workers: int = 2, progress=None):
 
     if progress:
         progress(1, desc=f"Completed {batch_id}")
-        
+
     return zip_filename
 
+
 def gradio_interface(batch_ids_input, progress=gr.Progress()):
     batch_ids = [batch_id.strip() for batch_id in batch_ids_input.split("\n") if batch_id.strip()]