from ultralytics import YOLO
import cv2
import numpy as np
import requests
import time

model = YOLO("yolo11n.pt") 

#try to fetch the image from the given url
def fetch_image(url):
    try:
        response = requests.get(url)
        response.raise_for_status()
        image_array = np.frombuffer(response.content, np.uint8)
        image = cv2.imdecode(image_array, cv2.IMREAD_COLOR)
        return image
    except requests.RequestException as e:
        print(f"Error: Could not fetch image - {e}")
        return None

# URL of the photostream
url = "http://145.92.224.21/image"

while True:
    frame = fetch_image(url)
    if frame is None:
        print("Error: Could not fetch image, retrying...")
        time.sleep(1)  # Wait for 1 second before retrying
        continue

    # Predict on the frame
    results = model(frame)

    # Display the results
    results[0].show()

    # Exit if 'q' is pressed
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

cv2.destroyAllWindows()