main.py

import cv2
import mediapipe as mp

from modules.face_analyzer import FaceAnalyzer
from modules.hand_analyzer import HandAnalyzer
from modules.canvas_manager import CanvasManager

def main():
    face_analyzer = FaceAnalyzer()
    hand_analyzer = HandAnalyzer()

    cap = cv2.VideoCapture(0)

    # Lecture d'une premiere frame pour obtenir la resolution
    success, image = cap.read()
    if not success:
        print("Erreur: Impossible d'acceder a la camera.")
        return

    img_h, img_w, _ = image.shape
    canvas_manager = CanvasManager(img_w, img_h)

    print("Appuyez sur 'q' pour quitter la fenetre de la camera.")

    while cap.isOpened():
        success, image = cap.read()
        if not success:
            break

        image = cv2.flip(image, 1)  # Effet miroir naturel

        image_rgb = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
        mp_image = mp.Image(image_format=mp.ImageFormat.SRGB, data=image_rgb)

        # Analyser le visage et dessiner le HUD
        face_analyzer.draw_face_hud(image, mp_image, img_w, img_h)

        # Analyser les mains et appliquer les interactions de dessin
        hand_analyzer.analyze_hands(image, mp_image, img_w, img_h, canvas_manager)

        # Fusionner la toile et la camera
        final_image = canvas_manager.overlay_on(image)

        cv2.imshow("Face Tracker & AR Drawing", final_image)

        if cv2.waitKey(5) & 0xFF == ord("q"):
            break

    cap.release()
    cv2.destroyAllWindows()


if __name__ == "__main__":
    main()