1、音频转文本

application.yml新增配置

      # 语音转文本
      audio:
        transcription:
          options:
            # 模型版本
            model: whisper-1

完整配置如下：

spring:
  application:
    name: spring-ai
  ai:
    openai:
      # 访问open ai接口的api key
      api-key: sk-3sfER03LDLG3SDFsdlwe283JSdw023lkrmrHDND32fmREKFD
      # 访问open ai的接口地址
      base-url: https://openai.com/

      # ai聊天设置
      chat:
        options:
          # chatGpt模型版本，32k是参数量，若当前代码中和application配置文件中同时声明，则代码中的配置会覆盖application配置文件中的
          model: gpt-4-32k
          # 温度越高，回答的准确率会下降，温度越低，回答的准确率越好,若当前代码中和application配置文件中同时声明，则代码中的配置会覆盖application配置文件中的
          temperature: 0.3F

      # ai绘图设置
      image:
        options:
          # 模型版本
          model: gpt-4-dalle
          # 图片质量
          quality: hd
          # 数量
          n: 2
          # 高度
          height: 1920
          # 宽度
          width: 1080

      # 语音转文本
      audio:
        transcription:
          options:
            # 模型版本
            model: whisper-1

View Code

模板模型版本可以debug查询，或者看文档

示例代码如下：

/**
 * 音频转文本
 * @author ithailin
 */
@RestController
@RequestMapping("/ai")
public class TranscriptionController {
    private static final Logger logger = LoggerFactory.getLogger(TranscriptionController.class);

    @Autowired
    private OpenAiAudioTranscriptionModel openAiAudioTranscriptionModel;

    /**
     * 音频转文本
     * public String call(Resource audioResource){}
     * @return
     */
    @RequestMapping("/transcription")
    public Object transcription(){
        Resource audioFile = new ClassPathResource("test.mp3");
        String call = openAiAudioTranscriptionModel.call(audioFile);
        logger.info("call:{}",call);
        return call;
    }
}

View Code

2、文本转音频

/**
 * 文本转音频
 * @author ithailin
 */
@RestController
@RequestMapping("/AI")
public class TTSController {
    private static final Logger logger = LoggerFactory.getLogger(TTSController.class);
    @Autowired
    private OpenAiAudioSpeechModel openAiAudioSpeechModel;

    /**
     * 音频转文本
     * public String call(Resource audioResource){}
     * @return
     */
    @RequestMapping("/tts")
    public Object tts(String msg){
        logger.info("msg：{}",msg );
        //msg若为空，转换会异常，msg中文、英文都支持
        byte[] call = openAiAudioSpeechModel.call(msg);
        FileUtils.save2File("D:\\SpringAI\\test.mp3",call);
        return "OK";
    }

}

View Code

FileUtils.java

public class FileUtils {

    /**
     * 方法功能：将字节数组写入到新建文件中。
     * @return boolean
     * */
    public static boolean save2File(String fname, byte[] msg){
        OutputStream fos = null;
        try{
            File file = new File(fname);
            File parent = file.getParentFile();
            boolean bool;
            if ((!parent.exists()) &&
                    (!parent.mkdirs())) {
                return false;
            }
            fos = new FileOutputStream(file);
            fos.write(msg);
            fos.flush();
            return true;
        }catch (FileNotFoundException e){
            return false;
        }catch (IOException e){
            e.printStackTrace();
            return false;
        }
        finally{
            if (fos != null) {
                try{
                    fos.close();
                } catch (IOException e) {}
            }
        }
    }

}