//! YOLO (Object Detection) processing module

use anyhow::Result;
use momentry_core::ui::progress::{ProcessorType, ProgressState, ProgressUi};
use momentry_core::OutputDir;
use std::path::Path;
use std::sync::{Arc, Mutex};

/// Process YOLO module
pub async fn process_yolo_module(
    yolo_path: &Path,
    video_path: &str,
    uuid: &str,
    progress_state: &Arc<Mutex<ProgressState>>,
    ui: &Arc<Mutex<Option<ProgressUi>>>,
) -> Result<()> {
    {
        let mut state = progress_state.lock().unwrap();
        state.get_processor(ProcessorType::Yolo).start(1);
    }

    let yolo_result = momentry_core::core::processor::process_yolo(
        video_path,
        yolo_path.to_str().unwrap(),
        Some(uuid),
    )
    .await?;

    let yolo_json = serde_json::to_string_pretty(&yolo_result)?;
    std::fs::write(yolo_path, &yolo_json)?;

    let output_dir = OutputDir::new();
    let _ = output_dir.backup_file(uuid, "yolo.json");

    println!("  ✓ YOLO saved: {} frames", yolo_result.frame_count);

    {
        let mut state = progress_state.lock().unwrap();
        state
            .get_processor(ProcessorType::Yolo)
            .complete(&format!("{} frames", yolo_result.frame_count));
    }

    if let Some(ref mut ui) = *ui.lock().unwrap() {
        let _ = ui.render();
    }

    Ok(())
}
