Merge pull request #51 from bioimage-io/abs_covers

resolve thumbnail sources
bioimage-io · Apr 23, 2024 · 1a751dc · 1a751dc
2 parents 172e12a + 0dcbe5b
commit 1a751dc
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/bioimageio_collection_backoffice/generate_collection_json.py b/bioimageio_collection_backoffice/generate_collection_json.py
@@ -80,7 +80,10 @@ def maybe_swap_with_thumbnail(
 
         if isinstance(src, str):
             clean_name = Path(src).name  # remove any leading './'
-            return thumbnails.get(clean_name, src)
+            if clean_name in thumbnails:
+                return p.get_file_url(thumbnails[clean_name])
+            else:
+                return src
 
         return src
 

diff --git a/bioimageio_collection_backoffice/remote_resource.py b/bioimageio_collection_backoffice/remote_resource.py
@@ -255,6 +255,9 @@ def get_uploader(self):
 
         return Uploader(email=email, name=name)
 
+    def get_file_url(self, path: str):
+        return self.client.get_file_url(f"{self.folder}files/{path}")
+
     def get_file_urls(self):
         return self.client.get_file_urls(f"{self.folder}files/")