#[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 BoxPrompt {
#[serde(skip_serializing_if = "Option::is_none")]
pub frame_index: Option<i64>,
#[serde(skip_serializing_if = "Option::is_none")]
pub x_max: Option<i64>,
#[serde(skip_serializing_if = "Option::is_none")]
pub x_min: Option<i64>,
#[serde(skip_serializing_if = "Option::is_none")]
pub y_max: Option<i64>,
#[serde(skip_serializing_if = "Option::is_none")]
pub y_min: Option<i64>,
}
#[derive(Debug, Serialize, Deserialize, Default)]
pub struct File {
#[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>,
pub url: String,
}
#[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 PointPrompt {
#[serde(skip_serializing_if = "Option::is_none")]
pub frame_index: Option<i64>,
#[serde(skip_serializing_if = "Option::is_none")]
pub label: Option<i64>,
#[serde(skip_serializing_if = "Option::is_none")]
pub x: Option<i64>,
#[serde(skip_serializing_if = "Option::is_none")]
pub y: Option<i64>,
}
#[derive(Debug, Serialize, Deserialize, Default)]
pub struct SAM2AutomaticSegmentationInput {
pub image_url: String,
#[serde(skip_serializing_if = "Option::is_none")]
pub min_mask_region_area: Option<i64>,
#[serde(skip_serializing_if = "Option::is_none")]
pub output_format: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub points_per_side: Option<i64>,
#[serde(skip_serializing_if = "Option::is_none")]
pub pred_iou_thresh: Option<f64>,
#[serde(skip_serializing_if = "Option::is_none")]
pub stability_score_thresh: Option<f64>,
#[serde(skip_serializing_if = "Option::is_none")]
pub sync_mode: Option<bool>,
}
#[derive(Debug, Serialize, Deserialize, Default)]
pub struct SAM2AutomaticSegmentationOutput {
pub combined_mask: Image,
pub individual_masks: Vec<Image>,
}
#[derive(Debug, Serialize, Deserialize, Default)]
pub struct SAM2ImageInput {
#[serde(skip_serializing_if = "Option::is_none")]
pub box_prompts: Option<Vec<Option<BoxPrompt>>>,
pub image_url: String,
#[serde(skip_serializing_if = "Option::is_none")]
pub output_format: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub prompts: Option<Vec<Option<PointPrompt>>>,
#[serde(skip_serializing_if = "Option::is_none")]
pub sync_mode: Option<bool>,
}
#[derive(Debug, Serialize, Deserialize)]
pub struct SAM2ImageOutput {
pub image: Image,
}
#[derive(Debug, Serialize, Deserialize, Default)]
pub struct SAM2RLEOutput {
pub rle: RleProperty,
}
#[derive(Debug, Serialize, Deserialize, Default)]
pub struct SAM2VideoOutput {
pub video: File,
}
#[derive(Debug, Serialize, Deserialize, Default)]
pub struct SAM2VideoRLEInput {
#[serde(skip_serializing_if = "Option::is_none")]
pub apply_mask: Option<bool>,
#[serde(skip_serializing_if = "Option::is_none")]
pub box_prompts: Option<Vec<Option<BoxPrompt>>>,
#[serde(skip_serializing_if = "Option::is_none")]
pub mask_url: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub prompts: Option<Vec<Option<PointPrompt>>>,
pub video_url: String,
}
#[derive(Debug, Serialize, Deserialize, Default)]
pub struct ValidationError {
pub loc: Vec<serde_json::Value>,
pub msg: String,
#[serde(rename = "type")]
pub ty: String,
}
#[derive(Debug, Serialize, Deserialize, smart_default::SmartDefault)]
#[allow(non_camel_case_types)]
pub enum RleProperty {
#[default]
String(String),
Array(Vec<String>),
}
pub fn image(params: SAM2ImageInput) -> FalRequest<SAM2ImageInput, SAM2ImageOutput> {
FalRequest::new("fal-ai/sam2/image", params)
}