Update app.py

app.py

@@ -3,6 +3,8 @@ from PIL import Image
 import sys
 import tempfile
 from facefusion import core
+from moviepy.editor import VideoFileClip
+
 
 def run_cli(cli_args):
   old_argv = sys.argv
@@ -12,13 +14,20 @@ def run_cli(cli_args):
   finally:
     sys.argv = old_argv
 
+
+def get_video_duration(video_path):
+    video = VideoFileClip(video_path)
+    duration = video.duration  # duration in seconds
+    return duration
+
+
 def swap_faces(source_image_path, target_image_path, enhance_face=True, enhance_frame=True):
   provider = 'cuda'
 
   target_ext = target_image_path.split('.')[-1]
   output_image_file = tempfile.NamedTemporaryFile(suffix=f'.{target_ext}')
   output_image_path = output_image_file.name
-
+  duration = get_video_duration(target_image_path)
   print(source_image_path)
   print(target_image_path)
   print(output_image_path)
@@ -34,7 +43,7 @@ def swap_faces(source_image_path, target_image_path, enhance_face=True, enhance_
   ]
 
   cli_args += [ '--frame-processors', 'face_swapper' ]
-  cli_args += ['--trim-frame-end','2']
+  cli_args += ['--trim-frame-end',str(duration)]
   if enhance_face:
     cli_args += [
       'face_enhancer',
@@ -67,5 +76,6 @@ demo = gr.Interface(
   allow_flagging="never"
 )
 
-if __name__ == "__main__":
-    demo.launch(show_error=True)
+if __name__ == "__main__":    
+    demo.queue(api_open=True)
+    demo.launch(show_error=True, show_api=True)