daniel543x/artcraft
1
0 Fork 0
mirror of https://github.com/storytold/artcraft.git synced 2026-11-03 00:47:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add tts result audio path to result endpoint

Browse Source
This commit is contained in:
Brandon Thomas committed 2023-03-16 10:31:28 -04:00
1 parent b8fd0486c7
commit 80da0d9649
1 file changed
+16 -9
Download Patch File Download Diff File Expand all files Collapse all files
crates/lib/database_queries/src/queries/tts/tts_results/list_tts_results_query_builder.rs
+16 -9
View File
@@ -29,6 +29,8 @@ pub struct TtsInferenceRecordForList {
pub tts_model_title: String,
pub raw_inference_text: String,
pub public_bucket_wav_audio_path: String,
pub maybe_creator_user_token: Option<String>,
pub maybe_creator_username: Option<String>,
pub maybe_creator_display_name: Option<String>,
@@ -125,17 +127,18 @@ impl ListTtsResultsQueryBuilder {
.map(|raw_result| raw_result.tts_result_id);
let inference_results = internal_results
.iter()
.into_iter()
.map(|r| {
TtsInferenceRecordForList {
tts_result_token: r.tts_result_token.clone(),
tts_model_token: r.tts_model_token.clone(),
tts_model_title: r.tts_model_title.clone(),
raw_inference_text: r.raw_inference_text.clone(),
maybe_creator_user_token: r.maybe_creator_user_token.clone(),
maybe_creator_username: r.maybe_creator_username.clone(),
maybe_creator_display_name: r.maybe_creator_display_name.clone(),
maybe_creator_result_id: r.maybe_creator_result_id.map(|v| v as u64).clone(),
tts_result_token: r.tts_result_token,
tts_model_token: r.tts_model_token,
tts_model_title: r.tts_model_title,
raw_inference_text: r.raw_inference_text,
public_bucket_wav_audio_path: r.public_bucket_wav_audio_path,
maybe_creator_user_token: r.maybe_creator_user_token,
maybe_creator_username: r.maybe_creator_username,
maybe_creator_display_name: r.maybe_creator_display_name,
maybe_creator_result_id: r.maybe_creator_result_id.map(|v| v as u64),
file_size_bytes: if r.file_size_bytes > 0 { r.file_size_bytes as u32 } else { 0 },
duration_millis: if r.duration_millis > 0 { r.duration_millis as u32 } else { 0 },
visibility: Visibility::from_str(&r.creator_set_visibility).unwrap_or(Visibility::Public),
@@ -202,6 +205,8 @@ SELECT
tts_models.title as tts_model_title,
tts_results.raw_inference_text as raw_inference_text,
tts_results.public_bucket_wav_audio_path,
users.token as maybe_creator_user_token,
users.username as maybe_creator_username,
users.display_name as maybe_creator_display_name,
@@ -312,6 +317,8 @@ pub struct RawInternalTtsRecord {
pub tts_model_title: String,
pub raw_inference_text: String,
pub public_bucket_wav_audio_path: String,
pub maybe_creator_user_token : Option<String>,
pub maybe_creator_username: Option<String>,
pub maybe_creator_display_name: Option<String>,