游雁
2023-10-19 b9bcf1f093c3053fdc4e2cf4a1d38e27bbf429fb
funasr/runtime/java/FunasrWsClient.java
@@ -79,6 +79,10 @@
      obj.put("chunk_size", array);
      obj.put("chunk_interval", new Integer(chunkInterval));
      obj.put("wav_name", wavName);
     if(FunasrWsClient.hotwords.trim().length()>0)
     {
        obj.put("hotwords", FunasrWsClient.hotwords.trim());
     }
     if(suffix.equals("wav")){
         suffix="pcm";
     }
@@ -185,6 +189,10 @@
    try {
      jsonObject = (JSONObject) jsonParser.parse(message);
      logger.info("text: " + jsonObject.get("text"));
     if(jsonObject.containsKey("timestamp"))
     {
        logger.info("timestamp: " + jsonObject.get("timestamp"));
     }
    } catch (org.json.simple.parser.ParseException e) {
      e.printStackTrace();
    }
@@ -222,6 +230,7 @@
  static String strChunkSize = "5,10,5";
  static int chunkInterval = 10;
  static int sendChunkSize = 1920;
  static String hotwords="";
  String wavName = "javatest";
@@ -270,6 +279,12 @@
        .setDefault("offline")
        .type(String.class)
        .required(false);
    parser
        .addArgument("--hotwords")
        .help("hotwords, splited by space")
        .setDefault("")
        .type(String.class)
        .required(false);
    String srvIp = "";
    String srvPort = "";
    String wavPath = "";
@@ -277,7 +292,7 @@
    String chunk_size = "";
    int chunk_interval = 10;
    String strmode = "offline";
    String hot="";
    try {
      Namespace ns = parser.parseArgs(args);
      srvIp = ns.get("host");
@@ -287,6 +302,7 @@
      chunk_size = ns.get("chunk_size");
      chunk_interval = ns.get("chunk_interval");
      strmode = ns.get("mode");
     hot=ns.get("hotwords");
      System.out.println(srvPort);
    } catch (ArgumentParserException ex) {
@@ -298,6 +314,7 @@
    FunasrWsClient.chunkInterval = chunk_interval;
    FunasrWsClient.wavPath = wavPath;
    FunasrWsClient.mode = strmode;
   FunasrWsClient.hotwords=hot;
    System.out.println(
        "serIp="
            + srvIp