Update app to load images from Hugging Face Hub instead of local files

app.py

@@ -11,6 +11,7 @@ from PIL import Image
 import colorsys
 from pathlib import Path
 import matplotlib.pyplot as plt
+from huggingface_hub import hf_hub_download, list_repo_files
 
 
 # --- Utility Functions ---
@@ -18,43 +19,60 @@ import matplotlib.pyplot as plt
 
 @st.cache_resource
 def load_image(image_name):
-    """Load an image from the images folder."""
-    img_path = Path("images") / image_name
-    if img_path.exists():
+    """Load an image from Hugging Face Hub."""
+    try:
+        # Download image from HF Hub dataset
+        img_path = hf_hub_download(
+            repo_id="amithjkamath/exampleimages",
+            filename=image_name,
+            repo_type="dataset"
+        )
         img = cv.imread(str(img_path))
         if img is not None:
             return cv.cvtColor(img, cv.COLOR_BGR2RGB)
+    except Exception as e:
+        st.warning(f"Could not load image {image_name}: {e}")
     return None
 
 
 def get_available_images(exclude_colorblind_tests=False):
-    """Get list of available images in the images folder.
+    """Get list of available images from Hugging Face Hub.
 
     Args:
         exclude_colorblind_tests: If True, exclude numbered test images (1-, 2-, etc.)
     """
-    images_dir = Path("images")
-    if images_dir.exists():
-        images = list(images_dir.glob("*.png")) + list(images_dir.glob("*.jpg"))
-        files = sorted([f.name for f in images])
+    try:
+        files = list_repo_files(
+            repo_id="amithjkamath/exampleimages",
+            repo_type="dataset"
+        )
+        # Filter for image files
+        image_files = [f for f in files if f.lower().endswith(('.png', '.jpg', '.jpeg'))]
+        image_files = sorted(image_files)
 
         if exclude_colorblind_tests:
             # Filter out colorblind test images (those starting with a digit)
-            files = [f for f in files if not f[0].isdigit()]
+            image_files = [f for f in image_files if not f[0].isdigit()]
 
-        return files
-    return []
+        return image_files
+    except Exception as e:
+        st.warning(f"Could not fetch images from Hugging Face Hub: {e}")
+        return []
 
 
 def get_colorblind_test_images():
-    """Get only colorblind test images (numbered)."""
-    images_dir = Path("images")
-    if images_dir.exists():
-        images = list(images_dir.glob("*.png")) + list(images_dir.glob("*.jpg"))
-        files = sorted([f.name for f in images])
-        # Return only images starting with a digit
-        return [f for f in files if f[0].isdigit()]
-    return []
+    """Get only colorblind test images (numbered) from Hugging Face Hub."""
+    try:
+        files = list_repo_files(
+            repo_id="amithjkamath/exampleimages",
+            repo_type="dataset"
+        )
+        # Filter for image files that start with a digit
+        image_files = [f for f in files if f.lower().endswith(('.png', '.jpg', '.jpeg')) and f[0].isdigit()]
+        return sorted(image_files)
+    except Exception as e:
+        st.warning(f"Could not fetch colorblind test images: {e}")
+        return []
 
 
 def generate_gradient_image(size=256):

requirements.txt

@@ -3,3 +3,4 @@ numpy==1.26.4
 opencv-python==4.10.0.84
 pillow==10.4.0
 matplotlib==3.9.2
+huggingface_hub==0.24.7