Update app_parallel.py

app_parallel.py

@@ -270,17 +270,17 @@ def generate_chunks(audio_chunks, preprocessed_data, args):
     global TEMP_DIR
     future_to_chunk = {executor.submit(process_chunk, chunk[1], preprocessed_data, args): chunk[0] for chunk in audio_chunks}
     
-    # try:
-    for future in as_completed(future_to_chunk):
-        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({'start_time': idx, 'base64_video': base64_video}).encode('utf-8')
-        except Exception as e:
-            yield f"Task for chunk {idx} failed: {e}\n"
-    # finally:
-    #     if TEMP_DIR:
-    #         custom_cleanup(TEMP_DIR.name)
+    try:
+        for future in as_completed(future_to_chunk):
+            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({'start_time': idx, 'base64_video': base64_video}).encode('utf-8')
+            except Exception as e:
+                yield f"Task for chunk {idx} failed: {e}\n"
+    finally:
+        if TEMP_DIR:
+            custom_cleanup(TEMP_DIR.name)
 
 @app.route("/run", methods=['POST'])
 def parallel_processing():
@@ -358,9 +358,6 @@ def parallel_processing():
         # base64_video, temp_file_path, duration = process_chunk(driven_audio_path, preprocessed_data, args)
     except Exception as e:
         return jsonify({'status': 'error', 'message': str(e)}), 500
-    finally:
-        if TEMP_DIR:
-            custom_cleanup(TEMP_DIR.name)
 
 # @app.route("/stream", methods=["GET"])
 # def stream_results():