diff --git a/invokeai/app/services/image_storage.py b/invokeai/app/services/image_storage.py index 510de7c581..e9edd85d6e 100644 --- a/invokeai/app/services/image_storage.py +++ b/invokeai/app/services/image_storage.py @@ -95,7 +95,10 @@ class DiskImageStorage(ImageStorageBase): count = len(all_images) page_of_images = all_images[page * per_page : (page + 1) * per_page] - page_count = int(count / per_page) + 1 + + page_count_trunc = int(count / per_page) + page_count_mod = count % per_page + page_count = page_count_trunc if page_count_mod == 0 else page_count_trunc + 1 return PaginatedResults[ImageField]( items=page_of_images, diff --git a/invokeai/app/services/sqlite.py b/invokeai/app/services/sqlite.py index fd089014bb..9c768eb63a 100644 --- a/invokeai/app/services/sqlite.py +++ b/invokeai/app/services/sqlite.py @@ -106,10 +106,12 @@ class SqliteItemStorage(ItemStorageABC, Generic[T]): finally: self._lock.release() - pageCount = int(count / per_page) + 1 + page_count_trunc = int(count / per_page) + page_count_mod = count % per_page + page_count = page_count_trunc if page_count_mod == 0 else page_count_trunc + 1 return PaginatedResults[T]( - items=items, page=page, pages=pageCount, per_page=per_page, total=count + items=items, page=page, pages=page_count, per_page=per_page, total=count ) def search(