Update app.py

app.py

@@ -31,7 +31,7 @@ INDEX_FILE = os.path.join(DATA_ROOT, "index.json")
 os.makedirs(VIDEO_DIR, exist_ok=True)
 
 # --------------------------------------------------
-# Album → Publisher mapping
+# Album → Publisher mapping (who made it)
 # --------------------------------------------------
 ALBUM_PUBLISHERS = {
     "B2c5n8hH4uWRoAW": "Alex Rose",
@@ -42,6 +42,18 @@ ALBUM_PUBLISHERS = {
     "B2c5ON9t3uz8kT7": "Cole Vandepoll",
 }
 
+# --------------------------------------------------
+# Album → Category mapping (what kind of content it is)
+# --------------------------------------------------
+ALBUM_CATEGORIES = {
+    "B2c5n8hH4uWRoAW": "Fun",
+    "B2c5yeZFhHXzdFg": "Rockets",
+    "B2cGI9HKKtaAF3T": "Sam Content Library",
+    "B2c59UlCquMMGkJ": "Serious",
+    "B2cJ0DiRHusi12z": "Music",
+    "B2c5ON9t3uz8kT7": "Cole Content Creator",  # "Same as Cole's profile"
+}
+
 # --------------------------------------------------
 # HTTP client
 # --------------------------------------------------
@@ -129,6 +141,39 @@ def save_index(data: dict):
     with open(INDEX_FILE, "w", encoding="utf-8") as f:
         json.dump(data, f, indent=2)
 
+def backfill_index_categories():
+    """
+    Fix previously cached entries that used the old bug:
+    category == publisher (or category missing).
+
+    After this runs once, your /feed/videos will be correct even
+    if you already had index.json created before the category fix.
+    """
+    try:
+        idx = load_index()
+        changed = False
+
+        for v in idx.get("videos", []):
+            token = v.get("source_album", "")
+            correct_category = ALBUM_CATEGORIES.get(token, "Uncategorized")
+            correct_publisher = ALBUM_PUBLISHERS.get(token, "Unknown")
+
+            # Ensure publisher is correct too (safe cleanup)
+            if v.get("publisher") in (None, "", "Unknown") and correct_publisher != "Unknown":
+                v["publisher"] = correct_publisher
+                changed = True
+
+            # Fix category if missing or wrong
+            if v.get("category") in (None, "", v.get("publisher")) or v.get("category") != correct_category:
+                v["category"] = correct_category
+                changed = True
+
+        if changed:
+            save_index(idx)
+            logging.info("Backfilled index.json categories/publishers")
+    except Exception:
+        logging.exception("Backfill failed (non-fatal)")
+
 # --------------------------------------------------
 # Downloader
 # --------------------------------------------------
@@ -155,7 +200,7 @@ async def poll_album(token: str):
     known = {v["id"] for v in index["videos"]}
 
     publisher = ALBUM_PUBLISHERS.get(token, "Unknown")
-    category = publisher
+    category = ALBUM_CATEGORIES.get(token, "Uncategorized")
 
     album_dir = os.path.join(VIDEO_DIR, token)
     os.makedirs(album_dir, exist_ok=True)
@@ -212,6 +257,9 @@ async def poll_album(token: str):
 # --------------------------------------------------
 @app.on_event("startup")
 async def start_polling():
+    # Fix any old cached feed entries first
+    backfill_index_categories()
+
     async def loop():
         while True:
             for token in ALBUM_PUBLISHERS: