Update app.py

app.py

@@ -20,7 +20,7 @@ import moviepy.editor as mp
 import requests
 import json
 import pickle
-import re
+# import re
 # from videoretalking import inference_function
 # import base64
 # import gfpgan_enhancer
@@ -284,10 +284,10 @@ def run_preprocessing(args):
 
     return preprocessed_data
 
-def remove_brackets(text):
-    # Use regex to remove content in brackets at the end of the text
-    cleaned_text = re.sub(r'\s*\[.*?\]\s*$', '', text)
-    return cleaned_text.strip()
+# def remove_brackets(text):
+#     # Use regex to remove content in brackets at the end of the text
+#     cleaned_text = re.sub(r'\s*\[.*?\]\s*$', '', text)
+#     return cleaned_text.strip()
 
 @app.route("/run", methods=['POST'])
 def generate_video():
@@ -360,9 +360,8 @@ def generate_video():
             source_image_path = save_uploaded_file(source_image, 'source_image.png',TEMP_DIR)
             print(source_image_path)
 
-            text_prompt_without_cititation = remove_brackets(text_prompt)
-            print('Final output text prompt for avatar: ',text_prompt_without_cititation)
-            driven_audio_path = generate_audio(voice_cloning, voice_gender, text_prompt_without_cititation)
+            
+            driven_audio_path = generate_audio(voice_cloning, voice_gender, text_prompt)
             
             save_dir = tempfile.mkdtemp(dir=TEMP_DIR.name)
             result_folder = os.path.join(save_dir, "results")
@@ -424,6 +423,7 @@ def generate_video():
                     "message": "Video processed and saved successfully.",
                     "video_url": video_url,
                     "text_prompt": text_prompt,
+                    "time_taken": time_taken,
                     "status": "success"
                 })
         else: