diff --git a/nodes.py b/nodes.py index b52d9c5..ffa56e9 100644 --- a/nodes.py +++ b/nodes.py @@ -612,6 +612,7 @@ class DeepFuzeFaceSwap: "required": { "source_images": ("IMAGE",), "target_images": ("IMAGE",), + "audio": ("VHS_AUDIO",), "enhancer": ("None,codeformer,gfpgan_1.2,gfpgan_1.3,gfpgan_1.4,gpen_bfr_256,gpen_bfr_512,gpen_bfr_1024,gpen_bfr_2048,restoreformer_plus_plus".split(","),{"default":'None'}), "faceswap_model":("blendswap_256,inswapper_128,inswapper_128_fp16,simswap_256,simswap_512_unofficial,uniface_256".split(","),{"default":"blendswap_256"}), "frame_enhancer": ("None,clear_reality_x4,lsdir_x4,nomos8k_sc_x4,real_esrgan_x2,real_esrgan_x2_fp16,real_esrgan_x4,real_esrgan_x4_fp16,real_hatgan_x4,span_kendata_x4,ultra_sharp_x4".split(","),{"default":'None'}), @@ -657,6 +658,7 @@ class DeepFuzeFaceSwap: self, source_images, target_images, + audio, enhancer, faceswap_model, frame_enhancer, @@ -1015,7 +1017,11 @@ class DeepFuzeFaceSwap: # faceswap_filename = temp_file print(result.stderr) - return load_video_cv(faceswap_filename,0,'Disabled',512,512,0,0,1) + audio_file = os.path.join(audio_dir,str(time.time()).replace(".","")+".wav") + open(audio_file,"wb").write(audio()) + subprocess.run(f"ffmpeg -i {faceswap_filename} -i {audio_file} -c copy {faceswap_filename.replace('.mp4','_.mp4')} -y".split()) + + return load_video_cv(faceswap_filename.replace('.mp4','_.mp4'),0,'Disabled',512,512,0,0,1)