add small format fix

benchmark.py

@@ -102,7 +102,8 @@ class MapGuesserBenchmark:
                     headless=self.headless,
                 ) as bot:
                     for i, sample in enumerate(test_samples):
-                        print(f"   📍 Sample {i + 1}/{len(test_samples)}")
+                        print('########################################################')
+                        print(f"📍 Sample {i + 1}/{len(test_samples)}")
                         try:
                             result = self.run_single_test_with_bot(bot, sample)
                             all_results.append(result)
@@ -114,7 +115,7 @@ class MapGuesserBenchmark:
                             dist_str = (
                                 f"{distance:.1f} km" if distance is not None else "N/A"
                             )
-                            print(f"   {status} (Distance: {dist_str})")
+                            print(f"{status} (Distance: {dist_str})")
 
                         except KeyboardInterrupt:
                             raise
@@ -165,6 +166,12 @@ class MapGuesserBenchmark:
         # **核心修复**: 从顶级的 "lat" 和 "lng" 键构造真实坐标字典
         true_coords = {"lat": location_data.get("lat"), "lng": location_data.get("lng")}
 
+        true_location = location_data["address"]
+        print(f"🔍 True location: {true_location}")
+        # print true coords
+        print(f"🔍 True coords: {true_coords}")
+        # print predicted coords
+        print(f"🔍 Predicted coords: {predicted_lat_lon}")
         distance_km = self.calculate_distance(true_coords, predicted_lat_lon)
 
         is_success = distance_km is not None and distance_km <= SUCCESS_THRESHOLD_KM

mapcrunch_controller.py

@@ -130,7 +130,7 @@ class MapCrunchController:
 
             # 策略B：优先尝试通过JS直接设置场景，速度最快
             if pano_id and pov:
-                print(f"✅ Loading location via JS Call: PanoID {pano_id[:10]}...")
+                # print(f"✅ Loading location via JS Call: PanoID {pano_id[:10]}...")
                 self.driver.execute_script(
                     "window.panorama.setPano(arguments[0]);"
                     "window.panorama.setPov(arguments[1]);",

readme.md

@@ -1,4 +1,6 @@
+uv sync
+
 python main.py --mode data --samples 50 --urban --no-indoor
 
 
-python main.py --mode benchmark --models gpt-4o
+python main.py --mode benchmark --models gpt-4o --samples 5

select_regions.py

@@ -0,0 +1,66 @@
+import pyautogui
+from pynput import keyboard
+import yaml
+
+
+regions = [
+    "screen_top_left",
+    "screen_bot_right",
+]
+
+map_regions = [
+    "map_top_left",
+    "map_bot_right",
+    "confirm_button",
+    "kodiak",
+    "hobart",
+]
+
+next_round_button = "next_round_button"
+
+coords = []
+
+PRESS_KEY = "a"
+
+
+def on_press(key):
+    try:
+        if key.char == PRESS_KEY:
+            x, y = pyautogui.position()
+            print(x, y)
+            coords.append([x, y])
+            return False
+    except AttributeError:
+        pass
+
+
+def get_coords(players=1):
+    for region in regions:
+        print(f"Move the mouse to the {region} region and press 'a'.")
+        with keyboard.Listener(on_press=on_press) as keyboard_listener:
+            keyboard_listener.join(timeout=40)
+    
+    for p in range(1, players+1):
+        for region in map_regions:
+            region = region + f"_{p}"
+            regions.append(region)
+            print(f"Move the mouse to the {region} region and press 'a'.")
+            with keyboard.Listener(on_press=on_press) as keyboard_listener:
+                keyboard_listener.join(timeout=40)
+
+    regions.append(next_round_button)
+    print(f"Move the mouse to the {next_round_button} region and press 'a'.")
+    with keyboard.Listener(on_press=on_press) as keyboard_listener:
+        keyboard_listener.join(timeout=40)
+
+    screen_regions = {reg: coord for reg, coord in zip(regions, coords)}
+
+    # save dict as a yaml file
+    with open("screen_regions.yaml", "w") as f:
+        yaml.dump(screen_regions, f)
+
+    return screen_regions
+
+
+if __name__ == "__main__":
+    _ = get_coords(players=1)