Update app.py

app.py

@@ -506,6 +506,14 @@ async def face_swap_api(
             selected_media_clicks_col = ai_enhancer_media_clicks_col
         elif normalized_appname == "collage-maker" and collage_media_clicks_col is not None:
             selected_media_clicks_col = collage_media_clicks_col
+        # ---------------------------------------------------------
+        # ACTIVE SUBCATEGORY COLLECTION (CRITICAL FIX)
+        # ---------------------------------------------------------
+        if normalized_appname == "ai-enhancer" and ai_enhancer_db is not None:
+            active_subcategories_col = ai_enhancer_db.subcategories
+        else:
+            active_subcategories_col = subcategories_col
+
 
         # ---------------------------------------------------------
         # REQUEST TRACE LOG
@@ -573,11 +581,16 @@ async def face_swap_api(
                 raise HTTPException(400, "Invalid new_subcategory_id format.")
 
             # 1. Find subcategory asset by asset_images._id
-            subcat_doc = await subcategories_col.find_one(
+            # subcat_doc = await subcategories_col.find_one(
+            #     {"asset_images._id": asset_oid},
+            #     {"asset_images.$": 1} # Only need the asset image URL
+            # )
+            subcat_doc = await active_subcategories_col.find_one(
                 {"asset_images._id": asset_oid},
-                {"asset_images.$": 1} # Only need the asset image URL
+                {"asset_images.$": 1}
             )
 
+
             if not subcat_doc or "asset_images" not in subcat_doc:
                 await log_faceswap_hit(user_email, "error", start_time, datetime.now(timezone.utc), appname=normalized_appname)
                 raise HTTPException(