Update main.py

main.py

@@ -577,7 +577,17 @@ async def process_video_task(
                    message="Downloading background video...", queue_position=None)
 
         try:
-            bg_path = await download_video_from_dataset(background, "minecraft")
+            # Auto-detect category from video name
+            # ss1, ss2, ss3, ss4, ss5 → subwaysurfers
+            # mc1, mc2, etc. → minecraft
+            if background.startswith('ss'):
+                category = 'subwaysurfers'
+            elif background.startswith('mc'):
+                category = 'minecraft'
+            else:
+                category = 'minecraft'  # default fallback
+            
+            bg_path = await download_video_from_dataset(background, category)
         except HTTPException as e:
             update_job(job_id, status="failed", error=str(e.detail))
             return