From 9fe0e3b7ea61bf8d8fce7c63dd3fff784696495d Mon Sep 17 00:00:00 2001
From: LonicaMewinsky <93007558+LonicaMewinsky@users.noreply.github.com>
Date: Fri, 10 Nov 2023 18:54:14 -0600
Subject: [PATCH] Added indices output

---
 LoadAnim.py | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/LoadAnim.py b/LoadAnim.py
index 1b26381..1d9544b 100644
--- a/LoadAnim.py
+++ b/LoadAnim.py
@@ -26,8 +26,8 @@ class BreakFrames:
                 }),
             },
         }
-    RETURN_TYPES = ("IMAGE", "IMAGE")
-    RETURN_NAMES = ("Frames","Keyframes")
+    RETURN_TYPES = ("IMAGE", "IMAGE", "STRING")
+    RETURN_NAMES = ("Frames","Keyframes", "Keyframe indices")
 
     FUNCTION = "breakframes"
     CATEGORY = "Frames"
@@ -59,7 +59,7 @@ class BreakFrames:
                 break
         video_capture.release()
         if num_keyframes > 0:
-            N = np.clip(num_keyframes, 2, len(tensors)-1)
+            N = np.clip(num_keyframes, 2, len(tensors)-1) # Save a spot for first frame
             differences = [torch.norm(tensors[i+1] - tensors[i], p=2) for i in range(len(tensors)-1)]
             _, top_indices = torch.topk(torch.tensor(differences), k=N, largest=True)
             keyframe_indices = sorted([index.item() + 1 for index in top_indices])
@@ -69,7 +69,8 @@ class BreakFrames:
         else:
             cat_keyframe_tensors = None
         cat_frame_tensors = torch.cat(tensors, dim = 0).permute(0, 2, 3, 1)
-        return (cat_frame_tensors, cat_keyframe_tensors)
+        print(str(keyframe_indices))
+        return (cat_frame_tensors, cat_keyframe_tensors, str(keyframe_indices))
 
 NODE_CLASS_MAPPINGS = {
     "BreakFrames": BreakFrames