report new prompt fields in API

This commit is contained in:
Brandon Thomas committed 2024-07-07 11:13:03 -04:00
1 parent 776c086470
commit 45792890b7
1 file changed
+18 -2
@@ -51,12 +51,15 @@ pub struct PromptInfo {
/// Note: Prompts may or may not be compatible across systems.
pub prompt_type: PromptType,
/// Positive prompt
/// Positive prompt (technically optional, but usually present)
pub maybe_positive_prompt: Option<String>,
/// Negative prompt
/// Negative prompt (optional)
pub maybe_negative_prompt: Option<String>,
/// Scheduled / travel prompt (optional)
pub maybe_travel_prompt: Option<String>,
/// If a "style" was used, this is the name of it.
/// This might not be present for all types of inference
/// and typically only applies to video style transfer.
@@ -69,6 +72,9 @@ pub struct PromptInfo {
/// Typically only for video style transfer.
pub maybe_strength: Option<f32>,
/// If a frame skip setting was used.
pub maybe_frame_skip: Option<u8>,
/// If a face detailer was used.
/// This might not be present for all types of inference
/// and typically only applies to video style transfer.
@@ -205,6 +211,10 @@ pub async fn get_prompt_handler(
let mut maybe_strength = None;
let mut maybe_inference_duration_millis = None;
let mut maybe_global_ipa_image_token = None;
let mut maybe_travel_prompt = None;
let mut maybe_frame_skip = None;
// Flags
let mut used_face_detailer = false;
let mut used_upscaler = false;
let mut lipsync_enabled = false;
@@ -223,6 +233,10 @@ pub async fn get_prompt_handler(
maybe_strength = inner_payload.strength;
maybe_inference_duration_millis = inner_payload.inference_duration_millis;
maybe_global_ipa_image_token = inner_payload.global_ipa_token.clone();
maybe_travel_prompt = inner_payload.travel_prompt.clone();
maybe_frame_skip = inner_payload.frame_skip;
// Flags
used_face_detailer = inner_payload.used_face_detailer.unwrap_or(false);
used_upscaler = inner_payload.used_upscaler.unwrap_or(false);
lipsync_enabled = inner_payload.lipsync_enabled.unwrap_or(false);
@@ -253,6 +267,7 @@ pub async fn get_prompt_handler(
maybe_strength,
maybe_positive_prompt: result.maybe_positive_prompt,
maybe_negative_prompt: result.maybe_negative_prompt,
maybe_travel_prompt,
maybe_style_name,
maybe_inference_duration_millis,
used_face_detailer,
@@ -264,6 +279,7 @@ pub async fn get_prompt_handler(
created_at: result.created_at,
maybe_moderator_fields,
maybe_global_ipa_image_token,
maybe_frame_skip,
},
};