#[allow(unused_imports)]
use crate::prelude::*;
#[allow(unused_imports)]
use serde::{Deserialize, Serialize};
#[allow(unused_imports)]
use std::collections::HashMap;
#[derive(Debug, Serialize, Deserialize, Default)]
pub struct DepthMapInput {
#[serde(skip_serializing_if = "Option::is_none")]
pub a: Option<f64>,
#[serde(skip_serializing_if = "Option::is_none")]
pub bg_th: Option<f64>,
#[serde(skip_serializing_if = "Option::is_none")]
pub depth_and_normal: Option<bool>,
pub image_url: String,
}
#[derive(Debug, Serialize, Deserialize, Default)]
pub struct DepthMapOutput {
pub image: Image,
}
#[derive(Debug, Serialize, Deserialize, Default)]
pub struct HTTPValidationError {
#[serde(skip_serializing_if = "Option::is_none")]
pub detail: Option<Vec<Option<ValidationError>>>,
}
#[derive(Debug, Serialize, Deserialize, Default)]
pub struct Image {
#[serde(skip_serializing_if = "Option::is_none")]
pub content_type: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub file_data: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub file_name: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub file_size: Option<i64>,
#[serde(skip_serializing_if = "Option::is_none")]
pub height: Option<i64>,
pub url: String,
#[serde(skip_serializing_if = "Option::is_none")]
pub width: Option<i64>,
}
#[derive(Debug, Serialize, Deserialize, Default)]
pub struct MarigoldDepthMapInput {
#[serde(skip_serializing_if = "Option::is_none")]
pub ensemble_size: Option<i64>,
pub image_url: String,
#[serde(skip_serializing_if = "Option::is_none")]
pub num_inference_steps: Option<i64>,
#[serde(skip_serializing_if = "Option::is_none")]
pub processing_res: Option<i64>,
}
#[derive(Debug, Serialize, Deserialize)]
pub struct MarigoldDepthMapOutput {
pub image: Image,
}
#[derive(Debug, Serialize, Deserialize, Default)]
pub struct NSFWImageDetectionInput {
pub image_url: String,
}
#[derive(Debug, Serialize, Deserialize, Default)]
pub struct NSFWImageDetectionOutput {
pub nsfw_probability: f64,
}
#[derive(Debug, Serialize, Deserialize, Default)]
pub struct RemoveBackgroundInput {
#[serde(skip_serializing_if = "Option::is_none")]
pub crop_to_bbox: Option<bool>,
pub image_url: String,
#[serde(skip_serializing_if = "Option::is_none")]
pub sync_mode: Option<bool>,
}
#[derive(Debug, Serialize, Deserialize, Default)]
pub struct RemoveBackgroundOutput {
pub image: Image,
}
#[derive(Debug, Serialize, Deserialize, Default)]
pub struct SamInput {
#[serde(skip_serializing_if = "Option::is_none")]
pub better_quality: Option<bool>,
#[serde(skip_serializing_if = "Option::is_none")]
pub black_white: Option<bool>,
#[serde(skip_serializing_if = "Option::is_none")]
pub box_prompt: Option<Vec<Option<Vec<Option<serde_json::Value>>>>>,
#[serde(skip_serializing_if = "Option::is_none")]
pub confidence: Option<f64>,
pub image_url: String,
#[serde(skip_serializing_if = "Option::is_none")]
pub invert: Option<bool>,
#[serde(skip_serializing_if = "Option::is_none")]
pub iou: Option<f64>,
#[serde(skip_serializing_if = "Option::is_none")]
pub point_label: Option<Vec<Option<i64>>>,
#[serde(skip_serializing_if = "Option::is_none")]
pub point_prompt: Option<Vec<Option<Vec<Option<serde_json::Value>>>>>,
#[serde(skip_serializing_if = "Option::is_none")]
pub retina: Option<bool>,
#[serde(skip_serializing_if = "Option::is_none")]
pub size: Option<i64>,
#[serde(skip_serializing_if = "Option::is_none")]
pub text_prompt: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub with_contours: Option<bool>,
}
#[derive(Debug, Serialize, Deserialize, Default)]
pub struct SamOutput {
#[serde(skip_serializing_if = "Option::is_none")]
pub image: Option<Option<Image>>,
}
#[derive(Debug, Serialize, Deserialize, Default)]
pub struct UpscaleInput {
#[serde(skip_serializing_if = "Option::is_none")]
pub face: Option<bool>,
pub image_url: String,
#[serde(skip_serializing_if = "Option::is_none")]
pub model: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub output_format: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub scale: Option<f64>,
#[serde(skip_serializing_if = "Option::is_none")]
pub tile: Option<i64>,
}
#[derive(Debug, Serialize, Deserialize, Default)]
pub struct UpscaleOutput {
pub image: Image,
}
#[derive(Debug, Serialize, Deserialize, Default)]
pub struct ValidationError {
pub loc: Vec<serde_json::Value>,
pub msg: String,
#[serde(rename = "type")]
pub ty: String,
}
pub fn marigold_depth(
params: MarigoldDepthMapInput,
) -> FalRequest<MarigoldDepthMapInput, MarigoldDepthMapOutput> {
FalRequest::new("fal-ai/imageutils/marigold-depth", params)
}