Update app.py

app.py

@@ -8,8 +8,6 @@ import uuid
 from requests.exceptions import RequestException, HTTPError, Timeout, ConnectionError
 from datasets import load_dataset
 
-# Assuming necessary imports are done above
-
 # Global initialization
 print("Initializing GPT-2 Tokenizer and Model...")
 tokenizer = GPT2Tokenizer.from_pretrained('gpt2')
@@ -20,33 +18,25 @@ print("Loading dataset...")
 dataset = load_dataset("visionlab/block-towers-10k-3s-trajectory-scale1", split='train')
 print("Dataset loaded successfully.")
 
-# Preprocess dataset to build a lookup table for fast access
-lookup_table = {}
-grid_size = 1.0  # Adjust based on your dataset's scale and distribution
-print("Building lookup table...")
-for item in dataset:
-    # Adjusting to access 'final_positions' as provided in the dataset structure
-    for position in item['data']['final_positions']:
-        xyz = [position['x'], position['y'], position['z']]
-        grid_key = (round(xyz[0]/grid_size), round(xyz[1]/grid_size), round(xyz[2]/grid_size))
-        if grid_key not in lookup_table:
-            lookup_table[grid_key] = xyz
-        else:
-            # For simplicity, this keeps the first encountered position for each grid key.
-            # You might want to handle multiple positions within the same grid cell differently.
-            continue
-print("Lookup table built.")
-
 class SecondLifeNavigator:
     def __init__(self):
-        self.lookup_table = lookup_table
-        print("SecondLifeNavigator initialized with the lookup table.")
+        self.dataset = dataset
+        print("SecondLifeNavigator initialized with the dataset.")
 
     def determine_action_sequence(self, current_xyz):
         print(f"Determining action sequence for position: {current_xyz}")
-        grid_key = (round(current_xyz[0]/grid_size), round(current_xyz[1]/grid_size), round(current_xyz[2]/grid_size))
-        nearest_position = self.lookup_table.get(grid_key)
-        
+        min_distance = float('inf')
+        nearest_position = None
+
+        # Iterate directly through dataset items to find the nearest position
+        for item in self.dataset:
+            for position in item['data']['final_positions']:
+                xyz = [position['x'], position['y'], position['z']]
+                distance = np.sqrt((xyz[0] - current_xyz[0])**2 + (xyz[1] - current_xyz[1])**2 + (xyz[2] - current_xyz[2])**2)
+                if distance < min_distance:
+                    min_distance = distance
+                    nearest_position = xyz
+
         if nearest_position:
             action_sequence = f"Move to position {nearest_position[0]}, {nearest_position[1]}, {nearest_position[2]}"
             print(f"Action sequence determined: {action_sequence}")
@@ -78,13 +68,12 @@ class SecondLifeNavigator:
         except RequestException as req_err:
             print(f"Error sending command to Corrade: {req_err}")
         return None
-            
+
 def create_interface():
     def navigate(current_state):
         endpoint = os.getenv('CORRADE_ENDPOINT', '')  # Ensure this environment variable is correctly set.
         sl_navigator = SecondLifeNavigator()
-        # Use determine_action_sequence instead of run
-        action_sequence = sl_navigator.determine_action_sequence(current_state)
+        action_sequence = sl_navigator.determine_action_sequence(json.loads(current_state))
         random_id = str(uuid.uuid4())
         return random_id + ": " + action_sequence