Fix image count

Author: dshabin

src/superannotate/lib/core/usecases.py

@@ -2430,14 +2430,10 @@ def execute(self):
             data = self._service.get_project_images_count(
                 project_id=self._project.uuid, team_id=self._project.team_id
             )
-            count = 0
+            count = data["images"]["count"]
             if self._with_all_sub_folders:
-                for i in data["data"]:
+                for i in data["folders"]["data"]:
                     count += i["imagesCount"]
-            else:
-                for i in data["data"]:
-                    if i["name"] == self._folder.name:
-                        count += i["imagesCount"]
             self._response.data = count
         return self._response
 

src/superannotate/lib/infrastructure/services.py

@@ -151,7 +151,7 @@ class SuperannotateBackendService(BaseBackendService):
     URL_S3_ACCESS_POINT = "/project/{}/get-image-s3-access-point"
     URL_S3_UPLOAD_STATUS = "/project/{}/getS3UploadStatus"
     URL_GET_EXPORTS = "exports"
-    URL_IMAGES_COUNT = "images/folders-list"
+    URL_IMAGES_COUNT = "images/images-folders"
     URL_GET_CLASS = "class/{}"
     URL_ANNOTATION_UPLOAD_PATH_TOKEN = "images/getAnnotationsPathsAndTokens"
     URL_PRE_ANNOTATION_UPLOAD_PATH_TOKEN = "images/getPreAnnotationsPathsAndTokens"
@@ -754,7 +754,7 @@ def get_upload_status(self, project_id: int, team_id: int, folder_id: int):
         return res.json().get("progress")
 
     def get_project_images_count(self, team_id: int, project_id: int):
-        get_images_count_url = urljoin(self.api_url, self.URL_IMAGES_COUNT)
+        get_images_count_url = urljoin(self.api_url, self.URL_FOLDERS_IMAGES)
         res = self._request(
             get_images_count_url,
             "get",