---

import cv2
import mss
import numpy as np
import pydirectinput

print("АГРЕССИВНЫЙ ИИ-ШТУРМОВИК ЗАПУЩЕН! Жду входа в бой...")

# Размеры захвата экрана (Твои эталонные 650x370)
monitor = {"top": 0, "left": 0, "width": 650, "height": 370}

# ИДЕАЛЬНАЯ ЦВЕТОВАЯ ПАЛИТРА ИЗ ТВОИХ СКРИНШОТОВ (HSV)
lower_green = np.array([35, 50, 50])
upper_green = np.array([85, 255, 255])

lower_blue = np.array([85, 50, 50])
upper_blue = np.array([135, 255, 255])

lower_red1 = np.array([0, 50, 50])
upper_red1 = np.array([10, 255, 255])
lower_red2 = np.array([170, 50, 50])
upper_red2 = np.array([180, 255, 255])

# Координаты центра кнопки джойстика атаки в BlueStacks на твоем экране
# Прицел атаки находится примерно в точке X: 570, Y: 240
ATTACK_JOYSTICK_X = 570
ATTACK_JOYSTICK_Y = 240

with mss.mss() as sct:
    # Переменная для отслеживания, зажата ли сейчас кнопка ходьбы
    is_moving_forward = False
    
    while True:
        screenshot = sct.grab(monitor)
        frame = np.array(screenshot)
        frame = cv2.cvtColor(frame, cv2.COLOR_BGRA2BGR)
        hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
        
        mask_g = cv2.inRange(hsv, lower_green, upper_green)
        mask_r1 = cv2.inRange(hsv, lower_red1, upper_red1)
        mask_r2 = cv2.inRange(hsv, lower_red2, upper_red2)
        mask_r = cv2.bitwise_or(mask_r1, mask_r2)
        
        contours_g, _ = cv2.findContours(mask_g, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
        contours_r, _ = cv2.findContours(mask_r, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)
        
        in_battle = False
        if len(contours_g) > 0 or len(contours_r) > 0:
            in_battle = True

        if not in_battle:
            if is_moving_forward:
                pydirectinput.keyUp('w')
                is_moving_forward = False
            cv2.imshow("AI Vision - Waiting for Battle", frame)
            if cv2.waitKey(1) & 0xFF == ord('q'):
                break
            continue

        me_x, me_y = None, None
        enemy_detected = False
        enemy_x, enemy_y = None, None
        
        screen_center_x = 650 // 2
        screen_center_y = 370 // 2
        closest_dist = 99999
        
        # 1. Находим себя
        for contour in contours_g:
            if cv2.contourArea(contour) > 120:
                x, y, w, h = cv2.boundingRect(contour)
                if w > h * 2.5 and w < h * 5.5:
                    dist = abs(x + w//2 - screen_center_x) + abs(y + h//2 - screen_center_y)
                    if dist < closest_dist:
                        closest_dist = dist
                        me_x, me_y = x + w//2, y + h//2
                        cv2.rectangle(frame, (x, y), (x + w, y + h), (0, 255, 0), 2)

        # 2. Находим ближайшего врага
        closest_enemy_dist = 99999
        for contour in contours_r:
            if cv2.contourArea(contour) > 120:
                x, y, w, h = cv2.boundingRect(contour)
                if w > h * 2.5 and w < h * 5.5:
                    enemy_detected = True
                    ex, ey = x + w//2, y + h//2
                    if me_x is not None:
                        e_dist = np.sqrt((ex - me_x)**2 + (ey - me_y)**2)
                        if e_dist < closest_enemy_dist:
                            closest_enemy_dist = e_dist
                            enemy_x, enemy_y = ex, ey
                    else:
                        enemy_x, enemy_y = ex, ey
                    cv2.rectangle(frame, (x, y), (x + w, y + h), (0, 0, 255), 2)

        # --- АКТИВНОЕ АГРЕССИВНОЕ УПРАВЛЕНИЕ БЕЗ ФРИЗОВ ---
        if enemy_detected and me_x is not None:
            # 1. Плавное непрерывное движение вперед к врагу
            if not is_moving_forward:
                pydirectinput.keyDown('w')
                is_moving_forward = True
                print("➡️ Иду в наступление на врага!")
            
            # 2. УМНОЕ КИБЕРСПОРТИВНОЕ ПРИЦЕЛИВАНИЕ (AIM)
            # Вычисляем направление вектора от нас к врагу
            dx = enemy_x - me_x
            dy = enemy_y - me_y
            distance = np.sqrt(dx**2 + dy**2)
            
            if distance > 0:
                # Нормализуем вектор прицеливания (делаем длину фиксированной для джойстика атаки)
                # Смещаем мышку в сторону врага на 40 пикселей от центра джойстика атаки
                aim_click_x = int(ATTACK_JOYSTICK_X + (dx / distance) * 40)
                aim_click_y = int(ATTACK_JOYSTICK_Y + (dy / distance) * 40)
                
                # Симулируем точный выстрел на опережение: зажимаем на джойстике, тянем к цели и отпускаем
                pydirectinput.moveTo(ATTACK_JOYSTICK_X, ATTACK_JOYSTICK_Y)
                pydirectinput.mouseDown()
                pydirectinput.moveTo(aim_click_x, aim_click_y)
                pydirectinput.mouseUp()
            
            # Прокаст ульты и гиперзаряда на ход ноги
            pydirectinput.press('f')
            pydirectinput.press('e')
            
        else:
            # Если врагов нет — плавно останавливаемся
            if is_moving_forward:
                pydirectinput.keyUp('w')
                is_moving_forward = False

        cv2.imshow("AI Vision - Pro Battle Mode", frame)
        if cv2.waitKey(1) & 0xFF == ord('q'):
            pydirectinput.keyUp('w')
            break

cv2.destroyAllWindows()
print("Бот полностью остановлен.")