Update src/facerender/animate.py

src/facerender/animate.py

@@ -240,7 +240,7 @@ class AnimateFromCoeff():
             video_name_full = x['video_name']  + '_full.mp4'
             full_video_path = os.path.join(video_save_dir, video_name_full)
             return_path = full_video_path
-            paste_pic(path, pic_path, crop_info, new_audio_path, full_video_path, extended_crop= True if 'ext' in preprocess.lower() else False)
+            base64_video, temp_file_path = paste_pic(path, pic_path, crop_info, new_audio_path, full_video_path, extended_crop= True if 'ext' in preprocess.lower() else False)
             print(f'The generated video is named {video_save_dir}/{video_name_full}') 
         else:
             full_video_path = av_path 
@@ -255,15 +255,15 @@ class AnimateFromCoeff():
             # print('return_path',return_path)
 
             try:
-                enhanced_images_gen_with_len = enhancer_generator_with_len(path, method=enhancer, bg_upsampler=background_enhancer)
+                enhanced_images_gen_with_len = enhancer_generator_with_len(temp_file_path, method=enhancer, bg_upsampler=background_enhancer)
                 imageio.mimsave(enhanced_path, enhanced_images_gen_with_len, fps=float(25))
             except:
-                enhanced_images_gen_with_len = enhancer_list(path, method=enhancer, bg_upsampler=background_enhancer)
+                enhanced_images_gen_with_len = enhancer_list(temp_file_path, method=enhancer, bg_upsampler=background_enhancer)
                 imageio.mimsave(enhanced_path, enhanced_images_gen_with_len, fps=float(25))
             
-            print('enhanced_images_gen_with_len',enhanced_images_gen_with_len)
-            print('enhanced_path',enhanced_path)
-            print('av_path_enhancer',av_path_enhancer)
+            # print('enhanced_images_gen_with_len',enhanced_images_gen_with_len)
+            # print('enhanced_path',enhanced_path)
+            # print('av_path_enhancer',av_path_enhancer)
             base64_video, temp_file_path = save_video_with_watermark(enhanced_path, new_audio_path, av_path_enhancer, watermark= False)
             print(f'The generated video is named {video_save_dir}/{video_name_enhancer}')
             os.remove(enhanced_path)