1

video-gen-api/app/tasks/shot_replicate_tasks.py

@@ -41,6 +41,7 @@ from app.services.shot_video_analysis_service import analyze_video_for_shot_spli
 from app.services.generation_billing_service import charge_shot_video_analysis_usage
 from app.services.shot_video_split_service import split_video_segment_async
 from app.services.upload_video_asset_service import validate_split_range
+from app.services.upload_resource import record_shot_segment_upload_resource
 from app.tasks.async_runner import run_async
 from app.tasks.celery_app import celery_app
 
@@ -467,6 +468,13 @@ async def _run_split_one_segment(segment_id: str) -> None:
                 return
             segment.segment_video_url = split_result.url
             segment.segment_video_path = split_result.path
+            await record_shot_segment_upload_resource(
+                db,
+                segment=segment,
+                storage_path=split_result.path,
+                resource_url=split_result.url,
+                file_size_bytes=split_result.file_size_bytes,
+            )
             segment.split_status = ShotSplitStatusEnum.COMPLETED.value
             segment.split_completed_at = _now()
             segment.split_lease_until = None