From b80733d5c136159dbc1e84b3dccde645004efdda Mon Sep 17 00:00:00 2001
From: Deb <133010426+aimfordeb@users.noreply.github.com>
Date: Thu, 31 Jul 2025 13:05:24 +0300
Subject: [PATCH] Add Continue Video node + Image Stitch division by zero error
 fixed

---
 funpack.py | 28 +++++++++++++++++++++++++++-
 1 file changed, 27 insertions(+), 1 deletion(-)

diff --git a/funpack.py b/funpack.py
index 212a5b32..fd67b651 100644
--- a/funpack.py
+++ b/funpack.py
@@ -316,6 +316,10 @@ class FunPackVideoStitch:
     }
 
     def linear_blend(self, batch_a, batch_b, blend_frames):
+        if blend_frames == 1:
+            blended_frame = 0.5 * batch_a[-1] + 0.5 * batch_b[0]
+            return blended_frame.unsqueeze(0)
+
         blended = []
         for i in range(blend_frames):
             alpha = i / (blend_frames - 1)
@@ -350,7 +354,27 @@ class FunPackVideoStitch:
 
         final_video = torch.cat(output_frames, dim=0)
         return (final_video,)
+        
+class FunPackContinueVideo:
+    CATEGORY = "FunPack"
+    RETURN_TYPES = ("IMAGE",)
+    RETURN_NAMES = ("CONTINUED",)
+    FUNCTION = "continue_video"
+    INPUT_TYPES = lambda: {
+        "required": {
+            "images": ("IMAGE",),
+            "frame_count": ("INT", {"default": 1, "min": 1, "max": 9999}),
+        }
+    }
 
+    def continue_video(self, images, frame_count):
+        total_frames = images.shape[0]
+
+        if frame_count > total_frames:
+            raise ValueError(f"Cannot extract {frame_count} frames from video with only {total_frames} frames.")
+
+        continued = images[-frame_count:]
+        return (continued,)
 
 # Update NODE_CLASS_MAPPINGS and NODE_DISPLAY_NAME_MAPPINGS
 NODE_CLASS_MAPPINGS = {
@@ -358,11 +382,13 @@ NODE_CLASS_MAPPINGS = {
     "FunPackCLIPLoader": FunPackCLIPLoader,
     "FunPackPromptEnhancer": FunPackPromptEnhancer,
     "FunPackVideoStitch": FunPackVideoStitch,
+    "FunPackContinueVideo": FunPackContinueVideo,
 }
 
 NODE_DISPLAY_NAME_MAPPINGS = {
     "FunPackImg2LatentInterpolation": "FunPack Img2Latent Interpolation",
     "FunPackCLIPLoader": "FunPack CLIP Loader",
     "FunPackPromptEnhancer": "FunPack Prompt Enhancer (Standalone)",
-    "FunPackVideoStitch": "FunPack Video Stitch"
+    "FunPackVideoStitch": "FunPack Video Stitch",
+    "FunPackContinueVideo": "FunPack Continue Video"
 }
\ No newline at end of file