Update app.py

app.py

@@ -8,9 +8,9 @@ def split_video(input_video, duration):
     num_segments = ceil(video_duration / duration)
     
     segments = []
-    for i in range(num_segments):
+    for i in range(num_segments - 1):
         start_time = i * duration
-        end_time = min((i + 1) * duration, video_duration)
+        end_time = (i + 1) * duration
         output_video = f"segment_{i+1}.mp4"
         
         stream = ffmpeg.input(input_video)
@@ -19,6 +19,19 @@ def split_video(input_video, duration):
         ffmpeg.run(stream, overwrite_output=True)
         
         segments.append(output_video)
+        
+    # Обработка последнего сегмента
+    remaining_duration = video_duration - (num_segments - 1) * duration
+    start_time = (num_segments - 1) * duration
+    end_time = video_duration
+    output_video = f"segment_{num_segments}.mp4"
+    
+    stream = ffmpeg.input(input_video)
+    stream = ffmpeg.trim(stream, start=start_time, end=end_time)
+    stream = ffmpeg.output(stream, output_video)
+    ffmpeg.run(stream, overwrite_output=True)
+    
+    segments.append(output_video)
     
     return segments
 
@@ -30,7 +43,7 @@ interface = gr.Interface(
     ],
     outputs=gr.Files(label="Video Segments"),
     title="Video Splitter",
-    description="Split a video into equal-duration segments.",
+    description="Split a video into equal-duration segments. The last segment can be of unequal duration.",
 )
 
-interface.launch()
+interface.launch()