direct output edit video

app.py

@@ -306,7 +306,8 @@ class VideoCoF_Controller(Wan_Controller):
                 cot=True,
             ).videos
 
-            final_video = sample
+            # Keep only the edited segment (drop reasoning/original parts)
+            final_video = sample[:, :, -source_frames_slider:, :, :]
 
         except Exception as e:
             print(f"Error: {e}")
@@ -324,7 +325,7 @@ class VideoCoF_Controller(Wan_Controller):
 
         # Save output
         save_sample_path = self.save_outputs(
-            False, length_slider, final_video, fps=fps
+            False, source_frames_slider, final_video, fps=fps
         )
         
         # Return input video to display it alongside output if needed?