Update app.py

app.py

@@ -24,11 +24,30 @@ print("Dataset loaded successfully.")
 from scipy.spatial import KDTree
 
 import time
-# Assuming positions_array is already defined
-# Extract 'final_positions' from each item in the dataset and compile them into a NumPy array
-final_positions = [position for item in dataset['data']['final_positions'] for position in item]
+
+# Initialize an empty list to store positions
+final_positions = []
+
+# Initialize a counter for progress tracking
+counter = 0
+
+# Total positions count (assuming each item['data']['final_positions'] is a list of positions)
+total_positions = sum(len(item['data']['final_positions']) for item in dataset)
+
+for item in dataset:
+    for position in item['data']['final_positions']:
+        final_positions.append(position)
+        counter += 1
+        
+        # Print progress every 100 steps
+        if counter % 100 == 0 or counter == total_positions:
+            print(f"Processed {counter} / {total_positions} positions...")
+
+# Convert the list of positions to a NumPy array
 positions_array = np.array([[p['x'], p['y'], p['z']] for p in final_positions])
 
+print("Completed processing positions into an array.")
+
 def estimate_time_and_build_kdtree(data):
     start_time = time.time()
     print("Estimating time based on dataset size...")