import asyncio

import RPi.GPIO as GPIO

from ultralytics import YOLO

# LED의 핀 번호 (BCM)

led = 12

# GPIO 모드 설정

GPIO.setmode(GPIO.BCM)

# LED를 위한 GPIO 설정

GPIO.setup(led, GPIO.OUT)

# YOLO 객체 생성

model = YOLO("/home/esp/sharefolder/best.onnx")

lock = asyncio.Lock()

async def detect_s():

    global class_names

    while True:

        class_names = []

        results = model.predict(source='0', show=True, imgsz=[320, 320])

        async with lock:

            for result in results.pandas().xyxy[0]['name']:

                class_names.append(result)

        await asyncio.sleep(0.1)

async def control_led():

    while True:

        with lock:

            if 'person' in class_names or 'vehicle' in class_names:

                print("Detected person or vehicle, turning on LED...")

                # LED를 켜는 코드를 작성하세요

                GPIO.output(led, GPIO.HIGH)

                await asyncio.sleep(1)

                GPIO.output(led, GPIO.LOW)

            else:

                print("No person or vehicle detected.")

        await asyncio.sleep(1)

async def main():

    # 객체 탐지 비동기 작업 생성

    detect_task = asyncio.create_task(detect_s())

    # LED 제어 비동기 작업 생성

    led_task = asyncio.create_task(control_led())

    # 비동기 작업이 완료될 때까지 대기

    await asyncio.gather(detect_task, led_task)

# 이벤트 루프 실행

asyncio.run(main())