Update app.py

app.py

@@ -29,8 +29,12 @@ def download_image(url: str, dest: str) -> None:
     """
     logging.info(f"Downloading image from {url} to {dest}")
     response = requests.get(url, stream=True)
-    with open(dest, "wb") as out_file:
-        shutil.copyfileobj(response.raw, out_file)
+    if response.status_code == 200:
+        with open(dest, "wb") as out_file:
+            shutil.copyfileobj(response.raw, out_file)
+        logging.info(f"Successfully downloaded image: {dest}")
+    else:
+        logging.error(f"Failed to download image from {url}. Status code: {response.status_code}")
     del response
 
 def download_image_by_image_id(image_id: str):
@@ -70,7 +74,10 @@ def download_batch_images(batch_id: str, workers: int = 2, progress=None):
     with ZipFile(zip_filename, 'w') as zipf:
         for image_id in image_ids:
             img_path = os.path.join(batch_id, f"{image_id}.jpg")
-            zipf.write(img_path, arcname=os.path.basename(img_path))
+            if os.path.exists(img_path):
+                zipf.write(img_path, arcname=os.path.basename(img_path))
+            else:
+                logging.warning(f"Image {img_path} does not exist and will not be zipped.")
 
     if progress:
         progress(1, desc=f"Completed {batch_id}")