Update main.py

main.py

@@ -832,10 +832,30 @@ async def queue_status():
 @app.get("/videos/{category}")
 async def list_videos(category: str = "minecraft"):
     try:
-        videos = await list_videos_from_dataset(category) or [f"mc{i}" for i in range(1, 7)]
+        # Map category aliases to dataset folder names
+        category_map = {
+            "minecraft": "minecraft",
+            "subwaysurfers": "subwaysurfers",
+            "ss": "subwaysurfers",
+        }
+        dataset_category = category_map.get(category.lower(), category.lower())
+        videos = await list_videos_from_dataset(dataset_category)
+        
+        # Fallback to default list if no videos found
+        if not videos:
+            if dataset_category == "subwaysurfers":
+                videos = [f"ss{i}" for i in range(1, 6)]  # ss1 through ss5
+            else:
+                videos = [f"mc{i}" for i in range(1, 7)]
+        
         return {"category": category, "videos": videos, "count": len(videos)}
     except Exception as e:
-        return {"category": category, "videos": [f"mc{i}" for i in range(1, 7)], "count": 6, "error": str(e)}
+        # Fallback defaults
+        if category.lower() in ["subwaysurfers", "ss"]:
+            videos = [f"ss{i}" for i in range(1, 6)]  # ss1 through ss5
+        else:
+            videos = [f"mc{i}" for i in range(1, 7)]
+        return {"category": category, "videos": videos, "count": len(videos), "error": str(e)}
 
 @app.post("/generate")
 async def generate_video(