c4181/whisper-worker
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

delete files after processing

Browse source
This commit is contained in:
Christopher Moyer 2024-12-17 23:25:00 -05:00
parent 2743c8ca3c
commit 7f1fa73bb0
1 changed files with 11 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
src/main.rs
View file

@ -91,11 +91,13 @@ _|"""""|_|"""""|_|"""""|_|"""""|_|"""""|_|"""""|_|"""""| {======|_|"""""|_|"""""
error!("Stdout: {}", String::from_utf8_lossy(&result.stdout)); error!("Stdout: {}", String::from_utf8_lossy(&result.stdout));
error!("Stderr: {}", String::from_utf8_lossy(&result.stderr)); error!("Stderr: {}", String::from_utf8_lossy(&result.stderr));
delete_file(path);
return delivery.ack(BasicAckOptions::default()).await.unwrap(); return delivery.ack(BasicAckOptions::default()).await.unwrap();
} }
Err(error) => { Err(error) => {
error!("Failed to transcribe {}, {}", &transcription_request.audio_file_path, error); error!("Failed to transcribe {}, {}", &transcription_request.audio_file_path, error);
delete_file(path);
return delivery.ack(BasicAckOptions::default()).await.unwrap(); return delivery.ack(BasicAckOptions::default()).await.unwrap();
} }
} }
@ -106,6 +108,7 @@ _|"""""|_|"""""|_|"""""|_|"""""|_|"""""|_|"""""|_|"""""| {======|_|"""""|_|"""""
Err(_) => { Err(_) => {
error!("Failed to read transcript file {}", &transcript_path.display()); error!("Failed to read transcript file {}", &transcript_path.display());
delete_file(path);
return delivery.ack(BasicAckOptions::default()).await.unwrap(); return delivery.ack(BasicAckOptions::default()).await.unwrap();
} }
}; };
@ -115,6 +118,8 @@ _|"""""|_|"""""|_|"""""|_|"""""|_|"""""|_|"""""|_|"""""| {======|_|"""""|_|"""""
Err(error) => { Err(error) => {
error!("Failed to send message to meilisearch, {}", error); error!("Failed to send message to meilisearch, {}", error);
delete_file(path);
delete_file(transcript_path.as_path());
return delivery.ack(BasicAckOptions::default()).await.unwrap(); return delivery.ack(BasicAckOptions::default()).await.unwrap();
} }
}; };
@ -128,6 +133,8 @@ _|"""""|_|"""""|_|"""""|_|"""""|_|"""""|_|"""""|_|"""""| {======|_|"""""|_|"""""
_ => {} _ => {}
} }
delete_file(path);
delete_file(transcript_path.as_path());
delivery.ack(BasicAckOptions::default()).await.unwrap() delivery.ack(BasicAckOptions::default()).await.unwrap()
} }
} }
@ -178,6 +185,10 @@ async fn wait_for_task_to_complete(client: &Client, task: &TaskInfo) -> Result<T
} }
} }
fn delete_file(file_path: &Path) {
fs::remove_file(file_path).unwrap_or_default();
}
#[derive(Serialize, Deserialize, Default, Debug)] #[derive(Serialize, Deserialize, Default, Debug)]
struct TranscriptionRequest { struct TranscriptionRequest {
audio_file_path: String, audio_file_path: String,