From 2a20f48efad692a8c2744f10c673bbdbe0c751b7 Mon Sep 17 00:00:00 2001 From: Val Kharitonov Date: Mon, 13 Mar 2023 12:24:18 -0400 Subject: Fix UTF-8 handling (including colors) (#79) --- main.cpp | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) (limited to 'main.cpp') diff --git a/main.cpp b/main.cpp index 98ccde5..d068761 100644 --- a/main.cpp +++ b/main.cpp @@ -939,6 +939,11 @@ int main(int argc, char ** argv) { break; } } + + // reset color to default if we there is no pending user input + if (!input_noecho && params.use_color && embd_inp.size() == input_consumed) { + printf(ANSI_COLOR_RESET); + } } // display text @@ -946,10 +951,6 @@ int main(int argc, char ** argv) { for (auto id : embd) { printf("%s", vocab.id_to_token[id].c_str()); } - // reset color to default if we there is no pending user input - if (params.use_color && embd_inp.size() <= input_consumed) { - printf(ANSI_COLOR_RESET); - } fflush(stdout); } -- cgit v1.2.3