feat: bigger arrow and label

geo_bot.py

@@ -12,6 +12,8 @@ from langchain_google_genai import ChatGoogleGenerativeAI
 
 from mapcrunch_controller import MapCrunchController
 
+# The "Golden" Prompt (v6): Combines clear mechanics with robust strategic principles.
+
 AGENT_PROMPT_TEMPLATE = """
 **Mission:** You are an expert geo-location agent. Your goal is to find clues to determine your location within a limited number of steps.
 
@@ -22,16 +24,19 @@ AGENT_PROMPT_TEMPLATE = """
 ---
 **Core Principles of an Expert Player:**
 
-1.  **Final Step Rule:** If `remaining_steps` is **exactly 1**, this is your last action and it **MUST be `GUESS`**. Do not use your final step for exploration.
-2.  **Be Decisive:** If you find a key clue (a specific address, a unique landmark, or text identifying a city/region), make a `GUESS` immediately. Don't waste steps.
-3.  **Efficient Exploration:**
-    - At intersections or when the view is unpromising, **pan first** to see all directions before moving.
-    - If a path looks barren, don't get stuck moving forward. It's often smarter to turn around (using `PAN` or `MOVE_BACKWARD`).
-4.  **Understand Your Path (The Arrow Heuristic):** The navigation arrows on the ground show the two directions of the **road**. `MOVE_FORWARD` follows the arrow that appears **physically higher on your screen**. `MOVE_BACKWARD` follows the lower arrow. Use this to navigate predictably.
+1.  **Navigate with Labels:** `MOVE_FORWARD` follows the green 'UP' arrow. `MOVE_BACKWARD` follows the red 'DOWN' arrow. These labels are your most reliable compass. If there are no arrows, you cannot move.
+
+2.  **Efficient Exploration (to avoid "Bulldozer" mode):**
+    - **Pan Before You Move:** At a new location or an intersection, it's often wise to use `PAN_LEFT` or `PAN_RIGHT` to quickly survey your surroundings before committing to a move.
+    - **Don't Get Stuck:** If you've moved forward 2-3 times down a path and found nothing but repetitive scenery (like an empty forest or highway), consider it a barren path. It's smarter to turn around (using `PAN`) and check another direction.
+
+3.  **Be Decisive:** If you find a truly definitive clue (like a full, readable address or a sign with a unique town name), `GUESS` immediately. Don't waste steps.
+
+4.  **Final Step Rule:** If `remaining_steps` is **exactly 1**, your action **MUST be `GUESS`**.
 
 ---
 **Context & Task:**
-You will receive a sequence of images from your journey. The last image is your **CURRENT** view. Analyze the full history and your current view, apply the Core Principles, and decide your next action.
+Analyze your full journey history and current view, apply the Core Principles, and decide your next action in the required JSON format.
 
 **Action History:**
 {history_text}
@@ -133,6 +138,8 @@ class GeoBot:
 
             self.controller.setup_clean_environment()
 
+            self.controller.label_arrows_on_screen()
+
             screenshot_bytes = self.controller.take_street_view_screenshot()
             if not screenshot_bytes:
                 print("Failed to take screenshot. Ending agent loop.")

mapcrunch_controller.py

@@ -36,6 +36,91 @@ class MapCrunchController:
             if (infoFirstView) infoFirstView.style.display = 'none';
         """)
 
+    def label_arrows_on_screen(self):
+        """Overlays 'UP' and 'DOWN' labels on the navigation arrows."""
+        try:
+            pov = self.driver.execute_script("return window.panorama.getPov();")
+            links = self.driver.execute_script("return window.panorama.getLinks();")
+        except Exception:
+            return
+
+        if not links or not pov:
+            return
+
+        current_heading = pov["heading"]
+        forward_link = None
+        backward_link = None
+
+        # This logic is identical to your existing `move` function
+        # to ensure stylistic and behavioral consistency.
+        min_forward_diff = 360
+        for link in links:
+            diff = 180 - abs(abs(link["heading"] - current_heading) - 180)
+            if diff < min_forward_diff:
+                min_forward_diff = diff
+                forward_link = link
+
+        target_backward_heading = (current_heading + 180) % 360
+        min_backward_diff = 360
+        for link in links:
+            diff = 180 - abs(abs(link["heading"] - target_backward_heading) - 180)
+            if diff < min_backward_diff:
+                min_backward_diff = diff
+                backward_link = link
+
+        js_script = """
+            document.querySelectorAll('.geobot-arrow-label').forEach(el => el.remove());
+            document.querySelectorAll('path[data-geobot-modified]').forEach(arrow => {
+                arrow.setAttribute('transform', arrow.getAttribute('data-original-transform') || '');
+                arrow.removeAttribute('data-geobot-modified');
+                arrow.removeAttribute('data-original-transform');
+            });
+
+            const modifyAndLabelArrow = (panoId, labelText, color) => {
+                const arrowElement = document.querySelector(`path[pano="${panoId}"]`);
+                if (!arrowElement) return;
+
+                const originalTransform = arrowElement.getAttribute('transform') || '';
+                arrowElement.setAttribute('data-original-transform', originalTransform);
+                arrowElement.setAttribute('transform', `${originalTransform} scale(1.8)`);
+                arrowElement.setAttribute('data-geobot-modified', 'true');
+
+                const rect = arrowElement.getBoundingClientRect();
+                const label = document.createElement('div');
+                label.className = 'geobot-arrow-label';
+                label.style.position = 'fixed';
+                label.style.left = `${rect.left + rect.width / 2}px`;
+                label.style.top = `${rect.top - 45}px`;
+                label.style.transform = 'translateX(-50%)';
+                label.style.padding = '5px 15px';
+                label.style.backgroundColor = 'rgba(0, 0, 0, 0.8)';
+                label.style.color = color;
+                label.style.borderRadius = '8px';
+                label.style.fontSize = '28px';
+                label.style.fontWeight = 'bold';
+                label.style.zIndex = '99999';
+                label.style.pointerEvents = 'none';
+                label.innerText = labelText;
+                document.body.appendChild(label);
+            };
+
+            const forwardPano = arguments[0];
+            const backwardPano = arguments[1];
+
+            if (forwardPano) {
+                modifyAndLabelArrow(forwardPano, 'UP', '#76FF03');
+            }
+            if (backwardPano && backwardPano !== forwardPano) {
+                modifyAndLabelArrow(backwardPano, 'DOWN', '#F44336');
+            }
+        """
+
+        forward_pano = forward_link["pano"] if forward_link else None
+        backward_pano = backward_link["pano"] if backward_link else None
+
+        self.driver.execute_script(js_script, forward_pano, backward_pano)
+        time.sleep(0.2)
+
     def get_available_actions(self) -> List[str]:
         """
         Checks for movement links via JavaScript.