diff --git a/crates/desktop/src-tauri/src/endpoints/inpaint_image_endpoint.rs b/crates/desktop/src-tauri/src/endpoints/inpaint_image_endpoint.rs new file mode 100644 index 0000000000..c8f351566c --- /dev/null +++ b/crates/desktop/src-tauri/src/endpoints/inpaint_image_endpoint.rs @@ -0,0 +1,24 @@ +use crate::ml::model_cache::ModelCache; +use crate::ml::prompt_cache::PromptCache; +use crate::state::app_config::AppConfig; +use crate::state::app_dir::AppDataRoot; +use image::{ImageReader, RgbImage}; +use log::info; +use tauri::{AppHandle, State}; + + +#[tauri::command] +pub async fn inpaint_image( + image: &str, + mask: &str, + prompt: String, + model_config: State<'_, AppConfig>, + model_cache: State<'_, ModelCache>, + prompt_cache: State<'_, PromptCache>, + app_data_root: State<'_, AppDataRoot>, + app: AppHandle, +) -> Result { + info!("inpaint_endpoint endpoint called."); + + Ok(mask.to_string()) +} diff --git a/crates/desktop/src-tauri/src/endpoints/mod.rs b/crates/desktop/src-tauri/src/endpoints/mod.rs index f49bdce06e..4f2e92c92b 100644 --- a/crates/desktop/src-tauri/src/endpoints/mod.rs +++ b/crates/desktop/src-tauri/src/endpoints/mod.rs @@ -1,5 +1,6 @@ pub mod download_models; pub mod flip_image; +pub mod inpaint_image_endpoint; pub mod realtime_image_endpoint; pub mod remove_background_endpoint; pub mod test_counter; diff --git a/crates/desktop/src-tauri/src/lib.rs b/crates/desktop/src-tauri/src/lib.rs index de8cbed04c..81410b93f6 100644 --- a/crates/desktop/src-tauri/src/lib.rs +++ b/crates/desktop/src-tauri/src/lib.rs @@ -7,6 +7,7 @@ pub mod transfer; pub mod utils; use crate::endpoints::download_models::download_models; +use crate::endpoints::inpaint_image_endpoint::inpaint_image; use crate::endpoints::realtime_image_endpoint::infer_image; use crate::endpoints::remove_background_endpoint::remove_background; use crate::endpoints::test_counter::test_counter; @@ -69,6 +70,7 @@ pub fn run() { .invoke_handler(tauri::generate_handler![ download_models, infer_image, + inpaint_image, remove_background, test_counter, text_to_image,