Add AI-Enhancer MongoDB support with environment variables and case-insensitive appname handling

api/main.py

@@ -108,6 +108,15 @@ collage_maker_admin_db = None
 collage_maker_media_clicks = None
 collage_maker_categories = None
 
+# AI-Enhancer MongoDB configuration
+AI_ENHANCER_MONGO_URI = os.environ.get("MONGODB_AI_ENHANCER")
+AI_ENHANCER_DB_NAME = os.environ.get("MONGODB_AI_ENHANCER_DB_NAME", "ai-enhancer")
+AI_ENHANCER_ADMIN_DB_NAME = os.environ.get("MONGODB_AI_ENHANCER_ADMIN_DB_NAME", "test")
+ai_enhancer_client = None
+ai_enhancer_db = None
+ai_enhancer_admin_db = None
+ai_enhancer_media_clicks = None
+
 
 def get_collage_maker_client() -> Optional[MongoClient]:
     """Get collage-maker MongoDB client."""
@@ -165,6 +174,59 @@ def _init_collage_maker_mongo() -> None:
 _init_collage_maker_mongo()
 
 
+def get_ai_enhancer_client() -> Optional[MongoClient]:
+    """Get AI-Enhancer MongoDB client."""
+    global ai_enhancer_client
+    if ai_enhancer_client is None and AI_ENHANCER_MONGO_URI:
+        try:
+            ai_enhancer_client = MongoClient(AI_ENHANCER_MONGO_URI)
+            log.info("AI-Enhancer MongoDB client initialized")
+        except Exception as err:
+            log.error("Failed to initialize AI-Enhancer MongoDB client: %s", err)
+            ai_enhancer_client = None
+    return ai_enhancer_client
+
+
+def get_ai_enhancer_database() -> Optional[Any]:
+    """Get AI-Enhancer database instance."""
+    global ai_enhancer_db
+    client = get_ai_enhancer_client()
+    if client is None:
+        return None
+    if ai_enhancer_db is None:
+        try:
+            ai_enhancer_db = client[AI_ENHANCER_DB_NAME]
+            log.info("AI-Enhancer database initialized: %s", AI_ENHANCER_DB_NAME)
+        except Exception as err:
+            log.error("Failed to get AI-Enhancer database: %s", err)
+            ai_enhancer_db = None
+    return ai_enhancer_db
+
+
+def _init_ai_enhancer_mongo() -> None:
+    """Initialize AI-Enhancer MongoDB connections."""
+    global ai_enhancer_admin_db, ai_enhancer_media_clicks
+    client = get_ai_enhancer_client()
+    if client is None:
+        log.info("AI-Enhancer Mongo URI not provided; AI-Enhancer features disabled")
+        return
+    try:
+        ai_enhancer_admin_db = client[AI_ENHANCER_ADMIN_DB_NAME]
+        ai_enhancer_media_clicks = ai_enhancer_admin_db["media_clicks"]
+        log.info(
+            "AI-Enhancer admin initialized: db=%s, media_clicks=%s",
+            AI_ENHANCER_ADMIN_DB_NAME,
+            ai_enhancer_media_clicks.name,
+        )
+    except Exception as err:
+        log.error("Failed to init AI-Enhancer admin Mongo: %s", err)
+        ai_enhancer_admin_db = None
+        ai_enhancer_media_clicks = None
+
+
+_init_ai_enhancer_mongo()
+
+
 def get_category_id_from_collage_maker() -> Optional[str]:
     """Query category ID from collage-maker categories collection."""
     if collage_maker_categories is None:
@@ -441,14 +503,22 @@ def log_media_click(user_id: Optional[str], category_id: Optional[str], appname:
     """Log to admin media_clicks collection only if user_id is provided.
     
     If appname='collage-maker', logs to collage-maker MongoDB instead of regular admin MongoDB.
+    If appname='AI-Enhancer' (case-insensitive), logs to AI-Enhancer MongoDB.
     """
     # Determine which media_clicks collection to use
     target_media_clicks = None
-    if appname == "collage-maker":
+    appname_lower = appname.lower() if appname else None
+    
+    if appname_lower == "collage-maker":
         target_media_clicks = collage_maker_media_clicks
         if target_media_clicks is None:
             log.warning("Collage-maker media_clicks not initialized, skipping log")
             return
+    elif appname_lower == "ai-enhancer":
+        target_media_clicks = ai_enhancer_media_clicks
+        if target_media_clicks is None:
+            log.warning("AI-Enhancer media_clicks not initialized, skipping log")
+            return
     else:
         target_media_clicks = admin_media_clicks
         if target_media_clicks is None:
@@ -533,7 +603,13 @@ def log_media_click(user_id: Optional[str], category_id: Optional[str], appname:
             )
     except Exception as err:
         err_str = str(err)
-        target_name = "collage-maker" if appname == "collage-maker" else "admin"
+        appname_lower = appname.lower() if appname else None
+        if appname_lower == "collage-maker":
+            target_name = "collage-maker"
+        elif appname_lower == "ai-enhancer":
+            target_name = "AI-Enhancer"
+        else:
+            target_name = "admin"
         if "Unauthorized" in err_str or "not authorized" in err_str.lower():
             log.warning(
                 "%s media click logging failed (permissions): user lacks read/write on db=%s collection=%s. "