"""
Realtime Detection untuk Abjad SIBI dengan visualisasi MediaPipe landmarks.
"""

import sys
import cv2
from pathlib import Path

sys.path.insert(0, str(Path(__file__).parent.parent))

from utils.inference import GestureRecognizer
from utils.config import MODEL_DIR, WINDOW_SIZE


def run_realtime(category: str = "abjad_sibi", camera_id: int = 0):
    model_path = MODEL_DIR / "abjad_sibi" / "model_abjad_sibi.pt"
    encoder_path = MODEL_DIR / "abjad_sibi" / "label_encoder.json"

    if not model_path.exists():
        print(f"Error: Model tidak ditemukan di {model_path}")
        return

    recognizer = GestureRecognizer(
        model_path,
        encoder_path,
        window_size=WINDOW_SIZE,
        draw_landmarks=True
    )
    cap = cv2.VideoCapture(camera_id)

    if not cap.isOpened():
        print(f"Error: Tidak dapat membuka camera {camera_id}")
        return

    print("\n" + "="*60)
    print(f"REALTIME RECOGNITION - {category.upper()}")
    print("="*60)
    print("Tekan 'q' untuk quit, 'r' untuk reset, 'l' toggle landmarks")
    print("="*60 + "\n")

    show_landmarks = True

    try:
        while True:
            ret, frame = cap.read()
            if not ret:
                break

            if show_landmarks:
                frame = recognizer.draw_landmarks_on_frame(frame)

            label, confidence = recognizer.process_frame(frame)

            if label is not None:
                text = f"{label} ({confidence*100:.1f}%)"
                cv2.putText(frame, text, (10, 50),
                           cv2.FONT_HERSHEY_SIMPLEX, 1.5, (0, 255, 0), 3)

            buffer_len = len(recognizer.buffer)
            cv2.putText(frame, f"Buffer: {buffer_len}/{recognizer.window_size}",
                       (10, 100), cv2.FONT_HERSHEY_SIMPLEX, 0.8, (255, 255, 255), 2)

            landmark_status = "Landmarks: ON" if show_landmarks else "Landmarks: OFF"
            cv2.putText(frame, landmark_status, (10, 130),
                       cv2.FONT_HERSHEY_SIMPLEX, 0.6, (0, 255, 255), 2)

            if label is None:
                cv2.putText(frame, "Tunjukkan gestur isyarat...", (10, 160),
                           cv2.FONT_HERSHEY_SIMPLEX, 0.7, (0, 0, 255), 2)

            cv2.imshow(f"Realtime - {category}", frame)
            key = cv2.waitKey(1) & 0xFF
            if key == ord('q'):
                break
            elif key == ord('r'):
                recognizer.reset()
            elif key == ord('l'):
                show_landmarks = not show_landmarks

    finally:
        cap.release()
        cv2.destroyAllWindows()
        recognizer.close()


def main():
    import argparse
    parser = argparse.ArgumentParser()
    parser.add_argument("--camera", type=int, default=0)
    args = parser.parse_args()
    run_realtime("abjad_sibi", args.camera)


if __name__ == "__main__":
    main()