Update app_celery.py

app_celery.py

@@ -150,42 +150,45 @@ def main(args):
                                                                             #  source_image_flag=True, pic_size=args.size)
 
 
-    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 image_hardcoded == "yes":
-        print("Loading preprocessed data...")
-        with open(preprocessed_data_path, "rb") as f:
-            preprocessed_data = pickle.load(f)
-        first_coeff_new_path = preprocessed_data["first_coeff_path"]
-        crop_pic_new_path = preprocessed_data["crop_pic_path"]
-        crop_info_path = preprocessed_data["crop_info_path"]
-        with open(crop_info_path, "rb") as f:
-            crop_info = pickle.load(f)
+    # 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 image_hardcoded == "yes":
+    #     print("Loading preprocessed data...")
+    #     with open(preprocessed_data_path, "rb") as f:
+    #         preprocessed_data = pickle.load(f)
+    #     first_coeff_new_path = preprocessed_data["first_coeff_path"]
+    #     crop_pic_new_path = preprocessed_data["crop_pic_path"]
+    #     crop_info_path = preprocessed_data["crop_info_path"]
+    #     with open(crop_info_path, "rb") as f:
+    #         crop_info = pickle.load(f)
             
-        print(f"Loaded existing preprocessed data from: {preprocessed_data_path}")
+    #     print(f"Loaded existing preprocessed data from: {preprocessed_data_path}")
     
-    else:
-        print("Running preprocessing...")
-        first_coeff_path, crop_pic_path, crop_info =  preprocess_model.generate(pic_path, 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)
+    # else:
+    #     print("Running preprocessing...")
+    #     first_coeff_path, crop_pic_path, crop_info =  preprocess_model.generate(pic_path, 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}
+    #     preprocessed_data = {"first_coeff_path": first_coeff_new_path,
+    #                         "crop_pic_path": crop_pic_new_path,
+    #                         "crop_info_path": crop_info_new_path}
+
+
+    #     with open(preprocessed_data_path, "wb") as f:
+    #         pickle.dump(preprocessed_data, f)
+    #     print(f"Preprocessed data saved to: {preprocessed_data_path}")
 
+    first_coeff_path, crop_pic_path, crop_info =  preprocess_model.generate(pic_path, first_frame_dir, args.preprocess, source_image_flag=True)
 
-        with open(preprocessed_data_path, "wb") as f:
-            pickle.dump(preprocessed_data, f)
-        print(f"Preprocessed data saved to: {preprocessed_data_path}")
     
     print('first_coeff_path ',first_coeff_new_path)
     print('crop_pic_path ',crop_pic_new_path)