NeuralFalcon's picture
Update app.py
1a90df2 verified
import os
import subprocess
def get_video_duration(video_path):
"""Return video duration as (hours, minutes, seconds)"""
try:
result = subprocess.run(
["ffprobe", "-v", "error", "-show_entries",
"format=duration", "-of",
"default=noprint_wrappers=1:nokey=1", video_path],
stdout=subprocess.PIPE,
stderr=subprocess.PIPE,
text=True
)
duration = float(result.stdout.strip())
hours = int(duration // 3600)
minutes = int((duration % 3600) // 60)
seconds = int(duration % 60)
return hours, minutes, seconds
except Exception as e:
print("Error:", e)
return None
def edit(video_path, keep_silence_up_to=0.2,high_quality=False,save_folder="./Remove-Silence/"):
os.makedirs(save_folder, exist_ok=True)
base_name = os.path.splitext(os.path.basename(video_path))[0]
save_path = os.path.join(save_folder, f"{base_name}_ALTERED.mp4")
cmd = [
"auto-editor",
os.path.abspath(video_path),
"--margin", f"{keep_silence_up_to}sec",
"--no-open",
"--temp-dir", "./temp/",
"-o", save_path
]
if high_quality:
cmd.extend([
"-c:v", "libx264",
"-b:v", "12M",
"-profile:v", "high",
"-c:a", "aac",
"-b:a", "192k"
])
try:
print("Running command:", " ".join(cmd))
subprocess.run(cmd, check=True)
save_path = os.path.abspath(save_path)
# get durations
input_duration = get_video_duration(os.path.abspath(video_path))
output_duration = get_video_duration(save_path)
# prepare logs
logs = f"""original Video Duration: {input_duration[0]} Hour {input_duration[1]} Minute {input_duration[2]} Second
After Removing Silences: {output_duration[0]} Hour {output_duration[1]} Minute {output_duration[2]} Second"""
return save_path, logs
except Exception as e:
print("Error:", e)
return None, None
import gradio as gr
def ui():
with gr.Blocks(title="Automatically Remove Silence From Video") as demo:
gr.Markdown("## 🎬 Automatically Remove Silence From Video")
gr.Markdown("Upload an .mp4 video, and silent parts will be removed automatically.")
with gr.Row():
with gr.Column():
video_input = gr.File(label="Upload Video")
run_btn = gr.Button("Process Video")
with gr.Accordion('πŸŽ₯ Video Settings', open=False):
silence_value = gr.Number(label="Keep Silence Upto (In seconds)", value=0.2)
high_quality = gr.Checkbox(value=False, label='Export in High Quality')
with gr.Column():
output_file = gr.File(label="Download Video File")
duration_info = gr.Textbox(label="Duration", interactive=False)
# hook function to button
run_btn.click(
fn=edit,
inputs=[video_input, silence_value,high_quality],
outputs=[output_file, duration_info]
)
return demo
# demo = ui()
# demo.queue().launch(debug=True, share=True)
import click
@click.command()
@click.option("--debug", is_flag=True, default=False, help="Enable debug mode.")
@click.option("--share", is_flag=True, default=False, help="Enable sharing of the interface.")
def main(debug, share):
demo=ui()
demo.queue().launch(debug=debug, share=share)
if __name__ == "__main__":
main()