Update app_parallel.py

app_parallel.py

@@ -267,21 +267,21 @@ def split_audio(audio_path, chunk_duration=5):
         with tempfile.NamedTemporaryFile(suffix=f"_chunk_{start_time}-{end_time}.wav", prefix="audio_chunk_", dir=TEMP_DIR.name, delete=False) as temp_file:
             chunk_path = temp_file.name
             chunk.write_audiofile(chunk_path)
-            audio_chunks.append(chunk_path)
+            audio_chunks.append((start_time, chunk_path))
     
     return audio_chunks
 
 # Generator function to yield chunk results as they are processed
 def generate_chunks(audio_chunks, preprocessed_data, args):
-    future_to_chunk = {executor.submit(process_chunk, chunk, preprocessed_data, args): chunk for chunk in audio_chunks}
+    future_to_chunk = {executor.submit(process_chunk, chunk[1], preprocessed_data, args): chunk[0] for chunk in audio_chunks}
     
     for future in as_completed(future_to_chunk):
-        chunk = future_to_chunk[future]  # Get the original chunk that was processed
+        idx = future_to_chunk[future]  # Get the original chunk that was processed
         try:
             base64_video, temp_file_path = future.result()  # Get the result of the completed task
-            yield json.dumps({'index': chunk, 'path': temp_file_path}).encode('utf-8')
+            yield json.dumps({'start_time': idx, 'path': temp_file_path}).encode('utf-8')
         except Exception as e:
-            yield f"Task for chunk {chunk} failed: {e}\n"
+            yield f"Task for chunk {idx} failed: {e}\n"
 
 @app.route("/run", methods=['POST'])
 def parallel_processing():