Update app.py

app.py

@@ -159,16 +159,23 @@ def main(args):
     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)
-        preprocessed_data = {"first_coeff_path": first_coeff_path,
-                            "crop_pic_path": crop_pic_path,
-                            "crop_info": crop_info}
+        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, os.path.basename(crop_info))
+        shutil.move(first_coeff_path, first_coeff_new_path)
+        shutil.move(crop_pic_path, crop_pic_new_path)
+        shutil.move(crop_info, crop_info_new_path)
+        preprocessed_data = {"first_coeff_path": first_coeff_new_path,
+                            "crop_pic_path": crop_pic_new_path,
+                            "crop_info": 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}")
     
-    print('first_coeff_path ',first_coeff_path)
-    print('crop_pic_path ',crop_pic_path)
+    print('first_coeff_path ',first_coeff_new_path)
+    print('crop_pic_path ',crop_pic_new_path)
+    print('crop_info ',crop_info_new_path)
 
     if first_coeff_path is None:
         print("Can't get the coeffs of the input")