Update main.py

main.py

@@ -33,17 +33,15 @@ _POOL_MAX = 64
 
 
 def _get_pinecone(api_key: str) -> Pinecone:
-    """Return a cached Pinecone client, creating one if needed."""
     if api_key not in _pinecone_pool:
         if len(_pinecone_pool) >= _POOL_MAX:
-            _pinecone_pool.popitem(last=False)   # evict oldest
+            _pinecone_pool.popitem(last=False)
         _pinecone_pool[api_key] = Pinecone(api_key=api_key)
-    _pinecone_pool.move_to_end(api_key)          # refresh LRU order
+    _pinecone_pool.move_to_end(api_key)
     return _pinecone_pool[api_key]
 
 
 def _configure_cloudinary(creds: dict) -> None:
-    """Configure cloudinary module only when needed, with simple caching."""
     key = creds["cloud_name"]
     if key not in _cloudinary_pool:
         cloudinary.config(
@@ -54,7 +52,6 @@ def _configure_cloudinary(creds: dict) -> None:
         _cloudinary_pool[key] = True
 
 
-# ── Lifespan: load models once at startup ─────────────────────────
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     global ai, _inference_sem
@@ -73,7 +70,7 @@ app = FastAPI(lifespan=lifespan)
 
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["*"],        # tighten to your Vercel domain in production
+    allow_origins=["*"],
     allow_credentials=True,
     allow_methods=["*"],
     allow_headers=["*"],
@@ -82,7 +79,6 @@ app.add_middleware(
 os.makedirs("temp_uploads", exist_ok=True)
 
 
-# ── Helpers ────────────────────────────────────────────────────────
 def standardize_category_name(name: str) -> str:
     clean = re.sub(r'\s+', '_', name.strip().lower())
     clean = re.sub(r'[^\w]', '', clean)
@@ -102,7 +98,6 @@ def get_cloudinary_creds(env_url: str) -> dict:
         "cloud_name": parsed.hostname,
     }
 
-
 # ══════════════════════════════════════════════════════════════════
 # 1. VERIFY KEYS & AUTO-BUILD INDEXES
 # ══════════════════════════════════════════════════════════════════
@@ -150,7 +145,7 @@ async def verify_keys(
 
 
 # ══════════════════════════════════════════════════════════════════
-# 2. UPLOAD (Cloudinary + Pinecone Only)
+# 2. UPLOAD (With Demo Fallback)
 # ══════════════════════════════════════════════════════════════════
 @app.post("/api/upload")
 async def upload_new_images(
@@ -160,15 +155,19 @@ async def upload_new_images(
     user_pinecone_key:   str  = Form(""),
     user_cloudinary_url: str  = Form(""),
 ):
-    if not user_pinecone_key or not user_cloudinary_url:
-        raise HTTPException(status_code=400, detail="Cloudinary URL and Pinecone API Key are required to upload.")
+    # FALLBACK LOGIC: Use user keys if provided, otherwise use Space secrets
+    actual_pc_key = user_pinecone_key or os.getenv("DEFAULT_PINECONE_KEY")
+    actual_cld_url = user_cloudinary_url or os.getenv("DEFAULT_CLOUDINARY_URL")
+
+    if not actual_pc_key or not actual_cld_url:
+        raise HTTPException(status_code=400, detail="Cloudinary URL and Pinecone API Key are required.")
 
     folder    = standardize_category_name(folder_name)
     uploaded_urls = []
 
-    cld_creds = get_cloudinary_creds(user_cloudinary_url)
+    cld_creds = get_cloudinary_creds(actual_cld_url)
     _configure_cloudinary(cld_creds)
-    pc        = _get_pinecone(user_pinecone_key)
+    pc        = _get_pinecone(actual_pc_key)
     idx_obj   = pc.Index("lens-objects")
     idx_face  = pc.Index("lens-faces")
 
@@ -202,7 +201,6 @@ async def upload_new_images(
                 }
                 (face_upserts if v["type"] == "face" else object_upserts).append(record)
 
-            # Fire both upserts concurrently
             upsert_tasks = []
             if face_upserts:
                 upsert_tasks.append(asyncio.to_thread(idx_face.upsert, vectors=face_upserts))
@@ -213,7 +211,6 @@ async def upload_new_images(
 
         except Exception as e:
             print(f"❌ Upload error for {file.filename}: {e}")
-            # Continue with the next file instead of aborting the whole batch
         finally:
             if os.path.exists(tmp_path):
                 os.remove(tmp_path)
@@ -222,16 +219,18 @@ async def upload_new_images(
 
 
 # ══════════════════════════════════════════════════════════════════
-# 3. SEARCH (Pinecone Only)
+# 3. SEARCH (With Demo Fallback)
 # ══════════════════════════════════════════════════════════════════
 @app.post("/api/search")
 async def search_database(
     file:                UploadFile = File(...),
     detect_faces:        bool = Form(True),
     user_pinecone_key:   str  = Form(""),
-    user_cloudinary_url: str  = Form(""), # Kept to match frontend form payload
+    user_cloudinary_url: str  = Form(""),
 ):
-    if not user_pinecone_key:
+    actual_pc_key = user_pinecone_key or os.getenv("DEFAULT_PINECONE_KEY")
+    
+    if not actual_pc_key:
         raise HTTPException(status_code=400, detail="Pinecone API Key is required to search.")
 
     safe_name = sanitize_filename(file.filename)
@@ -245,11 +244,10 @@ async def search_database(
         async with _inference_sem:
             vectors = await ai.process_image_async(tmp_path, is_query=True, detect_faces=detect_faces)
 
-        pc       = _get_pinecone(user_pinecone_key)
+        pc       = _get_pinecone(actual_pc_key)
         idx_obj  = pc.Index("lens-objects")
         idx_face = pc.Index("lens-faces")
 
-        # Fire ALL vector queries in parallel
         async def _query_one(vec_dict: dict) -> list[dict]:
             vec_list   = (vec_dict["vector"].tolist() if hasattr(vec_dict["vector"], "tolist") else vec_dict["vector"])
             target_idx = idx_face if vec_dict["type"] == "face" else idx_obj
@@ -270,7 +268,6 @@ async def search_database(
         nested = await asyncio.gather(*[_query_one(v) for v in vectors])
         all_results = [r for sub in nested for r in sub]
 
-        # Deduplicate, keep best score per URL
         seen: dict[str, dict] = {}
         for r in all_results:
             url = r["url"]
@@ -289,15 +286,17 @@ async def search_database(
 
 
 # ══════════════════════════════════════════════════════════════════
-# 4. CATEGORIES (Cloudinary Folders Only)
+# 4. CATEGORIES (With Demo Fallback)
 # ══════════════════════════════════════════════════════════════════
 @app.post("/api/categories")
 async def get_categories(user_cloudinary_url: str = Form("")):
-    if not user_cloudinary_url:
+    actual_cld_url = user_cloudinary_url or os.getenv("DEFAULT_CLOUDINARY_URL")
+
+    if not actual_cld_url:
         return {"categories": []}
     
     try:
-        creds = get_cloudinary_creds(user_cloudinary_url)
+        creds = get_cloudinary_creds(actual_cld_url)
         _configure_cloudinary(creds)
         result  = await asyncio.to_thread(cloudinary.api.root_folders)
         folders = [f["name"] for f in result.get("folders", [])]