Update app_parallel.py

app_parallel.py

@@ -83,6 +83,7 @@ args = None
 unique_id  = None
 m3u8_path = None
 audio_duration = None
+driven_audio_path = None
 
 app.config['temp_response'] = None
 app.config['generation_thread'] = None
@@ -338,7 +339,7 @@ def close_m3u8(m3u8_path: str):
       
 @app.route("/run", methods=['POST'])
 def parallel_processing():
-    global start_time
+    global start_time, driven_audio_path
     global audio_chunks, preprocessed_data, args, m3u8_path, audio_duration
     start_time = time.time()
     global TEMP_DIR
@@ -391,10 +392,14 @@ def parallel_processing():
             ref_pose_video_path = None
             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, image_hardcoded=image_hardcoded)
             preprocessed_data = run_preprocessing(args)
-            chunk_duration = 3
-            print(f"Splitting the audio into {chunk_duration}-second chunks...")
-            audio_chunks, audio_duration = split_audio(driven_audio_path, TEMP_DIR, chunk_duration=chunk_duration)
-            print(f"Audio has been split into {len(audio_chunks)} chunks: {audio_chunks}")
+            # chunk_duration = 3
+            # print(f"Splitting the audio into {chunk_duration}-second chunks...")
+            # audio_chunks, audio_duration = split_audio(driven_audio_path, TEMP_DIR, chunk_duration=chunk_duration)
+            # print(f"Audio has been split into {len(audio_chunks)} chunks: {audio_chunks}")
+            start_time = 0
+            audio_clip = mp.AudioFileClip(driven_audio_path)
+            audio_duration = audio_clip.duration
+            audio_chunks.append((start_time, driven_audio_path))
 
             os.makedirs('lives', exist_ok=True)
             print("Entering generate m3u8")