Update app.py

app.py

@@ -171,34 +171,33 @@ def infer(image, width=1024, height=1024, overlap_width=18, num_inference_steps=
     yield background, cnet_image
 
 def create_zoom_animation(previous_frame, next_frame, steps):
-    
     # List to store all frames
     interpolated_frames = []
     
-    for i in range(steps + 1):
-        t = i / steps  # Normalized time between 0 and 1
+    for i in range(steps):
+        t = i / (steps - 1)  # Normalized time between 0 and 1
         
         # Compute zoom factor (from 1 to 2)
         z = 1 + t  # Zoom factor increases from 1 to 2
         
-        # Compute crop size
-        crop_size = int(1024 / z)
-        
-        # Compute crop coordinates to center the crop
-        x0 = (1024 - crop_size) // 2
-        y0 = (1024 - crop_size) // 2
-        x1 = x0 + crop_size
-        y1 = y0 + crop_size
-        
-        # Crop the previous_frame
-        cropped_prev = previous_frame.crop((x0, y0, x1, y1))
-        # Resize to 512x512
-        if i == steps:
-            resized_frame = next_frame.resize((512, 512), Image.LANCZOS)
+        if i < steps - 1:
+            # Compute crop size
+            crop_size = int(1024 / z)
+            
+            # Compute crop coordinates to center the crop
+            x0 = (1024 - crop_size) // 2
+            y0 = (1024 - crop_size) // 2
+            x1 = x0 + crop_size
+            y1 = y0 + crop_size
+            
+            # Crop the previous_frame
+            cropped_prev = previous_frame.crop((x0, y0, x1, y1))
+            # Resize to 512x512
+            resized_frame = cropped_prev.resize((512, 512), Image.LANCZOS)
         else:
-            resized_prev = cropped_prev.resize((512, 512), Image.LANCZOS)
-            resized_frame = resized_prev
-  
+            # For the last frame, use the next_frame resized to 512x512
+            resized_frame = next_frame.resize((512, 512), Image.LANCZOS)
+        
         interpolated_frames.append(resized_frame)
 
     return interpolated_frames
@@ -251,15 +250,18 @@ def loop_outpainting(image, width=1024, height=1024, overlap_width=6, num_infere
     
     # Create interpolated frames
     final_frame_list = []
+    
     for i in range(len(reverse_image_list) - 1):
         larger_frame = reverse_image_list[i]
         smaller_frame = reverse_image_list[i + 1]
         interpolated_frames = create_zoom_animation(larger_frame, smaller_frame, num_interpolation_frames)
-        # Exclude the last frame to avoid duplication
-        final_frame_list.extend(interpolated_frames[:-1])
-    
-    # Add the last frame
-    final_frame_list.append(reverse_image_list[-1])
+        
+        if i == 0:
+            # Include all frames for the first sequence
+            final_frame_list.extend(interpolated_frames)
+        else:
+            # Exclude the first frame to avoid duplication
+            final_frame_list.extend(interpolated_frames[1:])
 
     # Create video from the final frame list
     video_path = create_video_from_images(final_frame_list, fps)