Update app_parallel.py

app_parallel.py

@@ -292,7 +292,7 @@ def split_audio(audio_path, TEMP_DIR, chunk_duration):
 #     return int(match.group(1)) if match else -1  # Return -1 if no match is found, handle appropriately.
 # Generator function to yield chunk results as they are processed
 
-def generate_chunks(audio_chunks, preprocessed_data, args, m3u8_path, audio_duration):
+def generate_chunks(audio_chunks, preprocessed_data, args, m3u8_path, audio_duration, start_time):
     global TEMP_DIR
     future_to_chunk = {executor.submit(process_chunk, chunk[1], preprocessed_data, args): chunk[0] for chunk in audio_chunks}
     processed_chunks = {chunk[0]: None for chunk in audio_chunks}
@@ -310,7 +310,9 @@ def generate_chunks(audio_chunks, preprocessed_data, args, m3u8_path, audio_dura
                         asyncio.set_event_loop(loop)
                         loop.run_until_complete(add_video(processed_chunks[expected_start_time], m3u8_path, audio_duration))
                         loop.close()
-                        yield json.dumps({'start_time': expected_start_time, 'video_index': yielded_count}).encode('utf-8')
+                        end_time = time.time()
+                        elapsed_time = end_time - start_time
+                        yield json.dumps({'start_time': expected_start_time, 'video_index': yielded_count, 'elapsed_time': elapsed_time}).encode('utf-8')
                         processed_chunks[expected_start_time] = None
                         yielded_count += 1
                     else:
@@ -396,12 +398,12 @@ def parallel_processing():
 
 @app.route("/stream", methods=["GET"])
 def stream_results():
-    global audio_chunks, preprocessed_data, args, m3u8_path, audio_duration
+    global audio_chunks, preprocessed_data, args, m3u8_path, audio_duration, start_time
     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, audio_duration))
+        return stream_with_context(generate_chunks(audio_chunks, preprocessed_data, args, m3u8_path, audio_duration, start_time))
     except Exception as e:
         return jsonify({'status': 'error', 'message': str(e)}), 500