Refactor progress reporting for annotation and image downloads using …

…tqdm_sly for better user feedback

src/functions.py

@@ -121,7 +121,8 @@ def _write_new_ann(path, content):
     img_names = [f"{ds.name}_{image_info.name}" for image_info in images_infos]
     ann_infos = []
 
-    coro = api.annotation.download_bulk_async(ds.id, img_ids)
+    ann_progress = sly.tqdm_sly(desc="Downloading annotations...", total=len(img_ids))
+    coro = api.annotation.download_bulk_async(ds.id, img_ids, progress_cb=ann_progress)
     loop = sly.utils.get_or_create_event_loop()
     if loop.is_running():
         future = asyncio.run_coroutine_threadsafe(coro, loop)
@@ -187,7 +188,7 @@ def _write_new_ann(path, content):
             image_processed = True
             train_count += 1
 
-        progress.iter_done_report()
+        progress(1)
 
     sly.logger.info(
         f"DATASET '{ds.name}': {train_count} images for train, {val_count} images for validation"

src/main.py

@@ -54,7 +54,7 @@ def process(self, context: sly.app.Export.Context):
         total_images_count = 0
         skipped = []
 
-        progress = sly.Progress("Transforming annotations...", images_count)
+        progress = sly.tqdm_sly(desc="Transforming annotations...", total=images_count)
         for ds in datasets:
             train_info, val_info = f.process_images(
                 api,
@@ -81,7 +81,7 @@ def process(self, context: sly.app.Export.Context):
             total_images_count += len(train_img_paths)
             total_images_count += len(val_image_paths)
 
-        download_progress = sly.Progress("Downloading images...", total_images_count)
+        download_progress = sly.tqdm_sly(desc="Downloading images...", total=total_images_count)
         img_ids = []
         img_paths = []
         for ds in datasets:
@@ -91,7 +91,7 @@ def process(self, context: sly.app.Export.Context):
             img_paths.extend(all_paths[ds.id]["val_paths"])
 
         coro = api.image.download_paths_async(
-            img_ids, img_paths, progress_cb=download_progress.iters_done_report
+            img_ids, img_paths, progress_cb=download_progress
         )
         loop = sly.utils.get_or_create_event_loop()
         if loop.is_running():