Updated app.py with reference pose path

app.py

@@ -25,11 +25,11 @@ import time
 start_time = time.time()
 
 class AnimationConfig:
-    def __init__(self, driven_audio_path, source_image_path, result_folder,pose_style,expression_scale,enhancer,still,preprocess):
+    def __init__(self, driven_audio_path, source_image_path, result_folder,pose_style,expression_scale,enhancer,still,preprocess,ref_pose_video_path):
         self.driven_audio = driven_audio_path
         self.source_image = source_image_path
         self.ref_eyeblink = None
-        self.ref_pose = None
+        self.ref_pose = ref_pose_video_path
         self.checkpoint_dir = './checkpoints'
         self.result_dir = result_folder
         self.pose_style = pose_style
@@ -213,6 +213,7 @@ def generate_video():
         print('still', still)
         preprocess = request.form.get('preprocess', 'crop')
         print('preprocess selected: ',preprocess)
+        ref_pose_video = request.files['ref_pose']
 
         if target_language != 'original_text':
             response = translate_text(text_prompt, target_language)
@@ -264,8 +265,13 @@ def generate_video():
         result_folder = os.path.join(save_dir, "results")
         os.makedirs(result_folder, exist_ok=True)
 
+        with tempfile.NamedTemporaryFile(suffix=".mp4", prefix="ref_pose_",dir=TEMP_DIR.name, delete=False) as temp_file:
+                ref_pose_video_path = temp_file.name
+                ref_pose_video.save(ref_pose_video_path)
+                print('ref_pose_video_path',ref_pose_video_path)
+
     # Example of using the class with some hypothetical paths
-    args = AnimationConfig(driven_audio_path=driven_audio_path, source_image_path=source_image_path, result_folder=result_folder, pose_style=pose_style, expression_scale=expression_scale, enhancer=enhancer,still=still,preprocess=preprocess)
+    args = AnimationConfig(driven_audio_path=driven_audio_path, source_image_path=source_image_path, result_folder=result_folder, pose_style=pose_style, expression_scale=expression_scale, enhancer=enhancer,still=still,preprocess=preprocess,ref_pose_video_path=ref_pose_video_path)
         
     if torch.cuda.is_available() and not args.cpu:
         args.device = "cuda"