Jeffgold commited on
Commit
6a86ebc
·
1 Parent(s): 928f6cf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -6
app.py CHANGED
@@ -3,16 +3,15 @@ import ffmpeg
3
  import subprocess
4
  from gradio import outputs
5
 
6
- quality = gr.inputs.Dropdown(choices=["18", "23", "28", "32"], label="Quality")
7
  video_file = gr.inputs.File(label="Video File")
 
 
8
 
9
-
10
- def convert_video(quality):
11
  output_path = "output.m3u8"
12
- ffmpeg_command = f"ffmpeg -i input.mp4 -c:v libx264 -crf {quality} -f hls {output_path}"
13
  subprocess.run(ffmpeg_command, shell=True)
14
 
15
  return gr.outputs.Video(output_path)
16
 
17
- gr.Interface(convert_video, inputs=[quality], outputs=[outputs.Video()]).launch()
18
-
 
3
  import subprocess
4
  from gradio import outputs
5
 
 
6
  video_file = gr.inputs.File(label="Video File")
7
+ quality = gr.inputs.Dropdown(choices=["18", "23", "28", "32"], label="Quality")
8
+ aspect_ratio = gr.inputs.Dropdown(choices=["16:9", "4:3", "3:2"], label="Aspect Ratio")
9
 
10
+ def convert_video(video_file, quality, aspect_ratio):
 
11
  output_path = "output.m3u8"
12
+ ffmpeg_command = f"ffmpeg -i {video_file} -c:v libx264 -crf {quality} -f hls -aspect {aspect_ratio} {output_path}"
13
  subprocess.run(ffmpeg_command, shell=True)
14
 
15
  return gr.outputs.Video(output_path)
16
 
17
+ gr.Interface(convert_video, inputs=[video_file, quality, aspect_ratio], outputs=[outputs.Video()]).launch()