fix dockerfile

Dockerfile

@@ -37,8 +37,11 @@ ENV ENV=production
 
 # Add script to download images from Hugging Face during build
 COPY download_images.py .
-RUN if [ -n "$HF_USERNAME" ] && [ -n "$HF_TOKEN" ] && [ -n "$HF_DATASET_REPO" ]; then \
-    echo "Downloading images from Hugging Face dataset..." && \
+# Use build secrets to access HF_TOKEN during build time
+RUN --mount=type=secret,id=HF_TOKEN \
+    if [ -n "$HF_USERNAME" ] && [ -f /run/secrets/HF_TOKEN ] && [ -n "$HF_DATASET_REPO" ]; then \
+    export HF_TOKEN=$(cat /run/secrets/HF_TOKEN) && \
+    echo "Downloading images from Hugging Face dataset for $HF_USERNAME/$HF_DATASET_REPO..." && \
     python download_images.py; \
     else \
     echo "Hugging Face credentials not set, skipping image download."; \

app.py

@@ -221,7 +221,7 @@ def get_hf_image_url(filename):
     """Get the URL for an image in the Hugging Face repo."""
     # Return the direct Space URL if available, otherwise use dataset URL
     if SPACE_NAME and HF_USERNAME:
-        return f"https://{HF_USERNAME}-{SPACE_NAME}.hf.space/static/uploads/{filename}"
+        return f"https://{HF_USERNAME.lower()}-{SPACE_NAME}.hf.space/static/uploads/{filename}"
     elif HF_USERNAME:
         return f"https://huggingface.co/datasets/{HF_USERNAME}/{DATASET_REPO}/resolve/main/{IMAGES_PATH}/{filename}"
     return None
@@ -533,7 +533,7 @@ async def upload_image(
             and HF_USERNAME
             and os.environ.get("ENV", "development") == "production"
         ):
-            full_url = f"https://{HF_USERNAME}-{SPACE_NAME}.hf.space/static/uploads/{unique_filename}"
+            full_url = f"https://{HF_USERNAME.lower()}-{SPACE_NAME}.hf.space/static/uploads/{unique_filename}"
         else:
             full_url = f"{request.base_url}static/uploads/{unique_filename}"
 
@@ -648,7 +648,7 @@ async def upload_with_replace(
             and HF_USERNAME
             and os.environ.get("ENV", "development") == "production"
         ):
-            full_url = f"https://{HF_USERNAME}-{SPACE_NAME}.hf.space/static/uploads/{unique_filename}"
+            full_url = f"https://{HF_USERNAME.lower()}-{SPACE_NAME}.hf.space/static/uploads/{unique_filename}"
         else:
             full_url = f"{request.base_url}static/uploads/{unique_filename}"
 
@@ -693,9 +693,7 @@ async def view_image(request: Request, file_name: str):
         and HF_USERNAME
         and os.environ.get("ENV", "development") == "production"
     ):
-        embed_url = (
-            f"https://{HF_USERNAME}-{SPACE_NAME}.hf.space/static/uploads/{file_name}"
-        )
+        embed_url = f"https://{HF_USERNAME.lower()}-{SPACE_NAME}.hf.space/static/uploads/{file_name}"
     else:
         embed_url = f"{request.base_url}static/uploads/{file_name}"
 
@@ -803,7 +801,9 @@ async def startup_event():
             print(f"  - Dataset: {DATASET_REPO}")
             if SPACE_NAME:
                 print(f"  - Space: {SPACE_NAME}")
-                print(f"  - Space URL: https://{HF_USERNAME}-{SPACE_NAME}.hf.space")
+                print(
+                    f"  - Space URL: https://{HF_USERNAME.lower()}-{SPACE_NAME}.hf.space"
+                )
             else:
                 print("  - Space name not set. Direct URLs will use dataset links.")
         else:

download_images.py

@@ -13,6 +13,18 @@ HF_USERNAME = os.environ.get("HF_USERNAME", "")
 HF_TOKEN = os.environ.get("HF_TOKEN", "")
 DATASET_REPO = os.environ.get("HF_DATASET_REPO", "image-uploader-data")
 
+# Validate required environment variables
+if not HF_USERNAME:
+    print("ERROR: HF_USERNAME environment variable is not set")
+    exit(1)
+
+if not HF_TOKEN:
+    print("ERROR: HF_TOKEN environment variable is not set")
+    exit(1)
+
+print(f"Using Hugging Face credentials for user: {HF_USERNAME}")
+print(f"Dataset repository: {DATASET_REPO}")
+
 # Set HF cache directory to a writable location
 os.environ["HF_HOME"] = os.path.join(tempfile.gettempdir(), "huggingface")
 os.environ["HUGGINGFACE_HUB_CACHE"] = os.path.join(
@@ -39,6 +51,7 @@ try:
     # Check if repo exists
     print(f"Checking if repository {HF_USERNAME}/{DATASET_REPO} exists")
     hf_api.repo_info(repo_id=f"{HF_USERNAME}/{DATASET_REPO}", repo_type="dataset")
+    print(f"Repository {HF_USERNAME}/{DATASET_REPO} exists")
 
     # Download metadata first
     print(f"Downloading metadata from {HF_USERNAME}/{DATASET_REPO}")
@@ -58,8 +71,10 @@ try:
         # Save metadata locally
         with open(METADATA_FILE, "w") as f:
             json.dump(metadata, f)
+            print(f"Metadata saved to {METADATA_FILE}")
     except Exception as e:
         print(f"Error downloading metadata: {e}")
+        print("Creating empty metadata file")
         metadata = {}
         # Initialize metadata file if it doesn't exist
         with open(METADATA_FILE, "w") as f:
@@ -76,10 +91,11 @@ try:
     print(f"Found {len(image_files)} images")
 
     # Download each image
-    for image_file in image_files:
+    success_count = 0
+    for i, image_file in enumerate(image_files):
         try:
             filename = os.path.basename(image_file)
-            print(f"Downloading {filename}")
+            print(f"[{i+1}/{len(image_files)}] Downloading {filename}")
 
             # Download file
             download_path = hf_api.hf_hub_download(
@@ -96,10 +112,13 @@ try:
                 dst.write(src.read())
 
             print(f"Saved {filename} to {destination}")
+            success_count += 1
         except Exception as e:
             print(f"Error downloading {image_file}: {e}")
 
-    print("Image download completed")
+    print(
+        f"Image download completed. Successfully downloaded {success_count}/{len(image_files)} images."
+    )
 
 except Exception as e:
     print(f"Error: {e}")