Update app.py

app.py

@@ -264,6 +264,24 @@ def run_image_generation(
 
 
 
+def get_admin_db(appname: Optional[str]):
+    """
+    Returns (categories_collection, media_clicks_collection)
+    based on appname
+    """
+    if appname == "collage-maker":
+        collage_uri = os.getenv("COLLAGE_MAKER_DB_URL")
+        if not collage_uri:
+            raise RuntimeError("COLLAGE_MAKER_DB_URL not set")
+
+        client = MongoClient(collage_uri)
+        db = client["adminPanel"]
+        return db.categories, db.media_clicks
+
+    # DEFAULT (existing behavior)
+    admin_client = MongoClient(os.getenv("ADMIN_MONGODB_URI"))
+    db = admin_client["adminPanel"]
+    return db.categories, db.media_clicks
 
 # ---------------------------------------------------------------------
 # Endpoints
@@ -294,6 +312,7 @@ async def generate(
     image2: Optional[UploadFile] = File(None),
     user_id: Optional[str] = Form(None),
     category_id: Optional[str] = Form(None),
+    appname: Optional[str] = Form(None),
     user=Depends(verify_firebase_token)
 ):
     start_time = time.time()
@@ -337,9 +356,10 @@ async def generate(
         try:
             admin_client = MongoClient(os.getenv("ADMIN_MONGODB_URI"))
             admin_db = admin_client["adminPanel"]
-    
-            categories_col = admin_db.categories
-            media_clicks_col = admin_db.media_clicks
+            
+            categories_col, media_clicks_col = get_admin_db(appname)
+            # categories_col = admin_db.categories
+            # media_clicks_col = admin_db.media_clicks
     
             # Validate user_oid & category_oid
             user_oid = ObjectId(user_id)
@@ -496,7 +516,8 @@ async def generate(
             "prompt": prompt,
             "user_email": user.get("email"),
             "error": str(e),
-            "response_time_ms": response_time_ms
+            "response_time_ms": response_time_ms,
+            "appname": appname
         })
         raise HTTPException(500, f"Inference failed: {e}")
 
@@ -553,7 +574,8 @@ async def generate(
         "output_id": str(out_id),
         "prompt": prompt,
         "user_email": user.get("email"),
-        "response_time_ms": response_time_ms
+        "response_time_ms": response_time_ms,
+        "appname": appname
     })
 
     return JSONResponse({