From b091d3057bca8ddb0073618fad487b4cbce3ee2f Mon Sep 17 00:00:00 2001
From: Sebastian Monroy <sebastian.monroy@gmail.com>
Date: Mon, 1 Sep 2025 10:36:20 +0100
Subject: [PATCH] MediaStreamInput: add "image_batch" option to format input,
 which downloads a video and outputs as tensor batch (WIP)

---
 media_stream.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/media_stream.py b/media_stream.py
index 686367d..5927987 100644
--- a/media_stream.py
+++ b/media_stream.py
@@ -41,7 +41,7 @@ class MediaStreamInput:
         return {
             "required": {
                 "input_name": ("STRING", {"default": "default_input", "multiline": False}),
-                "format": (["auto", "image", "video"],),
+                "format": (["auto", "image", "video", "image_batch"],),
                 "presigned_download_url": ("STRING", {
                     "multiline": True,
                     "default": "<auto-filled by system>"
@@ -75,6 +75,8 @@ class MediaStreamInput:
                     return self._process_image(media_bytes)
             elif format == 'video':
                 return self._process_video(media_bytes)
+            elif format == 'image_batch':
+                return self._process_video(media_bytes) # Treat as video for processing
             else: # format == 'image'
                 return self._process_image(media_bytes)