added api for ai; got backend working

This commit is contained in:
2026-03-26 23:39:31 -06:00
parent 164b2f87d4
commit 4a857d8cbf
20 changed files with 1436 additions and 280 deletions

View File

@ -0,0 +1,66 @@
use std::process::Command;
use serde_json;
use serde::{Deserialize, Serialize};
#[derive(Serialize, Deserialize, Clone, Debug)]
pub struct BackgroundRemovalStatus {
pub available: bool,
}
#[derive(Serialize, Deserialize, Clone, Debug)]
pub struct BackgroundRemovalResult {
pub output_path: String,
}
/// Check if background removal is available
pub fn is_available() -> Result<bool, String> {
let python_exe = crate::paths::python_exe();
let python_exe = python_exe.to_str().unwrap_or_default();
let script_path = crate::paths::backend_script("background_removal.py");
let script_path = script_path.to_str().unwrap_or_default();
let output = Command::new(python_exe)
.args(&[script_path, "is_available"])
.output()
.map_err(|e| format!("Failed to run Python script: {}", e))?;
if !output.status.success() {
let stderr = String::from_utf8_lossy(&output.stderr);
return Err(format!("Python script failed: {}", stderr));
}
let stdout = String::from_utf8_lossy(&output.stdout);
let result: BackgroundRemovalStatus = serde_json::from_str(&stdout.trim())
.map_err(|e| format!("Failed to parse JSON: {}", e))?;
Ok(result.available)
}
/// Remove background on export (placeholder for Phase 5)
pub fn remove_background_on_export(
input_path: &str,
output_path: &str,
replacement: &str,
replacement_value: &str,
) -> Result<String, String> {
let python_exe = crate::paths::python_exe();
let python_exe = python_exe.to_str().unwrap_or_default();
let script_path = crate::paths::backend_script("background_removal.py");
let script_path = script_path.to_str().unwrap_or_default();
let output = Command::new(python_exe)
.args(&[script_path, "remove_background_on_export", input_path, output_path, replacement, replacement_value])
.output()
.map_err(|e| format!("Failed to run Python script: {}", e))?;
if !output.status.success() {
let stderr = String::from_utf8_lossy(&output.stderr);
return Err(format!("Python script failed: {}", stderr));
}
let stdout = String::from_utf8_lossy(&output.stdout);
let result: BackgroundRemovalResult = serde_json::from_str(&stdout.trim())
.map_err(|e| format!("Failed to parse JSON: {}", e))?;
Ok(result.output_path)
}