Update app.py

app.py

@@ -190,43 +190,43 @@ def main(args):
     result, base64_video,temp_file_path,new_audio_path = animate_from_coeff.generate(data, save_dir, pic_path, crop_info, \
                                 enhancer=args.enhancer, background_enhancer=args.background_enhancer, preprocess=args.preprocess)
 
-    # face_path = temp_file_path
-    # audio_path = new_audio_path
-    # temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.mp4')
-    # video_lipsync_file_path = temp_file.name
-    # output_path = video_lipsync_file_path
-
-    # # Call the function
-    # inference_function.video_lipsync_correctness(
-    #     face=face_path,
-    #     audio_path=audio_path,
-    #     face3d_net_path = path_of_net_recon_model,
-    #     outfile=output_path,
-    #     tmp_dir="temp",
-    #     crop=[0, -1, 0, -1],
-    #     re_preprocess=True,  # Set to True if you want to reprocess; False otherwise
-    #     exp_img="neutral",  # Can be 'smile', 'neutral', or path to an expression image
-    #     one_shot=False,
-    #     up_face="original",  # Options: 'original', 'sad', 'angry', 'surprise'
-    #     LNet_batch_size=16,
-    #     without_rl1=False
-    # )
+    face_path = temp_file_path
+    audio_path = new_audio_path
+    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.mp4')
+    video_lipsync_file_path = temp_file.name
+    output_path = video_lipsync_file_path
+
+    # Call the function
+    inference_function.video_lipsync_correctness(
+        face=face_path,
+        audio_path=audio_path,
+        face3d_net_path = path_of_net_recon_model,
+        outfile=output_path,
+        tmp_dir="temp",
+        crop=[0, -1, 0, -1],
+        re_preprocess=True,  # Set to True if you want to reprocess; False otherwise
+        exp_img="neutral",  # Can be 'smile', 'neutral', or path to an expression image
+        one_shot=False,
+        up_face="original",  # Options: 'original', 'sad', 'angry', 'surprise'
+        LNet_batch_size=16,
+        without_rl1=False
+    )
     
     print('The video is generated')
 
-    # with open(output_path, 'rb') as f:
-    #     video_content = f.read()
+    with open(output_path, 'rb') as f:
+        video_content = f.read()
 
-    # base64_lipsync_video = base64.b64encode(video_content).decode('utf-8')
+    base64_lipsync_video = base64.b64encode(video_content).decode('utf-8')
 
-    video_clip = mp.VideoFileClip(temp_file_path)
+    video_clip = mp.VideoFileClip(output_path)
     duration = video_clip.duration
     
-    app.config['temp_response'] = base64_video
-    app.config['final_video_path'] = temp_file_path
+    app.config['temp_response'] = base64_lipsync_video
+    app.config['final_video_path'] = output_path
     app.config['final_video_duration'] = duration
     
-    return base64_video, temp_file_path, duration
+    return base64_lipsync_video, output_path, duration
 
     # shutil.move(result, save_dir+'.mp4')