Update app.py

app.py

@@ -85,39 +85,44 @@ _mongo_collection = None
 _admin_mongo_client = None
 _media_clicks_col = None
 
+import os
+
 try:
     from pymongo import MongoClient
-    # 1. API Logs/Submits Connection (existing logic)
+
+    # 1. API Logs/Submits Connection
     _mongo_uri = os.getenv("MONGODB_URI")
     _mongo_db = os.getenv("MONGODB_DB", "face_upscale")
     _mongo_col = os.getenv("MONGODB_COLLECTION", "submits")
+
     if _mongo_uri:
-        # Use existing client setup if you have one
-        pass
-        
-    # 2. Media Clicks Connection (using MONGODB_ADMIN_URI)
+        logger.info(f"Connecting to MongoDB for API logs: {_mongo_db}/{_mongo_col}")
+        _mongo_client = MongoClient(_mongo_uri, connect=False)
+        _mongo_collection = _mongo_client[_mongo_db][_mongo_col]
+    else:
+        logger.warning("MONGODB_URI not found → API logs will NOT be saved.")
+
+    # 2. Media Clicks Connection
     _admin_mongo_uri = os.getenv("MONGODB_ADMIN_URI")
-    _admin_mongo_db = os.getenv("MONGODB_ADMIN_DB", "adminPanel") 
-    _media_clicks_col_name = "media_clicks" 
-    
+    _admin_mongo_db = os.getenv("MONGODB_ADMIN_DB", "adminPanel")
+    _media_clicks_col_name = "media_clicks"
+
     if _admin_mongo_uri:
-        # --- CRITICAL LOGGING ADDED HERE ---
-        logger.info(f"Attempting connection to Admin DB: {_admin_mongo_db}...") 
-        # -----------------------------------
+        logger.info(f"Attempting connection to Admin DB: {_admin_mongo_db}...")
         _admin_mongo_client = MongoClient(_admin_mongo_uri, connect=False)
         _media_clicks_col = _admin_mongo_client[_admin_mongo_db][_media_clicks_col_name]
         logger.info("Admin MongoDB client established.")
+    else:
+        logger.warning("MONGODB_ADMIN_URI not provided → media clicks will NOT be saved.")
 
 except Exception as e:
-    # --- CRITICAL ERROR CATCHING ADDED HERE ---
-    logger.error(f"FATAL: Admin MongoDB connection failed. Media clicks will not be saved. Error: {e}")
-    # ------------------------------------------
-    # Ensure all related variables are None on failure
+    logger.error(f"FATAL: MongoDB connection failed. Error: {e}")
     _mongo_client = None
     _mongo_collection = None
     _admin_mongo_client = None
     _media_clicks_col = None
 
+
 input_images_limit = 5
 
 # Define URLs and their corresponding local storage paths