Spaces:
Sleeping
Sleeping
File size: 1,881 Bytes
c7810b4 f92b394 51ed60c f92b394 51ed60c f92b394 51ed60c f92b394 |
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 |
import os
import random
import ffmpeg
import gradio as gr
# Function to add music to video
def add_music_to_video(video_path, chunks_folder="chunks", output_path="output_with_music.mp4"):
# List all MP3 files in the chunks folder
music_files = [os.path.join(chunks_folder, f) for f in os.listdir(chunks_folder) if f.endswith('.mp3')]
if not music_files:
return "Error: No audio files found in the chunks folder."
# Randomly select one MP3 file
music_file_path = random.choice(music_files)
print(f"Selected music file: {music_file_path}")
# Get the duration of the video
video_info = ffmpeg.probe(video_path)
video_duration = float(video_info['streams'][0]['duration'])
# Trim the music file to match the video duration
trimmed_music_path = "trimmed_music.mp3"
ffmpeg.input(music_file_path).output(
trimmed_music_path, ss=0, t=video_duration
).run(overwrite_output=True)
# Combine video and audio
video_input = ffmpeg.input(video_path)
audio_input = ffmpeg.input(trimmed_music_path)
ffmpeg.concat(video_input, audio_input, v=1, a=1).output(
output_path, vcodec="libx264", acodec="aac", strict="experimental"
).run(overwrite_output=True)
return output_path
# Define Gradio interface
def process_video(uploaded_video):
video_path = uploaded_video.name
output_video = add_music_to_video(video_path, chunks_folder="/content/chunks")
return output_video
# Gradio Interface
interface = gr.Interface(
fn=process_video,
inputs=gr.Video(label="Upload Video"),
outputs=gr.Video(label="Video with Background Music"),
title="Add Background Music to Video",
description="Upload a video, and this app will add a 10-second background music track that matches the video duration."
)
# Launch Gradio app
interface.launch(share=True)
|