fix: 修复相应bug

src/renderer/plugins/live/PhotoRole.ts

@@ -351,7 +351,7 @@ export class PhotoRole extends EventEmitter {
     return new Promise<boolean>((resolve) => {
       let count = 0
       const keepCalling = async () => {
-        if (count >= 10) return resolve(false)
+        if (count >= 30) return resolve(false)
         count++
 
         const { status } = await this._getLiveStatus(taskId)

src/renderer/screens/ShowPhoto.vue

@@ -368,6 +368,7 @@ function initLLMSocket(): Promise<WebSocket> {
 
 async function endAudioInput() {
   microphoneState.value = 'waitInput'
+  inputContext.ws?.close()
   inputContext.mediaStream?.getTracks().forEach((track) => track.stop())
   inputContext.audioContext?.close()
   inputContext.audioContext2?.close()
@@ -443,8 +444,10 @@ async function onQ(question: string) {
     photoRole!.on('asyncAnswer', onAsyncAnswer)
 
     ws.onmessage = (message) => {
+      clearTimeout(messageTimeout)
+
       try {
-        const { text, event } = JSON.parse(message.data) as {
+        let { text, event } = JSON.parse(message.data) as {
           event: string
           message_num: number
           text: string
@@ -460,12 +463,12 @@ async function onQ(question: string) {
           return
         }
 
+        text = text.replace(/\u0000/g, '').trim()
         answer += text
         photoRole!.answerArgs!.answer += answer
         photoRole!.answerArgs!._typingAnswer.push(answer)
         isTime && console.time('sliceAnswer')
         isTime = false
-        clearTimeout(messageTimeout)
 
         const textArr = text.split('')
         for (let i = 0; i < textArr.length; i++) {

src/renderer/screens/ShowVideo.vue

@@ -263,6 +263,7 @@ function initLLMSocket(): Promise<WebSocket> {
 
 function endAudioInput() {
   microphoneState.value = 'waitInput'
+  inputContext.ws?.close()
   inputContext.mediaStream?.getTracks().forEach((track) => track.stop())
   inputContext.audioContext?.close()
   inputContext.audioContext2?.close()
@@ -360,12 +361,14 @@ async function onQ(question: string) {
     let sliceAnswerLength = 10
     inputContext.ws = ws
     ws.onmessage = (message) => {
+      clearTimeout(messageTimeout)
+
       if (microphoneState.value === 'input') {
         return
       }
 
       try {
-        const { text, event } = JSON.parse(message.data) as {
+        let { text, event } = JSON.parse(message.data) as {
           event: string
           message_num: number
           text: string
@@ -375,18 +378,15 @@ async function onQ(question: string) {
           inputContext.answerArray.push(sliceAnswer)
           runTTSTask()
           sliceAnswer = ''
-
-          inputContext.answerArray.push(sliceAnswer)
-          sliceAnswer = ''
           inputContext.ws?.close()
           console.log('----------------> answer: ', answer)
           return
         }
 
+        text = text.replace(/\u0000/g, '').trim()
         answer += text
         isTime && console.time('sliceAnswer')
         isTime = false
-        clearTimeout(messageTimeout)
 
         const textArr = text.split('')
         for (let i = 0; i < textArr.length; i++) {