update config of save data to save flag

config.py

@@ -3,17 +3,16 @@ import re
 import logging
 
 DEBUG = True
-TEST = False
-logging.getLogger("pywhispercpp").setLevel(logging.WARNING)
-
 
+logging.getLogger("pywhispercpp").setLevel(logging.WARNING)
 logging.basicConfig(
     level=logging.DEBUG if DEBUG else logging.INFO,
     format="%(asctime)s - %(levelname)s - %(message)s",
     filename='translator.log',
     datefmt="%H:%M:%S"
 )
-
+# save pipelines data to disk
+SAVE_DATA_SAVE = False
 # Add terminal log
 console_handler = logging.StreamHandler()
 console_handler.setLevel(logging.DEBUG if DEBUG else logging.INFO)

transcribe/whisper_llm_serve.py

@@ -66,17 +66,18 @@ class WhisperTranscriptionService:
         # for test
         self._transcrible_time_cost = 0.
         self._translate_time_cost = 0.
-        if config.TEST:
-            self._test_task_stop = threading.Event()
-            self._test_queue = queue.Queue()
-            self._test_thread = self._start_thread(self.test_data_loop)
+
+        if config.SAVE_DATA_SAVE:
+            self._save_task_stop = threading.Event()
+            self._save_queue = queue.Queue()
+            self._save_thread = self._start_thread(self.save_data_loop)
 
         # self._c = 0
 
-    def test_data_loop(self):
+    def save_data_loop(self):
         writer = TestDataWriter()
-        while not self._test_task_stop.is_set():
-            test_data = self._test_queue.get()
+        while not self._save_task_stop.is_set():
+            test_data = self._save_queue.get()
             writer.write(test_data)  # Save test_data to CSV
 
 
@@ -247,8 +248,8 @@ class WhisperTranscriptionService:
             )
             current_time = time.perf_counter()
             time_diff = current_time - start_time
-            if config.TEST:
-                self._test_queue.put(DebugResult(
+            if config.SAVE_DATA_SAVE:
+                self._save_queue.put(DebugResult(
                     seg_id=ana_result.seg_id,
                     transcrible_time=self._transcrible_time_cost,
                     translate_time=self._translate_time_cost,
@@ -277,6 +278,6 @@ class WhisperTranscriptionService:
         """停止所有处理线程并清理资源"""
         self._translate_thread_stop.set()
         self._frame_processing_thread_stop.set()
-        if config.TEST:
-            self._test_task_stop.set()
+        if config.SAVE_DATA_SAVE:
+            self._save_task_stop.set()
         logger.info(f"Stopping transcription service for client: {self.client_uid}")