Update app.py

app.py

@@ -117,7 +117,12 @@ def normalize_category_id(raw_category_id):
     if value == "":
         return DEFAULT_CATEGORY_FALLBACK, "default_empty"
 
-    return value, "provided"
+    # Always store categoryId as an ObjectId in Mongo
+    try:
+        return ObjectId(value), "provided_objectid"
+    except Exception:
+        # If somehow not a valid hex string, fall back to hashing
+        return _hash_to_object_id(value), "provided_hashed"
 
 ##-------NEW CATEGORY CLICK-------#####
 def log_media_click(raw_user_id, raw_category_id):
@@ -202,7 +207,26 @@ def log_media_click(raw_user_id, raw_category_id):
         )
 
         # --------------------------------------------------
-        # STEP 3: UPDATE EXISTING CATEGORY (DATES CAN REPEAT)
+        # STEP 3: NORMALIZE OLD STRING CATEGORY IDs TO ObjectId
+        # --------------------------------------------------
+        # If there is an existing category entry where categoryId was stored
+        # as a string, convert that field to an ObjectId so future updates
+        # always work with a consistent type.
+        category_id_str = str(category_id)
+        media_clicks_collection.update_one(
+            {
+                "userId": user_oid,
+                "categories.categoryId": category_id_str,
+            },
+            {
+                "$set": {
+                    "categories.$.categoryId": category_id,
+                }
+            },
+        )
+
+        # --------------------------------------------------
+        # STEP 4: UPDATE EXISTING CATEGORY (DATES CAN REPEAT)
         # --------------------------------------------------
         update_existing = media_clicks_collection.update_one(
             {
@@ -226,7 +250,7 @@ def log_media_click(raw_user_id, raw_category_id):
             return
 
         # --------------------------------------------------
-        # STEP 4: PUSH NEW CATEGORY (ORDER = TIME)
+        # STEP 5: PUSH NEW CATEGORY (ORDER = TIME)
         # --------------------------------------------------
         media_clicks_collection.update_one(
             {"userId": user_oid},
@@ -579,7 +603,6 @@ def download(filename: str):
     return FileResponse(file_path, media_type="image/png", filename=filename)
 
 
-
 # import hashlib
 # import logging
 # import os