refactor!: output the audio file

apepkuss · apepkuss · commit 5fed9cdefd6e · 2024-11-08T13:01:37.000+08:00
Signed-off-by: Xin Liu &lt;sam@secondstate.io&gt;
diff --git a/src/backend/piper.rs b/src/backend/piper.rs
@@ -53,7 +53,7 @@ pub(crate) async fn audio_speech_handler(req: Request<Body>) -> Response<Body> {
         }
     };
 
-    let file_obj = match llama_core::audio::create_speech(speech_request).await {
+    let audio_buffer = match llama_core::audio::create_speech(speech_request).await {
         Ok(obj) => obj,
         Err(e) => {
             let err_msg = format!("Failed to transcribe the audio. {}", e);
@@ -65,26 +65,14 @@ pub(crate) async fn audio_speech_handler(req: Request<Body>) -> Response<Body> {
         }
     };
 
-    // serialize the file object
-    let s = match serde_json::to_string(&file_obj) {
-        Ok(s) => s,
-        Err(e) => {
-            let err_msg = format!("Failed to serialize the file object. {}", e);
-
-            // log
-            error!(target: "stdout", "{}", &err_msg);
-
-            return error::internal_server_error(err_msg);
-        }
-    };
-
     // return response
     let result = Response::builder()
         .header("Access-Control-Allow-Origin", "*")
         .header("Access-Control-Allow-Methods", "*")
         .header("Access-Control-Allow-Headers", "*")
-        .header("Content-Type", "application/json")
-        .body(Body::from(s));
+        .header("Content-Type", "audio/wav")
+        .header("Content-Disposition", "attachment; filename=audio.wav")
+        .body(Body::from(audio_buffer));
 
     let res = match result {
         Ok(response) => response,
