import sys;sys.path.insert(0, '/home/pi/Install/lib/python3.13/site-packages')
import cv2,imutils
rumena_spodnja = (20, 100, 100) 
rumena_zgornja = (35, 255, 255)

kamera = cv2.VideoCapture("posnetki/posnetek.mp4")

while True:
    vrne, slika = kamera.read()
    if not vrne:
        print("Ne sprejmem slike, konec posnetka?")
        break
    plavo_rdece = cv2.GaussianBlur(slika, (11, 11), 0)
    width, height = slika.shape[:2]
    hsv = cv2.cvtColor(plavo_rdece, cv2.COLOR_BGR2HSV)
    maska = cv2.inRange(hsv, rumena_spodnja, rumena_zgornja)
    maska = cv2.erode(maska, None, iterations=2)
    maska = cv2.dilate(maska, None, iterations=2)
    meje = cv2.findContours(maska.copy(), cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_SIMPLE)
    meje = imutils.grab_contours(meje)
    center = None
    if len(meje) > 0:
        c = max(meje, key=cv2.contourArea)
        ((x, y), radius) = cv2.minEnclosingCircle(c)
        M = cv2.moments(c)
        center = (int(M["m10"] / M["m00"]), int(M["m01"] / M["m00"]))
        if radius > 10: # Lepši pogled centra
            cv2.circle(slika, (int(x), int(y)), int(radius), (0, 255, 255), 5)
            cv2.circle(slika, center, 5, (0, 0, 255), -1)
    cv2.imshow('ESC - konec', slika)
    if cv2.waitKey(100) == 27: #Bere tipkovnico, ESC za konec
        break
kamera.release()
cv2.destroyAllWindows()