use axum::{
    body::Body,
    response::Response
};
use crate::ApiResult;
use pyo3::prelude::*;

const MODULE: &str = include_str!("model/model.py");
const MODEL: &[u8] = include_bytes!("model/model.keras");

pub fn router() -> axum::Router {
    axum::Router::new()
        .route("/", 
            axum::routing::post(get_predict)
        )
}

async fn get_predict() -> ApiResult {
    //TODO: If api key is correct
    // Extract image from body of req
    let result = Python::with_gil(|py| -> PyResult<()> {
        let predict: Py<PyAny> = PyModule::from_code_bound(py, MODULE, "model.py", "model")?
            .getattr("predict")?.into();
        //TODO: Return result
        Ok(())
    });

    //TODO: Return results 
    Ok(Response::builder().body(Body::from(""))?)
}