Update app.py

app.py

@@ -25,7 +25,7 @@ import time
 start_time = time.time()
 
 class AnimationConfig:
-    def __init__(self, driven_audio_path, source_image_path, result_folder,pose_style,expression_scale,enhancer):
+    def __init__(self, driven_audio_path, source_image_path, result_folder,pose_style,expression_scale,enhancer,still):
         self.driven_audio = driven_audio_path
         self.source_image = source_image_path
         self.ref_eyeblink = None
@@ -43,7 +43,7 @@ class AnimationConfig:
         self.background_enhancer = None
         self.cpu = False
         self.face3dvis = False
-        self.still = False  
+        self.still = still  
         self.preprocess = 'crop'
         self.verbose = False
         self.old_version = False
@@ -209,6 +209,9 @@ def generate_video():
         expression_scale = int(request.form.get('expression_scale', 1))
         enhancer = request.form.get('enhancer', None)
         voice_gender = request.form.get('voice_gender', 'male')
+        still_str = request.form.get('still', 'False')
+        still = still_str.lower() == 'true'
+        print('still', still)
 
         if target_language != 'original_text':
             response = translate_text(text_prompt, target_language)
@@ -257,7 +260,7 @@ def generate_video():
         os.makedirs(result_folder, exist_ok=True)
 
     # 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)
+    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)
         
     if torch.cuda.is_available() and not args.cpu:
         args.device = "cuda"