Update app.py

app.py

@@ -269,6 +269,7 @@ def get_admin_db(appname: Optional[str]):
     Returns (categories_collection, media_clicks_collection)
     based on appname
     """
+     # ---- Collage Maker ----
     if appname == "collage-maker":
         collage_uri = os.getenv("COLLAGE_MAKER_DB_URL")
         if not collage_uri:
@@ -278,6 +279,16 @@ def get_admin_db(appname: Optional[str]):
         db = client["adminPanel"]
         return db.categories, db.media_clicks
 
+     # ---- AI ENHANCER ----
+    if appname == "AI-Enhancer":
+        enhancer_uri = os.getenv("AI_ENHANCER_DB_URL")
+        if not enhancer_uri:
+            raise RuntimeError("AI_ENHANCER_DB_URL not set")
+
+        client = MongoClient(enhancer_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"]