small tweaks

Author: dragonheart107

module/statistics/autosearch_reward.py

@@ -15,6 +15,10 @@ def wait_until_reward_stable(self, timeout=8.6, required_consecutive_matches=3,
         checks if all rewards have appeared, via first doing a predicted grid, 3 types to account for the Ui
         change due to Meta xp, then verifies via amount ocr, after that waits till the reward area has no changes
         e.g all drops appeared
+
+        timeout: may or may not need adjustment, after gathering data\\
+        required_consecutive_matches: 3 seems a good compromise\\
+        similarity_threshold: in testing the similarity ranged in 0.002- 0.01 ranges so opted for 0.999
         """
         logger.info('wait until area stable')
 
@@ -68,7 +72,7 @@ def wait_until_reward_stable(self, timeout=8.6, required_consecutive_matches=3,
                         logger.info("Grid 0 validated, checking Grid 1 for potentially more rewards...")
                         grid1_x, grid1_y = GRID_ORIGINS[1]
                         grid1 = ButtonGrid(origin=(grid1_x, grid1_y), **GRID_PARAMS)
-                        grid1_matched = self._verify_grid_via_ocr(grid1, 1, original_bgr)
+                        grid1_matched = self._verify_grid_via_ocr(grid1, 1, original_bgr, check_first_button_only=True)
 
                         if grid1_matched:
                             logger.info("Grid 1 also validated - using Grid 1 for wider coverage")
@@ -116,13 +120,14 @@ def wait_until_reward_stable(self, timeout=8.6, required_consecutive_matches=3,
             logger.error(f'Stabilization failure: {str(e)}')
             return False
 
-    def _verify_grid_via_ocr(self, grid, grid_idx, bgr_image):
+    def _verify_grid_via_ocr(self, grid, grid_idx, bgr_image, check_first_button_only=False):
         """
-        Double checks if grid correct via the item amount on the bottom right of grid buttons
-        Enhanced with better logging for grid comparison
+        double checks if grid correct via the item amount on the bottom right of grid buttons
+        special handling for grid0 vs grid1
         """
         amount_ocr = AmountOcr([], threshold=96, name='Amount_ocr')
         detected_amounts = []
+        first_button_has_amount = False
 
         for btn_idx, btn in enumerate(grid.buttons):
             try:
@@ -140,24 +145,39 @@ def _verify_grid_via_ocr(self, grid, grid_idx, bgr_image):
                 if len(amount_img.shape) == 2:
                     amount_img = cv2.cvtColor(amount_img, cv2.COLOR_GRAY2BGR)
 
-                # 4. OCR detection
+                # 4. OCR just like the one in ItemGrid in item.py
                 amount = amount_ocr.ocr([amount_img], direct_ocr=True)[0]
                 if amount > 0:
                     detected_amounts.append(amount)
                     logger.info(f"Grid {grid_idx}, Button {btn_idx}: amount={amount}")
+                    
+                    # Track if first button (index 0) has amount
+                    if btn_idx == 0:
+                        first_button_has_amount = True
 
             except Exception as e:
                 logger.warning(f"Grid {grid_idx}, Button {btn_idx} check failed: {str(e)}")
                 continue
 
-        # Log summary for this grid
+        # Determine validation result
         total_detected = len(detected_amounts)
-        if total_detected > 0:
-            logger.info(f"Grid {grid_idx} summary: {total_detected} buttons with amounts detected")
-            return True
+        
+        if check_first_button_only:
+            # For Grid 1 validation: only valid if first button has amount
+            if first_button_has_amount:
+                logger.info(f"Grid {grid_idx} summary: First button validated with amount - Grid 1 captures additional reward!")
+                return True
+            else:
+                logger.info(f"Grid {grid_idx} summary: First button has no amount - Grid 1 offers no advantage")
+                return False
         else:
-            logger.info(f"Grid {grid_idx} summary: No amounts detected")
-            return False
+            # Standard validation: any button with amount is sufficient
+            if total_detected > 0:
+                logger.info(f"Grid {grid_idx} summary: {total_detected} buttons with amounts detected")
+                return True
+            else:
+                logger.info(f"Grid {grid_idx} summary: No amounts detected")
+                return False
 
     def reward_area_stable_check(self, monitor_area, timeout, required_matches, similarity_threshold):
         """