jsonify response & increase no. of segments to send

This commit is contained in:
makaveli10
2023-05-26 12:39:16 +08:00
parent 0a3fa7081e
commit c8d292dbab
+4 -4
View File
@@ -2,7 +2,7 @@
import websockets import websockets
import pickle, struct, time, pyaudio import pickle, struct, time, pyaudio
import threading import threading
import os import os, json
import wave import wave
import textwrap import textwrap
@@ -63,7 +63,7 @@ class ServeClient:
self.show_prev_out_thresh = 5 # if pause(no output from whisper) show previous output for 5 seconds self.show_prev_out_thresh = 5 # if pause(no output from whisper) show previous output for 5 seconds
self.add_pause_thresh = 3 # add a blank to segment list as a pause(no speech) for 3 seconds self.add_pause_thresh = 3 # add a blank to segment list as a pause(no speech) for 3 seconds
self.transcript = [] self.transcript = []
self.send_last_n_segments = 2 self.send_last_n_segments = 10
# text formatting # text formatting
self.wrapper = textwrap.TextWrapper(width=50) self.wrapper = textwrap.TextWrapper(width=50)
@@ -155,7 +155,7 @@ class ServeClient:
'segments': segments 'segments': segments
} }
try: try:
self.websocket.send(str(out_dict)) self.websocket.send(json.dumps(out_dict))
except Exception as e: except Exception as e:
logging.info(f"[ERROR]: {e}") logging.info(f"[ERROR]: {e}")
else: else:
@@ -180,7 +180,7 @@ class ServeClient:
} }
try: try:
self.websocket.send(str(out_dict)) self.websocket.send(json.dumps(out_dict))
except Exception as e: except Exception as e:
logging.info(f"[INFO]: {e}") logging.info(f"[INFO]: {e}")
except Exception as e: except Exception as e: