Update app_parallel.py

app_parallel.py

@@ -80,6 +80,7 @@ preprocessed_data = None
 args = None
 unique_id  = None
 m3u8_path = None
+audio_duration = None
 
 app.config['temp_response'] = None
 app.config['generation_thread'] = None
@@ -287,7 +288,7 @@ def split_audio(audio_path, TEMP_DIR, chunk_duration):
     return audio_chunks, total_duration
 
 # Generator function to yield chunk results as they are processed
-def generate_chunks(audio_chunks, preprocessed_data, args, m3u8_path):
+def generate_chunks(audio_chunks, preprocessed_data, args, m3u8_path, audio_duration):
     global TEMP_DIR
     future_to_chunk = {executor.submit(process_chunk, chunk[1], preprocessed_data, args): chunk[0] for chunk in audio_chunks}
     
@@ -299,7 +300,7 @@ def generate_chunks(audio_chunks, preprocessed_data, args, m3u8_path):
                 # add video for streaming
                 loop = asyncio.new_event_loop()
                 asyncio.set_event_loop(loop)
-                loop.run_until_complete(add_video(temp_file_path, m3u8_path))
+                loop.run_until_complete(add_video(temp_file_path, m3u8_path, audio_duration))
                 loop.close()
                 yield json.dumps({'start_time': idx, 'video_index': chunk_idx}).encode('utf-8')
             except Exception as e:
@@ -311,7 +312,7 @@ def generate_chunks(audio_chunks, preprocessed_data, args, m3u8_path):
 @app.route("/run", methods=['POST'])
 def parallel_processing():
     global start_time
-    global audio_chunks, preprocessed_data, args, m3u8_path
+    global audio_chunks, preprocessed_data, args, m3u8_path, audio_duration
     start_time = time.time()
     global TEMP_DIR
     TEMP_DIR = create_temp_dir()
@@ -382,12 +383,12 @@ def parallel_processing():
 
 @app.route("/stream", methods=["GET"])
 def stream_results():
-    global audio_chunks, preprocessed_data, args, m3u8_path
+    global audio_chunks, preprocessed_data, args, m3u8_path, audio_duration
     print("audio_chunks",audio_chunks)
     print("preprocessed_data",preprocessed_data)
     print("args",args)
     try:
-        return stream_with_context(generate_chunks(audio_chunks, preprocessed_data, args, m3u8_path))
+        return stream_with_context(generate_chunks(audio_chunks, preprocessed_data, args, m3u8_path, audio_duration))
     except Exception as e:
         return jsonify({'status': 'error', 'message': str(e)}), 500