Update app.py

app.py

@@ -870,61 +870,7 @@ class Upscale:
         if is_success:
             im_buf_arr.tofile(save_path)
             
-    def sync_log_media_click(user_id_str: str, category_id_str: str):
-        """
-        Synchronously logs a click event to the media_clicks collection.
-        This function should be run using asyncio.to_thread().
-        """
-        if _media_clicks_col is None:
-            return
-    
-        try:
-            user_oid = ObjectId(user_id_str.strip())
-            category_oid = ObjectId(category_id_str.strip())
-            now = datetime.utcnow()
-            logger.info(f"Attempting background write for User:{user_id_str}, Category:{category_id_str}") # <--- NEW LOG
     
-            # 1. Try updating an existing category entry within the user's document
-            update_result = _media_clicks_col.update_one(
-                {
-                    "userId": user_oid,
-                    "categories.categoryId": category_oid
-                },
-                {
-                    "$set": {
-                        "updatedAt": now,
-                        "categories.$.lastClickedAt": now
-                    },
-                    "$inc": {
-                        "categories.$.click_count": 1
-                    }
-                }
-            )
-    
-            # 2. If no category entry exists (matched_count == 0) -> push a new one (or create user doc)
-            if update_result.matched_count == 0:
-                _media_clicks_col.update_one(
-                    { "userId": user_oid },
-                    {
-                        "$setOnInsert": { "createdAt": now },
-                        "$set": { "updatedAt": now },
-                        "$push": {
-                            "categories": {
-                                "categoryId": category_oid,
-                                "click_count": 1,
-                                "lastClickedAt": now
-                            }
-                        }
-                    },
-                    upsert=True
-                )
-            
-            logger.info(f"Media click logged for User {user_id_str} on Category {category_id_str}")
-            
-        except Exception as media_err:
-            logger.error(f"MEDIA_CLICK LOGGING ERROR: {media_err}")
-            pass # Handle logging error gracefully
-
 class Timer:
     def __init__(self):
         self.start_time = time.perf_counter()
@@ -992,6 +938,60 @@ def _verify_bearer_token(credentials: Optional[HTTPAuthorizationCredentials] = D
     if not _api_bearer_token or token != _api_bearer_token:
         raise HTTPException(status_code=status.HTTP_401_UNAUTHORIZED, detail="Invalid token")
     return True
+def sync_log_media_click(user_id_str: str, category_id_str: str):
+        """
+        Synchronously logs a click event to the media_clicks collection.
+        This function should be run using asyncio.to_thread().
+        """
+        if _media_clicks_col is None:
+            return
+    
+        try:
+            user_oid = ObjectId(user_id_str.strip())
+            category_oid = ObjectId(category_id_str.strip())
+            now = datetime.utcnow()
+            logger.info(f"Attempting background write for User:{user_id_str}, Category:{category_id_str}") # <--- NEW LOG
+    
+            # 1. Try updating an existing category entry within the user's document
+            update_result = _media_clicks_col.update_one(
+                {
+                    "userId": user_oid,
+                    "categories.categoryId": category_oid
+                },
+                {
+                    "$set": {
+                        "updatedAt": now,
+                        "categories.$.lastClickedAt": now
+                    },
+                    "$inc": {
+                        "categories.$.click_count": 1
+                    }
+                }
+            )
+    
+            # 2. If no category entry exists (matched_count == 0) -> push a new one (or create user doc)
+            if update_result.matched_count == 0:
+                _media_clicks_col.update_one(
+                    { "userId": user_oid },
+                    {
+                        "$setOnInsert": { "createdAt": now },
+                        "$set": { "updatedAt": now },
+                        "$push": {
+                            "categories": {
+                                "categoryId": category_oid,
+                                "click_count": 1,
+                                "lastClickedAt": now
+                            }
+                        }
+                    },
+                    upsert=True
+                )
+            
+            logger.info(f"Media click logged for User {user_id_str} on Category {category_id_str}")
+            
+        except Exception as media_err:
+            logger.error(f"MEDIA_CLICK LOGGING ERROR: {media_err}")
+            pass # Handle logging error gracefully
 
 DEFAULT_FACE_MODEL = 'GFPGANv1.4.pth'
 DEFAULT_UPSCALE_MODEL = 'SRVGG, realesr-general-x4v3.pth'