Update app_parallel.py

app_parallel.py

@@ -109,7 +109,7 @@ def process_chunk(audio_chunk, preprocessed_data, args):
 
     first_coeff_path = preprocessed_data["first_coeff_path"]
     crop_pic_path = preprocessed_data["crop_pic_path"]
-    crop_info_path = preprocessed_data["crop_info"]
+    crop_info_path = preprocessed_data["crop_info_path"]
     with open(crop_info_path , "rb") as f:
             crop_info = pickle.load(f)
 
@@ -220,39 +220,42 @@ def generate_audio(voice_cloning, voice_gender, text_prompt):
 
     return driven_audio_path
 
-# Preprocessing step that runs only once
 def run_preprocessing(args):
     global path_of_lm_croper, path_of_net_recon_model, dir_of_BFM_fitting
-    first_frame_dir = os.path.join(args.result_dir, 'first_frame_dir')
-    os.makedirs(first_frame_dir, exist_ok=True)
-
-    # Check if preprocessed data already exists
-    fixed_temp_dir = "C:/Users/fd01076/Downloads/preprocess_data"
+    fixed_temp_dir = "/tmp/preprocess_data"
     os.makedirs(fixed_temp_dir, exist_ok=True)
     preprocessed_data_path = os.path.join(fixed_temp_dir, "preprocessed_data.pkl")
 
     if os.path.exists(preprocessed_data_path) and args.image_hardcoded == "yes":
+        print("Loading preprocessed data...")
         with open(preprocessed_data_path, "rb") as f:
             preprocessed_data = pickle.load(f)
         print("Loaded existing preprocessed data from:", preprocessed_data_path)
     else:
+        print("Running preprocessing...")
         preprocess_model = CropAndExtract(path_of_lm_croper, path_of_net_recon_model, dir_of_BFM_fitting, args.device)
         first_coeff_path, crop_pic_path, crop_info = preprocess_model.generate(args.source_image, first_frame_dir, args.preprocess, source_image_flag=True)
+        first_coeff_new_path = os.path.join(fixed_temp_dir, os.path.basename(first_coeff_path))
+        crop_pic_new_path = os.path.join(fixed_temp_dir, os.path.basename(crop_pic_path))
+        crop_info_new_path = os.path.join(fixed_temp_dir, "crop_info.pkl")
+        shutil.move(first_coeff_path, first_coeff_new_path)
+        shutil.move(crop_pic_path, crop_pic_new_path)
+
+        with open(crop_info_new_path, "wb") as f:
+            pickle.dump(crop_info, f)
+
+        preprocessed_data = {"first_coeff_path": first_coeff_new_path,
+                            "crop_pic_path": crop_pic_new_path,
+                            "crop_info_path": crop_info_new_path}
 
-        if not first_coeff_path:
-            raise Exception("Failed to get coefficients")
 
-        # Save the preprocessed data
-        preprocessed_data = {
-            "first_coeff_path": first_coeff_path,
-            "crop_pic_path": crop_pic_path,
-            "crop_info": crop_info
-        }
         with open(preprocessed_data_path, "wb") as f:
             pickle.dump(preprocessed_data, f)
+        print(f"Preprocessed data saved to: {preprocessed_data_path}")
 
     return preprocessed_data
 
+    
 client = OpenAI(api_key="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")
 
 def openai_chat_avatar(text_prompt):