Jeffgold commited on
Commit
412d007
·
1 Parent(s): 6d721d7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -5
app.py CHANGED
@@ -12,8 +12,8 @@ from gradio import inputs
12
  File = Path
13
 
14
  video_file = gr.inputs.File(label="Video File")
15
- quality = gr.inputs.Dropdown(choices=["18", "23", "28", "32"], label="Quality", default="23")
16
- aspect_ratio = gr.inputs.Dropdown(choices=["16:9", "4:3", "3:2", "2:1"], label="Aspect Ratio", default="16:9")
17
 
18
  def convert_video(video_file: File, quality, aspect_ratio):
19
  if video_file is None:
@@ -21,12 +21,11 @@ def convert_video(video_file: File, quality, aspect_ratio):
21
  else:
22
  output_path = f"{temp_dir}/{Path(video_file).stem}.m3u8"
23
 
24
- temp_dir = tempfile.gettempdir()
25
- ffmpeg_command = f"ffmpeg -i {video_file or output_path} -c:v libx264 -crf {quality} -f {output_format} -aspect {aspect_ratio} {temp_dir}/{output_path}"
26
  subprocess.run(ffmpeg_command, shell=True)
27
 
28
  return components.Video(output_path)
29
 
30
  from gradio import outputs
31
 
32
- gr.Interface(convert_video, inputs="video", outputs=[outputs.Video()]).launch()
 
12
  File = Path
13
 
14
  video_file = gr.inputs.File(label="Video File")
15
+ quality = gr.inputs.RadioButtons(choices=["18", "23", "28", "32"], label="Quality", default="23")
16
+ aspect_ratio = gr.inputs.Slider(min=1, max=16, step=1, label="Aspect Ratio", default=16)
17
 
18
  def convert_video(video_file: File, quality, aspect_ratio):
19
  if video_file is None:
 
21
  else:
22
  output_path = f"{temp_dir}/{Path(video_file).stem}.m3u8"
23
 
24
+ ffmpeg_command = f"ffmpeg -i {video_file or output_path} -c:v libx264 -crf {quality} -f hls -aspect {aspect_ratio} {temp_dir}/{output_path}"
 
25
  subprocess.run(ffmpeg_command, shell=True)
26
 
27
  return components.Video(output_path)
28
 
29
  from gradio import outputs
30
 
31
+ gr.Interface(convert_video, inputs=[video_file, quality, aspect_ratio], outputs=[outputs.Video()]).launch()