Update app_multi.py

app_multi.py

@@ -13,6 +13,7 @@ import numpy as np
 from PIL import Image, ImageDraw, ImageFont
 from moviepy.editor import *
 from moviepy.video.io.VideoFileClip import VideoFileClip
+import moviepy.editor as mpe
 
 import asyncio
 import json
@@ -63,7 +64,17 @@ def ref_video_fn(path_of_ref_video):
         return gr.update(value=False)
 
 sad_talker = SadTalker("checkpoints", "src/config", lazy_load=True)
-    
+
+
+# combine video with music
+
+def combine_music(video, audio):
+    my_clip = mpe.VideoFileClip(video)
+    audio_background = mpe.AudioFileClip(audio)
+    final_audio = mpe.CompositeAudioClip([my_clip.audio, audio_background])
+    final_clip = my_clip.set_audio(final_audio)
+    return final_clip
+
 # Reference: https://huggingface.co/spaces/zomehwh/rvc-models/blob/main/app.py#L21  # noqa
 in_hf_space = getenv('SYSTEM') == 'spaces'
 
@@ -783,7 +794,13 @@ with app:
         
         
                     with gr.Tabs(elem_id="sadtalker_driven_audio"):
-                        with gr.TabItem('💡您还可以将视频下载到本地'):
+                        with gr.TabItem('💡 完整视频'):
+
+                            submit = gr.Button('想把我唱给你听', elem_id="sadtalker_generate", variant='primary')
+
+                            gen_mv = gr.Button('添加歌曲伴奏吧', variant='primary')
+
+                            music_video = gr.Video(label="视频+伴奏", format="mp4").style(width=256)
         
                             with gr.Row():
                                 driven_audio = output_audio
@@ -827,11 +844,11 @@ with app:
                                     batch_size = gr.Slider(label="Batch size (数值越大，生成速度越快；若显卡性能好，可增大数值)", step=1, maximum=32, value=2)
                                     enhancer = gr.Checkbox(label="GFPGAN as Face enhancer", value=True, visible=False)
         
-                                submit = gr.Button('想把我唱给你听', elem_id="sadtalker_generate", variant='primary')
         
                     with gr.Tabs(elem_id="sadtalker_genearted"):
                             gen_video = gr.Video(label="为您生成的专属视频", format="mp4").style(width=256)
         
+            gen_mv.click(fn=combine_music, inputs=[gen_video, as_audio_no_vocals], outputs=[music_video])
         
             submit.click(
                     fn=sad_talker.test,