Spaces:
Running
Running
| import logging | |
| import subprocess | |
| from pprint import pprint | |
| from tempfile import _TemporaryFileWrapper | |
| from ffmpy import FFmpeg | |
| import gradio as gr | |
| from functions import ( | |
| Clear, | |
| CommandBuilder, | |
| audio_channels, | |
| audio_codecs, | |
| audio_quality, | |
| audio_sample_rates, | |
| change_clipbox, | |
| containers, | |
| set_custom_bitrate, | |
| media_change, | |
| presets, | |
| supported_codecs, | |
| supported_presets, | |
| video_codecs, | |
| video_containers, | |
| VF, | |
| ) | |
| logging.basicConfig(level=logging.INFO) | |
| logging.info(msg=f"{video_containers}") | |
| def convert(file: _TemporaryFileWrapper, container_format: str, new_state: str): | |
| stderr = "" | |
| # stdout="" | |
| output_file = "" | |
| # video = "" | |
| ffmpeg = FFmpeg() | |
| try: | |
| logging.info("File name: %s", file.name) | |
| new_name, _ = file.name.split(".") | |
| logging.info("New filename:%s", new_name) | |
| output_file = f"{new_name}1.{container_format.lower()}" | |
| ffmpeg = FFmpeg( | |
| inputs={file.name: None}, | |
| outputs={output_file: ffmpeg_commands.commands.split()}, | |
| global_options=["-y", "-hide_banner"], | |
| ) | |
| print(ffmpeg) | |
| print(ffmpeg.cmd) | |
| ffmpeg.run(stderr=subprocess.PIPE) | |
| # pprint(f"{stdout} {stderr}") | |
| output = f"{ffmpeg.cmd}" | |
| # video=gr.update(label=output_file,value=output_file) | |
| except Exception as e: | |
| stderr = e | |
| output = f"{stderr}" | |
| return [None, None, None, output, new_state] | |
| new_state = output_file | |
| return [output_file, output_file, output_file, output, new_state] | |
| with gr.Blocks(css="./styles.css") as demo: | |
| # with gr.Tabs(): | |
| with gr.Tab("Format"): | |
| # Input Buttons | |
| with gr.Row(): | |
| with gr.Column() as inputs: | |
| file_input = gr.File() | |
| options = gr.Radio( | |
| label="options", choices=containers, value=containers[0] | |
| ) | |
| with gr.Row(): | |
| with gr.Row() as inputs_clip: | |
| clip = gr.Dropdown( | |
| choices=["None", "Enabled"], label="Clip:", value="None" | |
| ) | |
| start_time = gr.Textbox( | |
| label="Start Time:", | |
| placeholder="00:00", | |
| visible=False, | |
| interactive=True, | |
| ) | |
| stop_time = gr.Textbox( | |
| label="Stop Time:", | |
| placeholder="00:00", | |
| visible=False, | |
| interactive=True, | |
| ) | |
| with gr.Row(): | |
| clearBtn = gr.Button("Clear") | |
| convertBtn = gr.Button("Convert", variant="primary") | |
| # Output Buttons | |
| with gr.Column(): | |
| # media_output = gr.Audio(label="Output") | |
| with gr.Row(): | |
| video_button = gr.Button("Video") | |
| audio_button = gr.Button("Audio") | |
| file_button = gr.Button("File") | |
| media_output_audio = gr.Audio( | |
| type="filepath", label="Output", visible=True, interactive=False | |
| ) | |
| media_output_video = gr.Video(label="Output", visible=False) | |
| media_output_file = gr.File(label="Output", visible=False) | |
| with gr.Row() as command_output: | |
| output_textbox = gr.Textbox(label="command", elem_id="outputtext") | |
| resetFormat = Clear(inputs, inputs_clip) | |
| print(inputs_clip.children) | |
| # print(resetFormat) | |
| state = gr.State() | |
| clearBtn.click(resetFormat.clear, resetFormat(), resetFormat()) | |
| convertBtn.click( | |
| convert, | |
| inputs=[file_input, options, state], | |
| outputs=[ | |
| media_output_audio, | |
| media_output_file, | |
| media_output_video, | |
| output_textbox, | |
| state, | |
| ], | |
| ) | |
| with gr.Tab("Video"): | |
| with gr.Row() as video_inputs: | |
| video_options = gr.Dropdown( | |
| label="video", choices=video_codecs, value=video_codecs[-1] | |
| ) | |
| preset_options = gr.Dropdown( | |
| choices=presets, label="presets", value=presets[-1] | |
| ) | |
| with gr.Row(elem_id="button"): | |
| with gr.Column(): | |
| clearBtn = gr.Button("Clear") | |
| videoReset = Clear(video_inputs) | |
| clearBtn.click(videoReset.clear, videoReset(), videoReset()) | |
| with gr.Tab("Audio"): | |
| with gr.Row() as audio_inputs: | |
| # print(names[0]) | |
| audio_options = gr.Dropdown( | |
| label="audio", choices=audio_codecs, value=audio_codecs[-1] | |
| ) | |
| audio_bitrate = gr.Dropdown( | |
| choices=audio_quality, | |
| label="Audio Qualities", | |
| value=audio_quality[0], | |
| ) | |
| custom_bitrate = gr.Number(label="Audio Qualities", visible=False) | |
| gr.Dropdown( | |
| choices=audio_channels, | |
| label="Audio Channels", | |
| value=audio_channels[0], | |
| ) | |
| gr.Dropdown( | |
| choices=audio_sample_rates, | |
| label="Sample Rates", | |
| value=audio_sample_rates[0], | |
| ) | |
| with gr.Column(elem_id="button"): | |
| clearBtn = gr.Button("Clear") | |
| audioReset = Clear(audio_inputs) | |
| clearBtn.click(audioReset.clear, audioReset(), audioReset()) | |
| with gr.Tab("Filters") as filter_inputs: | |
| gr.Markdown("## Video") | |
| # equal_height=True | |
| with gr.Row(equal_height=True) as filter_inputs: | |
| for i in VF: | |
| # print(i.values()) | |
| # values = list(i.values()) | |
| values = list(i.values())[0] | |
| choices = [j for lst in values for j in [lst.get("name")]] | |
| a = gr.Dropdown( | |
| label=str(list(i.keys())[0]), choices=choices, value=choices[0] | |
| ) | |
| gr.Markdown("## Audio") | |
| with gr.Row(elem_id="acontrast") as filter_inputs_1: | |
| acontrastSlider = gr.Slider(label="Acontrast", elem_id="acontrast") | |
| with gr.Column(elem_id="button"): | |
| clearBtn = gr.Button("Clear") | |
| filterReset = Clear(filter_inputs, filter_inputs_1) | |
| clearBtn.click(filterReset.clear, filterReset(), filterReset()) | |
| """ Format Tab change functions""" | |
| ffmpeg_commands = CommandBuilder( | |
| inputs_clip, video_inputs, audio_inputs, filter_inputs, filter_inputs_1 | |
| ) | |
| ffmpeg_commands.do() | |
| # demo.load(fn=ffmpeg_commands.reset, inputs=[], outputs=[]) | |
| clip.change(fn=change_clipbox, inputs=clip, outputs=[start_time, stop_time]) | |
| # pprint(ffmpeg_commands.commands) | |
| ffmpeg_commands.update(output_textbox) | |
| # file_input.change(fn=updateOutput,inputs=file_input,outputs=output_textbox) | |
| options.change(supported_codecs, [options], [video_options, audio_options]) | |
| # options.change(mediaChange,[options],[media_output_audio,media_output_video]) | |
| # video_button.click(fn=videoChange,inputs=media_output_file,outputs=media_output_video) | |
| audio_button.click( | |
| media_change, | |
| [audio_button, state], | |
| [media_output_audio, media_output_video, media_output_file], | |
| ) | |
| video_button.click( | |
| media_change, | |
| [video_button, state], | |
| [media_output_audio, media_output_video, media_output_file], | |
| ) | |
| # media_output_audio.change(lambda x:gr.update(value=x),[media_output_audio],[media_output_video]) | |
| file_button.click( | |
| media_change, | |
| [file_button, state], | |
| [media_output_audio, media_output_video, media_output_file], | |
| ) | |
| """Video Tab change functions""" | |
| video_options.change(supported_presets, [video_options], [preset_options]) | |
| """Audio Tab change functions""" | |
| audio_bitrate.change(set_custom_bitrate, [audio_bitrate], [custom_bitrate]) | |
| if __name__ == "__main__": | |
| demo.launch(debug=True) | |