xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-08 05:17:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

Make sure bytesIO objects can be read many times

Browse Source
This commit is contained in:
Jacob Segal 2025-04-27 21:56:43 -07:00
parent 3e0be20f9c
commit 48f9eb3ec7
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy/comfy_types/input_types.py
View File

@ -132,6 +132,8 @@ class VideoFromFile(VideoInput):
Returns: Returns:
Tuple of (width, height) Tuple of (width, height)
""" """
if isinstance(self.file, io.BytesIO):
self.file.seek(0) # Reset the BytesIO object to the beginning
with av.open(self.file, mode='r') as container: with av.open(self.file, mode='r') as container:
for stream in container.streams: for stream in container.streams:
if stream.type == 'video': if stream.type == 'video':
@ -180,6 +182,8 @@ class VideoFromFile(VideoInput):
return VideoComponents(images=images, audio=audio, frame_rate=frame_rate, metadata=metadata) return VideoComponents(images=images, audio=audio, frame_rate=frame_rate, metadata=metadata)
def get_components(self) -> VideoComponents: def get_components(self) -> VideoComponents:
if isinstance(self.file, io.BytesIO):
self.file.seek(0) # Reset the BytesIO object to the beginning
with av.open(self.file, mode='r') as container: with av.open(self.file, mode='r') as container:
return self.get_components_internal(container) return self.get_components_internal(container)
raise ValueError(f"No video stream found in file '{self.file}'") raise ValueError(f"No video stream found in file '{self.file}'")
@ -191,6 +195,8 @@ class VideoFromFile(VideoInput):
codec: VideoCodec = VideoCodec.AUTO, codec: VideoCodec = VideoCodec.AUTO,
metadata: Optional[dict] = None metadata: Optional[dict] = None
): ):
if isinstance(self.file, io.BytesIO):
self.file.seek(0) # Reset the BytesIO object to the beginning
with av.open(self.file, mode='r') as container: with av.open(self.file, mode='r') as container:
container_format = container.format.name container_format = container.format.name
video_encoding = container.streams.video[0].codec.name if len(container.streams.video) > 0 else None video_encoding = container.streams.video[0].codec.name if len(container.streams.video) > 0 else None