From 3c01a76405c2317303dadf0da830d7b5fde7470e Mon Sep 17 00:00:00 2001
From: heimoshuiyu <heimoshuiyu@gmail.com>
Date: Thu, 21 Nov 2024 22:44:27 +0800
Subject: [PATCH] Convert Traditional Chinese to Simplified Chinese

---
 whisper_fastapi.py | 11 ++++-------
 1 file changed, 4 insertions(+), 7 deletions(-)

diff --git a/whisper_fastapi.py b/whisper_fastapi.py
index 3cdc336..21311fe 100644
--- a/whisper_fastapi.py
+++ b/whisper_fastapi.py
@@ -141,13 +141,10 @@ def stream_builder(
         % (info.language, info.language_probability)
     )
     def wrap():
-        last_pos = 0
-        with tqdm.tqdm(total=info.duration, unit="seconds", disable=True) as pbar:
-            for segment in segments:
-                start, end, text = segment.start, segment.end, segment.text
-                pbar.update(end - last_pos)
-                last_pos = end
-                yield segment
+        for segment in segments:
+            if info.language == "zh":
+                segment.text = ccc.convert(segment.text)
+            yield segment
 
     return wrap(), info