Update app.py

app.py

@@ -2,7 +2,7 @@ import os
 import moviepy.editor as mp
 import gradio as gr
 
-def split_video(video_file, parts):
+def split_video(video_file, parts, resolution, quality, audio_enabled, volume):
     video = mp.VideoFileClip(video_file.name)
     duration = video.duration
     part_duration = duration / parts
@@ -12,8 +12,31 @@ def split_video(video_file, parts):
         start_time = part * part_duration
         end_time = (part + 1) * part_duration
         part_clip = video.subclip(start_time, end_time)
+
+        # Настройка параметров кодирования
+        if quality == "Низкое":
+            bitrate = "1M"
+        elif quality == "Среднее":
+            bitrate = "2M"
+        else:
+            bitrate = "4M"
+
+        # Настройка параметров звука
+        if not audio_enabled:
+            part_clip.audio = None
+        else:
+            part_clip.audio = part_clip.audio.volumex(volume)
+
+        # Сохранение видео
         output_filename = f"part_{part + 1}.mp4"
-        part_clip.write_videofile(output_filename, codec="libx264", audio_codec="aac")
+        part_clip.write_videofile(
+            output_filename,
+            codec="libx264",
+            audio_codec="aac",
+            bitrate=bitrate,
+            preset="medium",
+            ffmpeg_params=["-vf", f"scale={resolution}"],
+        )
         output_files.append(output_filename)
 
     return output_files
@@ -21,12 +44,17 @@ def split_video(video_file, parts):
 iface = gr.Interface(
     fn=split_video,
     inputs=[
-        gr.File(label="Upload Video"), 
-        gr.Slider(minimum=2, maximum=10, value=2, label="Number of Parts")
+        gr.File(label="Upload Video"),
+        gr.Slider(minimum=2, maximum=10, value=2, label="Number of Parts"),
+        gr.Dropdown(choices=["720p", "1080p", "4K"], label="Resolution"),
+        gr.Dropdown(choices=["Низкое", "Среднее", "Высокое"], label="Качество"),
+        gr.Checkbox(label="Включить звук"),
+        gr.Slider(minimum=0, maximum=2, value=1, label="Громкость"),
     ],
     outputs=gr.Files(label="Download Split Videos"),
     title="Video Splitter",
     description="Upload your video and select how many parts you want to split it into."
 )
+
 if __name__ == "__main__":
     iface.launch()