# moviepy.video.io.ffmpeg_tools.ffmpeg_extract_subclip

def ffmpeg_extract_subclip(filename, t1, t2, targetname=None):
    """ Makes a new video file playing video file ``filename`` between
        the times ``t1`` and ``t2``. """
    name, ext = os.path.splitext(filename)
    if not targetname:
        T1, T2 = [int(1000*t) for t in [t1, t2]]
        targetname = "%sSUB%d_%d.%s" % (name, T1, T2, ext)
    
    cmd = [get_setting("FFMPEG_BINARY"),"-y",
           "-ss", "%0.2f"%t1,
           "-i", filename,
           "-t", "%0.2f"%(t2-t1),
           "-map", "0", "-vcodec", "copy", "-acodec", "copy", targetname]

    subprocess_call(cmd)

subclip = VideoUrl("<subclip_path>")

tensors = subclip.load()

---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
Cell In[21], line 1
----> 1 tensors = subclip.load()

File ~/Projects/chrisammon3000/experiments/docarray/docarray-test/.venv/lib/python3.11/site-packages/docarray/typing/url/video_url.py:96, in VideoUrl.load(self, **kwargs)
     33 """
     34 Load the data from the url into a `NamedTuple` of
     35 [`VideoNdArray`][docarray.typing.VideoNdArray],
   (...)
     93     [`NdArray`][docarray.typing.NdArray] of the key frame indices.
     94 """
     95 buffer = self.load_bytes(**kwargs)
---> 96 return buffer.load()

File ~/Projects/chrisammon3000/experiments/docarray/docarray-test/.venv/lib/python3.11/site-packages/docarray/typing/bytes/video_bytes.py:86, in VideoBytes.load(self, **kwargs)
     84     audio = parse_obj_as(AudioNdArray, np.array(audio_frames))
     85 else:
---> 86     audio = parse_obj_as(AudioNdArray, np.stack(audio_frames))
     88 video = parse_obj_as(VideoNdArray, np.stack(video_frames))
     89 indices = parse_obj_as(NdArray, keyframe_indices)

File ~/Projects/chrisammon3000/experiments/docarray/docarray-test/.venv/lib/python3.11/site-packages/numpy/core/shape_base.py:449, in stack(arrays, axis, out, dtype, casting)
    447 shapes = {arr.shape for arr in arrays}
    448 if len(shapes) != 1:
--> 449     raise ValueError('all input arrays must have the same shape')
    451 result_ndim = arrays[0].ndim + 1
    452 axis = normalize_axis_index(axis, result_ndim)

ValueError: all input arrays must have the same shape

import os
from pathlib import Path
import numpy as np
from docarray.typing import VideoUrl
from moviepy.video.io.ffmpeg_tools import ffmpeg_extract_subclip


def generate_subclips(parent_path, video_id, video_uri, video_duration, duration=60):
    subclips_path = Path(parent_path) / "subclips"
    subclips_path.mkdir(exist_ok=True)

    start_times = np.arange(0, video_duration, duration)
    end_times = np.append(start_times[1:], video_duration)
    clip_times = list(zip(start_times, end_times))

    for start_time, end_time in clip_times:
        # filename should have start_end seconds as part of the name
        output_file_path = subclips_path / f"{video_id}__{start_time}_{end_time}.{video_uri.suffix[1:]}"
        ffmpeg_extract_subclip(video_uri, start_time, end_time, targetname=output_file_path)

# Example usage
# parent_path = 'path/to/parent/directory'
# video_id = 'example_video_id'
# video_uri = Path('path/to/video.mp4')
# video_duration = 1200  # for example, 20 minutes
# generate_subclips(parent_path, video_id, video_uri, video_duration, duration=60)

def sort_key(path):
    """Sorts by the start time in the subclip file name
    For example: Fu7YkoRWKB8_Y__0_60.mp4 will sort by `0`
    """
    # Extract the integer after "__" from the filename
    return int(path.stem.split('__')[1].split('_')[0])

subclips_dir = Path(os.getcwd()).parent / "subclips"

# create subclips
generate_subclips(subclips_dir, <video_id>, <video_uri>, <video_duration>, duration=60)
subclips_paths = sorted(subclips_dir.iterdir(), key=sort_key)
video_urls = [VideoUrl(f"{str(subclip)}") for subclip in subclips_paths]

# load tensors
# the first subclip might work...
subclip0 = VideoUrl(str(subclips_paths[0]))
subclip0_tensors = subclip.load()

# but the second and other subclips throw the shape mismatch error
subclip1 = VideoUrl(str(subclips_paths[1]))
subclip1_tensors = subclip.load()

0.40.0