f

.gitignore

@@ -0,0 +1,15 @@
+# Environments
+.env
+venv/
+__pycache__/
+
+# Local Storage (prevents uploading test photos to git)
+data/
+
+# Node
+node_modules/
+.expo/
+
+# OS Files
+.DS_Store
+Thumbs.db

keystone-backend/main.py

@@ -1,28 +1,35 @@
 import os
-import shutil
 from fastapi import FastAPI, HTTPException, Header, Depends, UploadFile, File
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel
 from dotenv import load_dotenv
 
-# Load environment variables
+# Load local environment variables
 load_dotenv()
 
 app = FastAPI(title="KeyStone API", version="0.1.0")
 
-# Fetch credentials
+# Fetch credentials from environment variables
 DEV_USERNAME = os.getenv("DEV_USERNAME", "admin")
 DEV_PASSWORD = os.getenv("DEV_LOGIN_PASSWORD", "password123")
 MOCK_TOKEN = "mock-jwt-token-xyz123"
 
-# --- PERSISTENT STORAGE SETUP ---
-# If running in HF Spaces with persistent storage, it mounts to /data.
-# Locally, we will default to creating a folder called "data/photos" in the project root.
-BASE_STORAGE_DIR = os.getenv("STORAGE_DIR", "./data/photos")
+# --- SMART BUCKET STORAGE SETUP ---
+# If the root "/data" folder exists, Hugging Face mounted the bucket.
+if os.path.exists("/data"):
+    BASE_STORAGE_DIR = "/data"
+else:
+    # Otherwise, fall back to a local folder for Windows testing.
+    BASE_STORAGE_DIR = os.getenv("STORAGE_DIR", "./data")
 
-# Ensure the directory exists when the server starts
-os.makedirs(BASE_STORAGE_DIR, exist_ok=True)
-# --------------------------------
+# Define your specific bucket paths
+ORIGINALS_DIR = os.path.join(BASE_STORAGE_DIR, "originals")
+THUMBNAILS_DIR = os.path.join(BASE_STORAGE_DIR, "thumbnails")
+
+# Ensure these directories exist when the server starts
+os.makedirs(ORIGINALS_DIR, exist_ok=True)
+os.makedirs(THUMBNAILS_DIR, exist_ok=True)
+# ----------------------------------
 
 app.add_middleware(
     CORSMiddleware,
@@ -47,7 +54,11 @@ def verify_mock_token(authorization: str = Header(None)):
 
 @app.get("/health")
 async def health_check():
-    return {"status": "ok", "service": "KeyStone API", "storage_path": BASE_STORAGE_DIR}
+    return {
+        "status": "ok", 
+        "service": "KeyStone API", 
+        "storage_path": BASE_STORAGE_DIR
+    }
 
 @app.post("/auth/login")
 async def mock_login(payload: LoginPayload):
@@ -59,22 +70,21 @@ async def mock_login(payload: LoginPayload):
         }
     raise HTTPException(status_code=400, detail="Incorrect username or password")
 
-# --- NEW UPLOAD ENDPOINT ---
 @app.post("/upload")
 async def upload_photo(file: UploadFile = File(...), token: str = Depends(verify_mock_token)):
-    # Create the full file path safely
-    file_path = os.path.join(BASE_STORAGE_DIR, file.filename)
+    # Save directly to the originals folder
+    file_path = os.path.join(ORIGINALS_DIR, file.filename)
     
     try:
-        # Save the file in chunks to keep memory usage low
+        # Save in 1MB chunks to prevent memory spikes
         with open(file_path, "wb") as buffer:
-            while content := await file.read(1024 * 1024):  # Read in 1MB chunks
+            while content := await file.read(1024 * 1024): 
                 buffer.write(content)
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"Could not save file: {str(e)}")
         
     return {
-        "message": "Photo uploaded successfully!",
+        "message": "Photo securely uploaded!",
         "filename": file.filename,
         "path": file_path
     }