From 9d741f3122a23dde811f9252eb56ffc632fbfaea Mon Sep 17 00:00:00 2001
From: poal <2689766486@qq.com>
Date: Sat, 25 Oct 2025 13:57:04 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8Dtts=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../speech/tts/model/SherpaTtsModel.java      |  6 ++---
 .../smartjavaai/speech/utils/AudioUtils.java  | 24 +++++++++++++++++++
 2 files changed, 27 insertions(+), 3 deletions(-)

diff --git a/speech/src/main/java/cn/smartjavaai/speech/tts/model/SherpaTtsModel.java b/speech/src/main/java/cn/smartjavaai/speech/tts/model/SherpaTtsModel.java
index e7d38e3..51c6169 100644
--- a/speech/src/main/java/cn/smartjavaai/speech/tts/model/SherpaTtsModel.java
+++ b/speech/src/main/java/cn/smartjavaai/speech/tts/model/SherpaTtsModel.java
@@ -81,10 +81,10 @@ public class SherpaTtsModel implements TtsModel{
             }
         }
         try {
-            int sid = 100;
+            int sid = 0;
             float speed = 1.0f;
             if(params != null){
-                sid = sherpaTtsParams.getSpeakerId() > 0 ? sherpaTtsParams.getSpeakerId() : 100;
+                sid = sherpaTtsParams.getSpeakerId();
                 speed = sherpaTtsParams.getSpeed() > 0.0f ? sherpaTtsParams.getSpeed() : 1.0f;
             }
             GeneratedAudio audio = null;
@@ -104,7 +104,7 @@ public class SherpaTtsModel implements TtsModel{
     @Override
     public R<Audio> generate(String text, TtsParams params) {
         GeneratedAudio audio = generateCore(text, params);
-        return R.ok(new Audio(audio.getSamples()));
+        return R.ok(new Audio(audio.getSamples(), audio.getSampleRate(), 1));
     }
 
     @Override
diff --git a/speech/src/main/java/cn/smartjavaai/speech/utils/AudioUtils.java b/speech/src/main/java/cn/smartjavaai/speech/utils/AudioUtils.java
index 5d00d7d..4f154d1 100644
--- a/speech/src/main/java/cn/smartjavaai/speech/utils/AudioUtils.java
+++ b/speech/src/main/java/cn/smartjavaai/speech/utils/AudioUtils.java
@@ -238,6 +238,30 @@ public class AudioUtils {
         }
     }
 
+    /**
+     * 将 float[] 音频数据保存为 WAV 文件
+     */
+    public static void saveToWav(Audio audio, String savePath) throws IOException {
+        AudioFormat audioFormat = new AudioFormat(
+                AudioFormat.Encoding.PCM_SIGNED,
+                audio.getSampleRate(),
+                16,
+                1,
+                2,
+                audio.getSampleRate(),
+                false
+        );
+        // 1. 转换为 16-bit PCM
+        byte[] bytes = floatsToPCM16(audio.getData());
+        // 2. 使用 ByteArrayInputStream 封装为音频流
+        try (ByteArrayInputStream bais = new ByteArrayInputStream(bytes);
+             AudioInputStream ais = new AudioInputStream(bais, audioFormat, audio.getData().length)) {
+            // 3. 保存到本地文件
+            AudioSystem.write(ais, AudioFileFormat.Type.WAVE, new File(savePath));
+        }
+    }
+
+
     public static AudioFormat getDefaultAudioFormat(){
         return new AudioFormat(
                 AudioFormat.Encoding.PCM_SIGNED,
-- 
Gitee