upload thumbnail cover image

crates/schema/database/mysql_queries/src/queries/media_files/create/generic_insert/insert_media_file_generic.rs

@@ -88,6 +88,9 @@ pub struct InsertArgs<'a> {
     pub generated_by_worker: Option<&'a str>,
     pub generated_by_cluster: Option<&'a str>,
 
+    // Cover image (e.g. thumbnail for 3D splats)
+    pub maybe_cover_image_media_file_token: Option<&'a MediaFileToken>,
+
     // Moderation details (deprecated)
     pub maybe_mod_user_token: Option<&'a UserToken>,
 }
@@ -179,9 +182,11 @@ pub async fn insert_media_file_generic(
 
             extra_file_modification_info = ?,
 
-            maybe_mod_user_token = ?, 
-            is_generated_on_prem = ?, 
-            generated_by_worker = ?, 
+            maybe_cover_image_media_file_token = ?,
+
+            maybe_mod_user_token = ?,
+            is_generated_on_prem = ?,
+            generated_by_worker = ?,
             generated_by_cluster = ?
         "#,
         result_token,
@@ -233,6 +238,8 @@ pub async fn insert_media_file_generic(
 
         extra_file_modification_info,
 
+        args.maybe_cover_image_media_file_token.map(|t| t.as_str()),
+
         args.maybe_mod_user_token,
         args.is_generated_on_prem,
         args.generated_by_worker,

crates/schema/database/mysql_queries/src/queries/media_files/create/generic_insert/insert_media_file_generic_from_job.rs

@@ -120,6 +120,7 @@ pub async fn insert_media_file_generic_from_job(
         maybe_text_transcript: args.maybe_text_transcript,
         maybe_title: args.maybe_title,
         maybe_video_encoding: args.maybe_video_encoding,
+        maybe_cover_image_media_file_token: None,
         media_class: args.media_class,
         media_type: args.media_type,
         origin_category: args.origin_category,

crates/schema/database/mysql_queries/src/queries/media_files/create/insert_builder/media_file_insert_builder.rs

@@ -73,6 +73,9 @@ pub struct MediaFileInsertBuilder {
   // // If batch generated, this is the batch token.
   maybe_batch_generation_token: Option<BatchGenerationToken>,
 
+  // Cover image (e.g. thumbnail for 3D splats)
+  maybe_cover_image_media_file_token: Option<MediaFileToken>,
+
   // Storage details
   public_bucket_directory_hash: Option<MediaFileBucketPath>, // NB: Non-nullable field(s)
 
@@ -107,6 +110,7 @@ impl MediaFileInsertBuilder {
       maybe_origin_filename: None,
       maybe_engine_category: None,
       maybe_batch_generation_token: None,
+      maybe_cover_image_media_file_token: None,
       public_bucket_directory_hash: None,
       maybe_prompt_token: None,
     }
@@ -250,6 +254,11 @@ impl MediaFileInsertBuilder {
     self
   }
 
+  pub fn maybe_cover_image_media_file_token(mut self, maybe_token: Option<&MediaFileToken>) -> Self {
+    self.maybe_cover_image_media_file_token = maybe_token.cloned();
+    self
+  }
+
   pub fn public_bucket_directory_hash(mut self, public_bucket_directory_hash: &MediaFileBucketPath) -> Self {
     self.public_bucket_directory_hash = Some(public_bucket_directory_hash.clone());
     self
@@ -308,6 +317,7 @@ impl MediaFileInsertBuilder {
       maybe_creator_file_synthetic_id_category: IdCategory::MediaFile, // TODO: Remove this
       maybe_creator_category_synthetic_id_category: IdCategory::FileUpload, // TODO: Remove this
       maybe_extra_media_info: None, // TODO
+      maybe_cover_image_media_file_token: self.maybe_cover_image_media_file_token.as_ref(),
       is_generated_on_prem: false, // TODO
       generated_by_worker: None, // TODO
       generated_by_cluster: None, // TODO

crates/schema/database/mysql_queries/src/queries/media_files/create/insert_media_file_from_studio_scene_render.rs

@@ -112,6 +112,7 @@ pub async fn insert_media_file_from_studio_scene_render(
     maybe_prompt_token: None,
     maybe_mod_user_token: None,
     maybe_engine_category: None,
+    maybe_cover_image_media_file_token: None,
   }).await?;
 
   Ok(new_media_token)

crates/service/job/worldlabs_job/src/process_job/process_successful_job.rs

@@ -1,5 +1,5 @@
 use anyhow::anyhow;
-use log::{error, info};
+use log::{error, info, warn};
 
 use bucket_paths::legacy::typified_paths::public::media_files::bucket_file_path::MediaFileBucketPath;
 use enums::by_table::generic_inference_jobs::inference_result_type::InferenceResultType;
@@ -12,11 +12,15 @@ use hashing::sha256::sha256_hash_bytes::sha256_hash_bytes;
 use mysql_queries::queries::generic_inference::worldlabs::list_pending_worldlabs_jobs::PendingWorldlabsJob;
 use mysql_queries::queries::media_files::create::insert_builder::media_file_insert_builder::MediaFileInsertBuilder;
 use mysql_queries::queries::generic_inference::web::mark_generic_inference_job_successfully_done_by_token::mark_generic_inference_job_successfully_done_by_token;
+use tokens::tokens::media_files::MediaFileToken;
 use world_labs_api::api::requests::get_operation::get_operation::GetOperationResponse;
 use crate::job_dependencies::JobDependencies;
 
-const PREFIX: &str = "artcraft_";
-const SUFFIX: &str = ".ceramic.spz"; // NB: "ceramic" triggers some heuristics in Artcraft, such as flipping the model upside down
+const SPLAT_PREFIX: &str = "artcraft_";
+const SPLAT_SUFFIX: &str = ".ceramic.spz"; // NB: "ceramic" triggers some heuristics in Artcraft, such as flipping the model upside down
+
+const THUMB_PREFIX: &str = "artcraft_";
+const THUMB_SUFFIX: &str = ".png";
 
 /// Download the completed splat, upload to bucket, create media file record, and mark job done.
 pub async fn process_successful_job(
@@ -36,6 +40,29 @@ pub async fn process_successful_job(
       operation.operation_id
     ))?;
 
+  // --- Step 1: Download and upload the thumbnail as a cover image. ---
+
+  let maybe_cover_token = match world.assets.as_ref().and_then(|a| a.thumbnail_url.as_ref()) {
+    Some(thumbnail_url) => {
+      match download_and_upload_thumbnail(deps, job, operation, thumbnail_url).await {
+        Ok(token) => Some(token),
+        Err(err) => {
+          warn!(
+            "Failed to create thumbnail cover image for operation {}: {:?}. Continuing without cover.",
+            operation.operation_id, err
+          );
+          None
+        }
+      }
+    }
+    None => {
+      info!("No thumbnail_url for operation {}; skipping cover image.", operation.operation_id);
+      None
+    }
+  };
+
+  // --- Step 2: Download and upload the splat. ---
+
   info!(
     "Downloading splat for operation {} from: {}",
     operation.operation_id, splat_url
@@ -61,7 +88,7 @@ pub async fn process_successful_job(
     .map_err(|err| anyhow!("error hashing splat: {:?}", err))?;
 
   // Build the bucket path.
-  let bucket_path = MediaFileBucketPath::generate_new(Some(PREFIX), Some(SUFFIX));
+  let bucket_path = MediaFileBucketPath::generate_new(Some(SPLAT_PREFIX), Some(SPLAT_SUFFIX));
 
   let object_path = bucket_path.get_full_object_path_str();
 
@@ -96,6 +123,7 @@ pub async fn process_successful_job(
     .file_size_bytes(splat_bytes.len() as u64)
     .checksum_sha2(&checksum)
     .maybe_prompt_token(job.maybe_prompt_token.as_ref())
+    .maybe_cover_image_media_file_token(maybe_cover_token.as_ref())
     .public_bucket_directory_hash(&bucket_path)
     .insert_pool(&deps.mysql_pool)
     .await
@@ -131,3 +159,70 @@ pub async fn process_successful_job(
 
   Ok(())
 }
+
+/// Download a thumbnail image, upload it to the public bucket, and create a media file record for it.
+async fn download_and_upload_thumbnail(
+  deps: &JobDependencies,
+  job: &PendingWorldlabsJob,
+  operation: &GetOperationResponse,
+  thumbnail_url: &str,
+) -> AnyhowResult<MediaFileToken> {
+  info!(
+    "Downloading thumbnail for operation {} from: {}",
+    operation.operation_id, thumbnail_url
+  );
+
+  let thumb_bytes: Vec<u8> = reqwest::get(thumbnail_url)
+    .await
+    .map_err(|err| anyhow!("reqwest error downloading thumbnail: {:?}", err))?
+    .bytes()
+    .await
+    .map_err(|err| anyhow!("error reading thumbnail bytes: {:?}", err))?
+    .to_vec();
+
+  info!(
+    "Downloaded {} thumbnail bytes for operation {}",
+    thumb_bytes.len(),
+    operation.operation_id
+  );
+
+  let checksum = sha256_hash_bytes(&thumb_bytes)
+    .map_err(|err| anyhow!("error hashing thumbnail: {:?}", err))?;
+
+  let bucket_path = MediaFileBucketPath::generate_new(Some(THUMB_PREFIX), Some(THUMB_SUFFIX));
+  let object_path = bucket_path.get_full_object_path_str();
+
+  info!("Uploading thumbnail to public bucket at path: {}", object_path);
+
+  deps
+    .public_bucket_client
+    .upload_file_with_content_type_process(object_path, &thumb_bytes, "image/png")
+    .await
+    .map_err(|err| anyhow!("error uploading thumbnail to bucket: {:?}", err))?;
+
+  let thumb_token = MediaFileInsertBuilder::new()
+    .maybe_creator_user(job.maybe_creator_user_token.as_ref())
+    .maybe_creator_anonymous_visitor(job.maybe_creator_anonymous_visitor_token.as_ref())
+    .creator_ip_address(&job.creator_ip_address)
+    .creator_set_visibility(job.creator_set_visibility)
+    .media_file_class(MediaFileClass::Image)
+    .media_file_type(MediaFileType::Png)
+    .media_file_origin_category(MediaFileOriginCategory::Inference)
+    .media_file_origin_product_category(MediaFileOriginProductCategory::WorldGeneration)
+    .mime_type("image/png")
+    .file_size_bytes(thumb_bytes.len() as u64)
+    .checksum_sha2(&checksum)
+    .maybe_prompt_token(job.maybe_prompt_token.as_ref())
+    .public_bucket_directory_hash(&bucket_path)
+    .insert_pool(&deps.mysql_pool)
+    .await
+    .map_err(|err| anyhow!("error inserting thumbnail media file record: {:?}", err))?;
+
+  info!(
+    "Created thumbnail media file {} for operation {}.",
+    thumb_token.as_str(),
+    operation.operation_id
+  );
+
+  Ok(thumb_token)
+}