File size: 3,054 Bytes
b0aacf7 71aad4e b0aacf7 71aad4e b0aacf7 71aad4e b0aacf7 71aad4e b0aacf7 71aad4e b0aacf7 da53346 b0aacf7 da53346 b0aacf7 71aad4e 10c7829 d8f4957 4ea7ae6 d8f4957 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 |
import subprocess
import gradio as gr
import tempfile
from pathlib import Path
class Predictor:
def predict(self,
audio: str,
bg_color: str = "#000000",
fg_alpha: float = 0.75,
bars_color: str = "#ffffff",
bar_count: int = 100,
bar_width: float = 0.4,
caption_text: str = "",
) -> str:
"""Make waveform video from audio file"""
waveform_video = gr.make_waveform(
audio,
bg_color=bg_color,
fg_alpha=fg_alpha,
bars_color=bars_color,
bar_count=bar_count,
bar_width=bar_width,
)
if caption_text == "" or caption_text is None:
return waveform_video
else:
padded_waveform_path = tempfile.mktemp(suffix=".mp4")
background_image_path = tempfile.mktemp(suffix=".png")
final_video_path = tempfile.mktemp(suffix=".mp4")
# Add padding to the top of the waveform video
subprocess.run([
'ffmpeg', '-y', '-i', waveform_video, '-vf',
f'pad=width=1000:height=667:x=0:y=467:color={bg_color[1:]}',
padded_waveform_path
], check=True)
# Create an image using ImageMagick with provided font
subprocess.run([
'convert', '-background', bg_color, '-fill', bars_color, '-font', '/src/fonts/Roboto-Black.ttf',
'-pointsize', '48', '-size', '900x367', '-gravity', 'center', f'caption:{caption_text}',
'-bordercolor', bg_color, '-border', '40', background_image_path
], check=True)
# Overlay the image on the padded waveform video
subprocess.run([
'ffmpeg', '-y', '-i', padded_waveform_path, '-i', background_image_path,
'-filter_complex', 'overlay=0:0', final_video_path
], check=True)
return final_video_path
# Gradio user interface
def gradio_predict(audio, bg_color, fg_alpha, bars_color, bar_count, bar_width, caption_text):
predictor = Predictor()
result = predictor.predict(
audio=audio,
bg_color=bg_color,
fg_alpha=fg_alpha,
bars_color=bars_color,
bar_count=bar_count,
bar_width=bar_width,
caption_text=caption_text
)
return result
# Launch Gradio interface
interface = gr.Interface(
fn=gradio_predict,
inputs=[
gr.Audio(source="upload", type="filepath", label="Audio File"),
gr.Textbox(value="#000000", label="Background Color"),
gr.Slider(0, 1, value=0.75, label="Foreground Opacity"),
gr.ColorPicker(value="#ffffff", label="Bars Color"),
gr.Slider(10, 500, value=100, step=1, label="Number of Bars"),
gr.Slider(0, 1, value=0.4, step=0.1, label="Bar Width"),
gr.Textbox(value="", label="Caption Text")
],
outputs=gr.Video(label="Waveform Video"),
live=False
)
if __name__ == "__main__":
interface.launch(server_name="0.0.0.0", server_port=7860) |