feat: add video job cancellation functionality and error tracking in generated videos

Co-authored-by: Copilot <copilot@github.com>

backend/app/db.py

@@ -121,6 +121,9 @@ def _run_migrations(conn: duckdb.DuckDBPyConnection) -> None:
     conn.execute("""
         ALTER TABLE generated_videos ADD COLUMN IF NOT EXISTS generation_type VARCHAR DEFAULT 'text_to_video'
     """)
+    conn.execute("""
+        ALTER TABLE generated_videos ADD COLUMN IF NOT EXISTS error VARCHAR
+    """)
     _seed_admin(conn)
 
 

backend/app/routers/generate.py

@@ -322,7 +322,7 @@ async def list_generated_videos(
     user_id = current_user.get("id") or current_user.get("sub")
     conn = get_conn()
     rows = conn.execute(
-        """SELECT id, job_id, model_id, prompt, polling_url, status, video_url, created_at
+        """SELECT id, job_id, model_id, prompt, polling_url, status, video_url, error, created_at
            FROM generated_videos
            WHERE user_id = ?
            ORDER BY created_at DESC""",
@@ -337,7 +337,8 @@ async def list_generated_videos(
             "polling_url": r[4],
             "status": r[5],
             "video_url": r[6],
-            "created_at": r[7].isoformat() if r[7] else None,
+            "error": r[7],
+            "created_at": r[8].isoformat() if r[8] else None,
         }
         for r in rows
     ]
@@ -352,7 +353,7 @@ async def get_generated_video(
     user_id = current_user.get("id") or current_user.get("sub")
     conn = get_conn()
     row = conn.execute(
-        """SELECT id, job_id, model_id, prompt, polling_url, status, video_url, created_at, updated_at
+        """SELECT id, job_id, model_id, prompt, polling_url, status, video_url, error, created_at, updated_at
            FROM generated_videos
            WHERE id = ? AND user_id = ?""",
         [video_id, user_id],
@@ -367,6 +368,36 @@ async def get_generated_video(
         "polling_url": row[4],
         "status": row[5],
         "video_url": row[6],
-        "created_at": row[7].isoformat() if row[7] else None,
-        "updated_at": row[8].isoformat() if row[8] else None,
+        "error": row[7],
+        "created_at": row[8].isoformat() if row[8] else None,
+        "updated_at": row[9].isoformat() if row[9] else None,
     }
+
+
+@router.post("/videos/{video_id}/cancel", status_code=200)
+async def cancel_video_job(
+    video_id: str,
+    current_user: dict = Depends(get_current_user),
+) -> dict[str, str]:
+    """Mark a video job as 'cancelled' if it belongs to the current user and is not terminal."""
+    user_id = current_user.get("id") or current_user.get("sub")
+    conn = get_conn()
+    row = conn.execute(
+        "SELECT status FROM generated_videos WHERE id = ? AND user_id = ?",
+        [video_id, user_id],
+    ).fetchone()
+    if not row:
+        raise HTTPException(status_code=404, detail="Video job not found")
+    job_status = row[0]
+    if job_status in ("completed", "failed", "cancelled"):
+        raise HTTPException(
+            status_code=status.HTTP_400_BAD_REQUEST,
+            detail=f"Cannot cancel job with status '{job_status}'",
+        )
+    now = datetime.now(timezone.utc).replace(tzinfo=None)
+    async with get_write_lock():
+        conn.execute(
+            "UPDATE generated_videos SET status = 'cancelled', updated_at = ? WHERE id = ?",
+            [now, video_id],
+        )
+    return {"status": "ok", "job_id": video_id}

backend/app/services/video_worker.py

@@ -60,8 +60,8 @@ async def process_queued_jobs(conn: duckdb.DuckDBPyConnection, lock: asyncio.Loc
             now = datetime.now(timezone.utc).replace(tzinfo=None)
             async with lock:
                 conn.execute(
-                    "UPDATE generated_videos SET status = 'failed', updated_at = ? WHERE id = ?",
-                    [now, db_id],
+                    "UPDATE generated_videos SET status = 'failed', error = ?, updated_at = ? WHERE id = ?",
+                    [str(exc), now, db_id],
                 )
             continue
 
@@ -116,14 +116,15 @@ async def process_processing_jobs(conn: duckdb.DuckDBPyConnection, lock: asyncio
 
         urls = result.get("unsigned_urls") or result.get("video_urls")
         video_url = (urls or [None])[0]
+        error_msg = result.get("error")
         now = datetime.now(timezone.utc).replace(tzinfo=None)
 
         async with lock:
             conn.execute(
                 """UPDATE generated_videos
-                   SET status = ?, video_url = ?, updated_at = ?
+                   SET status = ?, video_url = ?, error = ?, updated_at = ?
                    WHERE id = ?""",
-                [job_status, video_url, now, db_id],
+                [job_status, video_url, error_msg, now, db_id],
             )
         updated += 1
         logger.info("Video job %s → %s", db_id, job_status)