Update app.py

app.py

@@ -22,6 +22,7 @@ from flask_cors import CORS, cross_origin
 import uuid
 import time
 from PIL import Image
+import moviepy.editor as mp
 
 start_time = time.time()
 
@@ -69,6 +70,7 @@ app.config['temp_response'] = None
 app.config['generation_thread'] = None
 app.config['text_prompt'] = None
 app.config['final_video_path'] = None
+app.config['final_video_duration'] = None
 
 
 
@@ -186,10 +188,15 @@ def main(args):
     result, base64_video,temp_file_path = animate_from_coeff.generate(data, save_dir, pic_path, crop_info, \
                                 enhancer=args.enhancer, background_enhancer=args.background_enhancer, preprocess=args.preprocess)
     
-    print('The generated video is named:')
+    print('The video is generated')
+
+    video_clip = mp.VideoFileClip(temp_file_path)
+    duration = video_clip.duration
+    
     app.config['temp_response'] = base64_video
     app.config['final_video_path'] = temp_file_path
-    return base64_video, temp_file_path
+    app.config['final_video_duration'] = duration
+    return base64_video, temp_file_path, duration
 
     # shutil.move(result, save_dir+'.mp4')
 
@@ -382,6 +389,7 @@ def check_generation_status():
             final_response = app.config['temp_response']
             response["base64_video"] = final_response
             response["text_prompt"] = app.config.get('text_prompt')
+            response["duration"] = app.config.get('final_video_duration')
             response["status"] = "completed"
 
             final_video_path = app.config['final_video_path']