Update app.py

app.py

@@ -1719,6 +1719,7 @@
 #     uvicorn.run(fastapi_app, host="0.0.0.0", port=7860)
 
 
+# --------------------- List Images Endpoint ---------------------
 # --------------------- List Images Endpoint ---------------------
 import os
 os.environ["OMP_NUM_THREADS"] = "1"
@@ -1913,13 +1914,13 @@ async def startup_db():
 @fastapi_app.on_event("shutdown")
 async def shutdown_db():
     global client, admin_client, collage_maker_client
-    if client:
+    if client is not None:
         client.close()
         logger.info("MongoDB connection closed")
-    if admin_client:
+    if admin_client is not None:
         admin_client.close()
         logger.info("Admin MongoDB connection closed")
-    if collage_maker_client:
+    if collage_maker_client is not None:
         collage_maker_client.close()
         logger.info("Collage Maker MongoDB connection closed")
 
@@ -1941,7 +1942,7 @@ def get_media_clicks_collection(appname: Optional[str] = None):
     if appname:
         normalized = appname.strip().lower()
         if normalized == "collage-maker":
-            if collage_media_clicks_col:
+            if collage_media_clicks_col is not None:
                 return collage_media_clicks_col
             logger.warning("COLLAGE_MAKER_DB_URL not configured; falling back to default media_clicks collection")
     return media_clicks_col
@@ -2302,7 +2303,7 @@ async def face_swap_api(
             # ------------------------------------------------------------------#
             # # MEDIA_CLICKS (ONLY IF user_id PRESENT)
             # ------------------------------------------------------------------#
-            if user_id and media_clicks_collection:
+            if user_id and media_clicks_collection is not None:
                 try:
                     user_id_clean = user_id.strip()
                     if not user_id_clean:
@@ -2462,7 +2463,7 @@ async def face_swap_api(
 
                 except Exception as media_err:
                     logger.error(f"MEDIA_CLICK ERROR: {media_err}")
-            elif user_id and not media_clicks_collection:
+            elif user_id and media_clicks_collection is None:
                 logger.warning("Media clicks collection unavailable; skipping media click tracking")
 
 #         # ------------------------------------------------------------------
@@ -2764,7 +2765,7 @@ async def face_swap_api(
             target_url = asset["url"]
             subcategory_oid = doc["_id"]
 
-            if user_id and media_clicks_collection:
+            if user_id and media_clicks_collection is not None:
                 try:
                     user_id_clean = user_id.strip()
                     if not user_id_clean:
@@ -2897,7 +2898,7 @@ async def face_swap_api(
 
                 except Exception as media_err:
                     logger.error(f"MEDIA_CLICK ERROR: {media_err}")
-            elif user_id and not media_clicks_collection:
+            elif user_id and media_clicks_collection is None:
                 logger.warning("Media clicks collection unavailable; skipping media click tracking")
 
         if target_category_id:
@@ -3084,3 +3085,4 @@ fastapi_app = mount_gradio_app(
 if __name__ == "__main__":
     uvicorn.run(fastapi_app, host="0.0.0.0", port=7860)
 
+