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 - 1):
+    for i in range(num_segments):
         start_time = i * duration
-        end_time = (i + 1) * duration
+        end_time = min((i + 1) * duration, video_duration)
         output_video = f"segment_{i+1}.mp4"
         
         stream = ffmpeg.input(input_video)
@@ -19,19 +19,6 @@ 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
 
@@ -43,7 +30,7 @@ interface = gr.Interface(
     ],
     outputs=gr.Files(label="Video Segments"),
     title="Video Splitter",
-    description="Split a video into equal-duration segments. The last segment can be of unequal duration.",
+    description="Split a video into equal-duration segments.",
 )
 
 interface.launch()