Update app_parallel.py

app_parallel.py

@@ -78,6 +78,8 @@ start_time = None
 audio_chunks = []
 preprocessed_data = None
 args = None
+unique_id  = None
+m3u8_path = None
 
 app.config['temp_response'] = None
 app.config['generation_thread'] = None
@@ -285,7 +287,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):
+def generate_chunks(audio_chunks, preprocessed_data, args, m3u8_path):
     global TEMP_DIR
     future_to_chunk = {executor.submit(process_chunk, chunk[1], preprocessed_data, args): chunk[0] for chunk in audio_chunks}
     
@@ -297,7 +299,7 @@ def generate_chunks(audio_chunks, preprocessed_data, args):
                 # add video for streaming
                 loop = asyncio.new_event_loop()
                 asyncio.set_event_loop(loop)
-                loop.run_until_complete(add_video(temp_file_path))
+                loop.run_until_complete(add_video(temp_file_path, m3u8_path))
                 loop.close()
                 yield json.dumps({'start_time': idx, 'video_index': chunk_idx}).encode('utf-8')
             except Exception as e:
@@ -309,10 +311,11 @@ def generate_chunks(audio_chunks, preprocessed_data, args):
 @app.route("/run", methods=['POST'])
 def parallel_processing():
     global start_time
-    global audio_chunks, preprocessed_data, args
+    global audio_chunks, preprocessed_data, args, m3u8_path
     start_time = time.time()
     global TEMP_DIR
     TEMP_DIR = create_temp_dir()
+    global unique_id
     unique_id = str(uuid.uuid4())
     print('request:',request.method)
     try:
@@ -367,7 +370,8 @@ def parallel_processing():
 
             os.makedirs('lives', exist_ok=True)
             print("Entering generate m3u8")
-            generate_m3u8(audio_duration, f'lives/{unique_id}.m3u8')
+            m3u8_path = f'lives/{unique_id}.m3u8'
+            generate_m3u8(audio_duration, m3u8_path)
             
             return jsonify({'video_url': f'{unique_id}.m3u8'}), 200
                     
@@ -378,12 +382,12 @@ def parallel_processing():
 
 @app.route("/stream", methods=["GET"])
 def stream_results():
-    global audio_chunks, preprocessed_data, args
+    global audio_chunks, preprocessed_data, args, m3u8_path
     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))
+        return stream_with_context(generate_chunks(audio_chunks, preprocessed_data, args, m3u8_path))
     except Exception as e:
         return jsonify({'status': 'error', 'message': str(e)}), 500