fix: 优化控制台闪烁的问题

Author: 潕量

cli_demo.py

@@ -39,20 +39,12 @@ def main():
             os.system(clear_command)
             print("欢迎使用 ChatGLM-6B 模型，输入内容即可进行对话，clear 清空对话历史，stop 终止程序")
             continue
-        count = 0
+        prev_response = ""
+        print("ChatGLM-6B：", end="", flush=True)
         for response, history in model.stream_chat(tokenizer, query, history=history):
-            if stop_stream:
-                stop_stream = False
-                break
-            else:
-                count += 1
-                if count % 8 == 0:
-                    os.system(clear_command)
-                    print(build_prompt(history), flush=True)
-                    signal.signal(signal.SIGINT, signal_handler)
-        os.system(clear_command)
-        print(build_prompt(history), flush=True)
-
+            print(response[len(prev_response):], end="", flush=True)
+            prev_response = response
+        print("\n", end="", flush=True)
 
 if __name__ == "__main__":
     main()