From d20c030e5b75306dd67e8fe9924d5d94eac1bf30 Mon Sep 17 00:00:00 2001
From: wusong <63332221+wusong1128@users.noreply.github.com>
Date: 星期三, 25 九月 2024 15:11:50 +0800
Subject: [PATCH] 解决python ws服务针对尾部非人声录音无结束标识返回的问题 (#2102)

---
 funasr/utils/timestamp_tools.py |    6 ++++--
 1 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/funasr/utils/timestamp_tools.py b/funasr/utils/timestamp_tools.py
index 84a02b2..369dc1b 100644
--- a/funasr/utils/timestamp_tools.py
+++ b/funasr/utils/timestamp_tools.py
@@ -141,6 +141,8 @@
     )
     for punc_stamp_text in punc_stamp_text_list:
         punc_id, timestamp, text = punc_stamp_text
+        if sentence_start is None and timestamp is not None:
+            sentence_start = timestamp[0]
         # sentence_text += text if text is not None else ''
         if text is not None:
             if "a" <= text[0] <= "z" or "A" <= text[0] <= "Z":
@@ -156,7 +158,6 @@
 
         punc_id = int(punc_id) if punc_id is not None else 1
         sentence_end = timestamp[1] if timestamp is not None else sentence_end
-        sentence_start = timestamp[0] if timestamp is not None else sentence_start
         sentence_text_seg = (
             sentence_text_seg[:-1] if sentence_text_seg[-1] == " " else sentence_text_seg
         )
@@ -184,6 +185,7 @@
             sentence_text = ""
             sentence_text_seg = ""
             ts_list = []
+            sentence_start = None
     return res
 
 
@@ -272,4 +274,4 @@
             sentence_text = ""
             sentence_text_seg = ""
             ts_list = []
-    return res
\ No newline at end of file
+    return res

--
Gitblit v1.9.1