Update app.py

app.py

@@ -23,7 +23,7 @@ import uuid
 import time
 from PIL import Image
 import moviepy.editor as mp
-from videoretalking import inference_function
+# from videoretalking import inference_function
 
 start_time = time.time()
 
@@ -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(output_path)
+    video_clip = mp.VideoFileClip(temp_file_path)
     duration = video_clip.duration
     
-    app.config['temp_response'] = base64_lipsync_video
-    app.config['final_video_path'] = output_path
+    app.config['temp_response'] = base64_video
+    app.config['final_video_path'] = temp_file_path
     app.config['final_video_duration'] = duration
     
-    return base64_lipsync_video, output_path, duration
+    return base64_video, temp_file_path, duration
 
     # shutil.move(result, save_dir+'.mp4')