Update Echox_copy_stream.py

Echox_copy_stream.py

@@ -85,96 +85,6 @@ def load_speech_model(device):
 #     vocoder = load_units_vocoder(voc_cfg, device)
 #     return vocoder, voc_cfg
 
-class ScrollingDisplay:
-    def __init__(self, width=71, height=10):
-        self.width = width
-        self.height = height
-        self.lines = [""] * height
-        self.current_line = 0
-        self.current_pos = 0
-        
-    def add_token(self, token):
-        if '\n' in token:
-            parts = token.split('\n')
-            for i, part in enumerate(parts):
-                if i > 0:
-                    self._new_line()
-                if part:
-                    self._add_text_to_current_line(part)
-        else:
-            self._add_text_to_current_line(token)
-    
-    def _add_text_to_current_line(self, text):
-        if not text:
-            return
-        
-        is_continuation = (not text.startswith(' ') and 
-                           self.current_pos > 0 and 
-                           not self.lines[self.current_line].endswith(' '))
-        
-        if is_continuation:
-            current_line_content = self.lines[self.current_line]
-            last_space_pos = current_line_content.rfind(' ')
-            
-            if last_space_pos != -1:
-                last_word_start = last_space_pos + 1
-            else:
-                last_word_start = 0
-            
-            full_word = current_line_content[last_word_start:] + text
-            
-            if self.current_pos + len(text) > self.width:
-                new_content = current_line_content[:last_word_start].rstrip()
-                self.lines[self.current_line] = new_content
-                self.current_pos = len(new_content)
-                self._new_line()
-                text = full_word
-            else:
-                self.lines[self.current_line] += text
-                self.current_pos += len(text)
-                return
-        
-        if self.current_pos > 0 and len(text) > (self.width - self.current_pos) and len(text) <= self.width:
-            self._new_line()
-        
-        while len(text) > 0:
-            if self.current_pos == 0:
-                text = text.lstrip(' ')
-                if not text:
-                    break
-            
-            remaining = self.width - self.current_pos
-            if len(text) > remaining:
-                part = text[:remaining]
-                self.lines[self.current_line] += part
-                self.current_pos += len(part)
-                text = text[remaining:]
-                self._new_line()
-            else:
-                self.lines[self.current_line] += text
-                self.current_pos += len(text)
-                break
-    
-    def _new_line(self):
-        self.current_line += 1
-        self.current_pos = 0
-        
-        if self.current_line >= self.height:
-            self.lines = self.lines[1:] + [""]
-            self.current_line = self.height - 1
-    
-    def display(self):
-        res = ""
-        for i, line in enumerate(self.lines):
-            padded_line = line.ljust(self.width)
-            if i == self.current_line:
-                cursor_pos = min(self.current_pos, self.width - 1)
-                display_line = padded_line[:cursor_pos] + "█" + padded_line[cursor_pos + 1:]
-                res += f"{display_line}\n"
-            else:
-                res += f"{padded_line}\n"
-        return res
-
 class EchoxAssistant():
     def __init__(self):
         class BasicSetting:
@@ -368,8 +278,6 @@ class EchoxAssistant():
         do_sample = kwargs.get('do_sample', True)
         
         current_text = ""
-        display = ScrollingDisplay()
-
         accumulated_hidden_states = []
         accumulated_tokens = []
         similarity_scores = []
@@ -443,8 +351,7 @@ class EchoxAssistant():
                 similarity_scores.append(similarity)
                 
                 token_text = self.tokenizer.decode([next_token.item()], skip_special_tokens=True)
-                display.add_token(token_text)
-                current_text = display.display()
+                current_text += token_text
                 
                 yield current_text, None