Update app.py

app.py

@@ -41,7 +41,7 @@ def split_video_into_chunks(video_path, chunk_size):
     # Calculate the number of chunks needed
     num_chunks = math.ceil(total_frames / chunk_size)
 
-    # Create a list to store the chunks
+    # Create a list to store the chunks and their frame counts
     chunks = []
 
     # Split the video into chunks
@@ -53,13 +53,18 @@ def split_video_into_chunks(video_path, chunk_size):
         # Extract the chunk from the video
         chunk = video.subclip(start_frame / video.fps, end_frame / video.fps)
 
-        # Add the chunk to the list
-        chunks.append(chunk)
+        # Calculate the frame count for the chunk
+        frame_count = end_frame - start_frame
+
+        # Add the chunk and its frame count to the list
+        chunks.append((chunk, frame_count))
 
     # If the last chunk is smaller than the chunk size
-    if len(chunks) > 0 and len(chunks[-1]) < chunk_size:
+    if len(chunks) > 0 and chunks[-1][1] < chunk_size:
         # Adjust the end frame of the last chunk to the total frames
-        chunks[-1] = video.subclip(chunks[-1].t_start, video.duration)
+        last_chunk = chunks[-1][0]
+        last_frame_count = chunks[-1][1]
+        chunks[-1] = (video.subclip(last_chunk.t_start, video.duration), last_frame_count)
 
     return chunks