Just a quick commit gotta modify the seeding script to get the bucket hashes for downloading

crates/lib/mysql_queries/src/payloads/generic_inference_args/tts_payload.rs

@@ -1,7 +1,7 @@
 #[derive(Clone, Debug, Serialize, Deserialize)]
 
 pub struct TTSArgs {
-  #[serde(rename = "e")] 
+  #[serde(rename = "vt")] 
   pub voice_token: String //  varchar 32
 }
 

crates/service/job/inference_job/src/job/job_types/tts/vall_e_x/download_voice_file.rs

@@ -0,0 +1,113 @@
+// TODO when media file is clearified and know where the prompts are stored 
+// use std::path::PathBuf;
+
+// use anyhow::anyhow;
+// use log::{error, warn};
+// use sqlx::MySqlPool;
+// use tempdir::TempDir;
+
+// use buckets::public::media_uploads::original_file::MediaUploadOriginalFilePath;
+// use buckets::public::voice_conversion_results::original_file::VoiceConversionResultOriginalFilePath;
+// use cloud_storage::bucket_client::BucketClient;
+// use jobs_common::job_progress_reporter::job_progress_reporter::JobProgressReporter;
+// use mysql_queries::payloads::generic_inference_args::lipsync_payload::LipsyncAnimationAudioSource;
+// use mysql_queries::queries::generic_inference::job::list_available_generic_inference_jobs::AvailableInferenceJob;
+// use mysql_queries::queries::media_uploads::get_media_upload_for_inference::get_media_upload_for_inference;
+// use mysql_queries::queries::voice_conversion::results::get_voice_conversion_result_for_inference::get_voice_conversion_result_for_inference;
+// use tokens::tokens::media_uploads::MediaUploadToken;
+
+// use crate::job::job_loop::process_single_job_error::ProcessSingleJobError;
+// use crate::util::maybe_download_file_from_bucket::maybe_download_file_from_bucket;
+// use crate::util::scoped_temp_dir_creator::ScopedTempDirCreator;
+
+// pub struct AudioFile {
+//   pub filesystem_path: PathBuf,
+// }
+
+// pub async fn download_voice_file(
+//   audio_source: &LipsyncAnimationAudioSource,
+//   public_bucket_client: &BucketClient,
+//   job_progress_reporter: &mut Box<dyn JobProgressReporter>,
+//   job: &AvailableInferenceJob,
+//   temp_dir_creator: &ScopedTempDirCreator,
+//   work_temp_dir: &TempDir,
+//   mysql_pool: &MySqlPool,
+// ) -> Result<AudioFile, ProcessSingleJobError> {
+
+//   let bucket_object_path;
+
+//   match audio_source {
+//     LipsyncAnimationAudioSource::F(media_file_token) => {
+//       // TODO(bt, 2023-09-08): Implement
+//       return Err(ProcessSingleJobError::NotYetImplemented)
+//     }
+    
+//     LipsyncAnimationAudioSource::U(media_upload_token) => {
+//       let media_upload_token = MediaUploadToken::new_from_str(media_upload_token);
+//       let media_upload_result = get_media_upload_for_inference(
+//         &media_upload_token,
+//         mysql_pool
+//       ).await;
+
+//       let media_upload_result = match media_upload_result {
+//         Ok(Some(result)) => result,
+//         Ok(None) => {
+//           warn!("could not find media upload: {:?}", media_upload_token);
+//           return Err(ProcessSingleJobError::from_anyhow_error(
+//             anyhow!("could not find media upload: {:?}", media_upload_token)))
+//         }
+//         Err(e) => {
+//           error!("could not query media upload: {:?}", e);
+//           return Err(ProcessSingleJobError::from_anyhow_error(e))
+//         }
+//       };
+
+//       let media_upload_bucket_path = MediaUploadOriginalFilePath::from_object_hash(&media_upload_result.public_bucket_directory_hash);
+//       bucket_object_path = media_upload_bucket_path.to_full_object_pathbuf();
+//     }
+//     LipsyncAnimationAudioSource::T(tts_result_token) => {
+//       // TODO(bt, 2023-09-08): Implement
+//       return Err(ProcessSingleJobError::NotYetImplemented)
+//     }
+//     LipsyncAnimationAudioSource::V(voice_conversion_result_token) => {
+//       let voice_conversion_result = get_voice_conversion_result_for_inference(
+//         voice_conversion_result_token,
+//         false,
+//         &mysql_pool,
+//       ).await;
+
+//       let voice_conversion_result = match voice_conversion_result {
+//         Ok(Some(result)) => result,
+//         Ok(None) => {
+//           warn!("could not find voice conversion result: {:?}", voice_conversion_result_token);
+//           return Err(ProcessSingleJobError::from_anyhow_error(
+//             anyhow!("could not find voice conversion result: {:?}", voice_conversion_result_token)))
+//         }
+//         Err(e) => {
+//           error!("could not query voice conversion result: {:?}", e);
+//           return Err(ProcessSingleJobError::from_anyhow_error(e))
+//         }
+//       };
+
+//       let result_bucket_path = VoiceConversionResultOriginalFilePath::from_object_hash(&voice_conversion_result.public_bucket_hash);
+//       bucket_object_path = result_bucket_path.to_full_object_pathbuf();
+//     }
+//   }
+
+//   let downloaded_filesystem_path = work_temp_dir.path().join("audio.bin");
+
+//   maybe_download_file_from_bucket(
+//     "audio file",
+//     &downloaded_filesystem_path,
+//     &bucket_object_path,
+//     public_bucket_client,
+//     job_progress_reporter,
+//     "downloading",
+//     job.id.0,
+//     &temp_dir_creator,
+//   ).await?;
+
+//   Ok(AudioFile {
+//     filesystem_path: downloaded_filesystem_path,
+//   })
+// }

crates/service/job/inference_job/src/job/job_types/tts/vall_e_x/process_job.rs

@@ -42,7 +42,10 @@ pub async fn process_job(args: VALLEXProcessJobArgs<'_>) -> Result<JobSuccessRes
     }
   }
 
-  // Download embeddings files
+  // Download embeddings file using embedding token
+  println!("");
+
+
 
   // Create a temp dir to download things to
 
@@ -55,6 +58,9 @@ pub async fn process_job(args: VALLEXProcessJobArgs<'_>) -> Result<JobSuccessRes
 
   // save result ?
 
+
+   // ==================== UPLOAD AUDIO TO BUCKET ==================== //
+
   Err(ProcessSingleJobError::InvalidJob(anyhow!("this job flow is not yet implemented")))
 }