use debian:bookworm-slim as runtime image

Dockerfile

@@ -1,4 +1,4 @@
-FROM rust:latest as chef
+FROM rust:slim as chef
 RUN apt-get update && apt-get install -y curl
 RUN cargo install cargo-chef
 WORKDIR /app
@@ -14,8 +14,7 @@ COPY . .
 RUN cargo chef cook --release --recipe-path recipe.json
 RUN cargo build --release
 
-FROM ubuntu:latest as runtime
-
+FROM debian:bookworm-slim as runtime
 RUN apt-get update && apt-get install -y ca-certificates && rm -rf /var/lib/apt/lists/*
 WORKDIR /app
 COPY --from=builder /app/target/release/polyhedron .

src/asr/aws.rs

@@ -12,8 +12,8 @@ use tokio::select;
 use tokio::sync::broadcast::Receiver;
 use tokio_stream::Stream;
 use futures_util::TryStreamExt;
-use tracing::trace;
-use crate::asr::{ASR, Event, slice_i16_to_u8};
+use tracing::{trace, warn};
+use crate::asr::{ASR, Event, slice_i16_to_u8_le};
 
 pub struct AwsAsr {
     speaker_voice_channel: tokio::sync::mpsc::Sender<Vec<i16>>,
@@ -42,7 +42,7 @@ impl AwsAsr {
             let fut = async {
                 let input_stream = stream! {
                     while let Some(raw) = speaker_voice_rx.recv().await {
-                        let reshape = slice_i16_to_u8(&raw);
+                        let reshape = slice_i16_to_u8_le(&raw);
                         yield Ok(AudioStream::AudioEvent(AudioEvent::builder().audio_chunk(Blob::new(reshape)).build()));
                     }
                 };
@@ -59,7 +59,6 @@ impl AwsAsr {
                 let output_stream = to_stream(output);
                 output_stream
                     .try_for_each(|text| async {
-                        trace!("here");
                         let _ = shared_speaker_transcript.send(text);
                         Ok(())
                     })
@@ -69,7 +68,7 @@ impl AwsAsr {
             select! {
                 res = fut => {
                     if let Err(e) = res {
-                        println!("Error: {:?}", e);
+                        warn!("Error: {:?}", e);
                     }
                 }
                 _ = drop_rx => {}

src/asr/mod.rs

@@ -18,7 +18,7 @@ pub(crate) trait ASR {
 }
 
 
-pub fn slice_i16_to_u8(slice: &[i16]) -> Vec<u8> {
+pub fn slice_i16_to_u8_le(slice: &[i16]) -> Vec<u8> {
     slice
         .iter()
         .flat_map(|&sample| {
@@ -27,3 +27,13 @@ pub fn slice_i16_to_u8(slice: &[i16]) -> Vec<u8> {
         .collect()
 }
 
+pub fn slice_i16_to_u8_be(slice: &[i16]) -> Vec<u8> {
+    slice
+        .iter()
+        .flat_map(|&sample| {
+            [(sample >> 8) as u8, sample as u8]
+        })
+        .collect()
+}
+
+

src/lesson.rs

@@ -8,7 +8,7 @@ use std::io::BufRead;
 use std::ops::Deref;
 use std::sync::{Arc, Weak};
 use tokio::sync::RwLock;
-use tracing::{error};
+use tracing::{error, warn};
 
 use tokio::select;
 use crate::asr::{Event, aws::AwsAsr, ASR};
@@ -296,7 +296,7 @@ impl LangLesson {
             select! {
                 res = fut => {
                     if let Err(e) = res {
-                        println!("Error: {:?}", e);
+                        warn!("Error: {:?}", e);
                     }
                 }
                 _ = drop_rx => {}
@@ -411,7 +411,7 @@ impl InnerVoiceLesson {
                 res = fut => match res {
                     Ok(_) => {}
                     Err(e) => {
-                        println!("Error: {:?}", e);
+                        warn!("Error: {:?}", e);
                     }
                 },
                 _ = rx => {}

src/main.rs

@@ -4,9 +4,7 @@
  */
 
 use aws_config::BehaviorVersion;
-use aws_sdk_transcribestreaming::meta::PKG_VERSION;
 use tokio::select;
-use tracing::{debug};
 use tracing_subscriber::{fmt, prelude::*, EnvFilter};
 
 use polyhedron::app;
@@ -19,11 +17,8 @@ async fn main() -> Result<(), std::io::Error> {
         .with(EnvFilter::from_default_env())
         .init();
 
-    debug!("Transcribe client version: {}", PKG_VERSION);
-
     let shared_config = aws_config::load_defaults(BehaviorVersion::latest()).await;
 
-
     select! {
         res = app(&shared_config) => res,
         _ = tokio::signal::ctrl_c() => {

static/index.html

@@ -5,7 +5,7 @@
     <link rel="icon" type="image/svg+xml" href="/vite.svg" />
     <meta name="viewport" content="width=device-width, initial-scale=1.0" />
     <title>Vite + React</title>
-    <script type="module" crossorigin src="/assets/index-1bd6d507.js"></script>
+    <script type="module" crossorigin src="/assets/index-f4f6a4b0.js"></script>
     <link rel="stylesheet" href="/assets/index-983f9492.css">
   </head>
   <body>

tests/stream_single.rs

@@ -19,7 +19,7 @@ use tracing::{info, error, debug};
 use polyhedron::{
     stream_single,
     SingleEvent,
-    asr::slice_i16_to_u8,
+    asr::slice_i16_to_u8_le,
     Context
 };
 
@@ -51,6 +51,7 @@ async fn test_single() {
     let url = format!(
         "ws://{}/ws/voice?id=123abc&from=zh-CN&to=en-US&voice=Amy", addr
     );
+    // let url = "ws://localhost:8080/ws/voice?id=123abc&from=zh-CN&to=en-US&voice=Amy".to_string();
     let (mut client_stream, _) = connect_async(url)
         .await
         .unwrap();
@@ -75,8 +76,8 @@ async fn test_single() {
 
     let audio_stream = stream! {
         for chunk in chunks {
-            yield slice_i16_to_u8(&chunk);
-            sleep(Duration::from_millis(10)).await;
+            yield slice_i16_to_u8_le(&chunk);
+            sleep(Duration::from_millis(100)).await;
         }
     };
     pin!(audio_stream);
@@ -106,7 +107,7 @@ async fn test_single() {
                 assert!(false, "Error: {}", e);
             }
         }
-        _ = sleep(Duration::from_secs(10)) => {
+        _ = sleep(Duration::from_secs(15)) => {
             assert!(false, "timeout");
         }
     };