remove the first empty line in the chat

pythops · Jan 25, 2024 · cbf6650 · cbf6650
1 parent 3db7d93
commit cbf6650
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 6 deletions.
diff --git a/src/chat.rs b/src/chat.rs
@@ -64,7 +64,6 @@ impl Chat<'_> {
 
                 self.answer.formatted_answer =
                     formatter.format(format!("🤖: {}", &self.answer.plain_answer).as_str());
-                // self.move_to_bottom();
             }
 
             LLMAnswer::EndAnswer => {

diff --git a/src/handler.rs b/src/handler.rs
@@ -12,7 +12,7 @@ use crossterm::event::{KeyCode, KeyEvent, KeyModifiers};
 use std::sync::mpsc::Sender;
 use std::{collections::HashMap, thread};
 
-use ratatui::text::Line;
+use ratatui::text::{Line, Text};
 
 use crate::notification::{Notification, NotificationLevel};
 use std::sync::Arc;
@@ -237,10 +237,16 @@ pub fn handle_key_events(
 
                 app.chat.plain_chat.push(format!("👤 : {}\n", user_input));
 
-                app.chat.formatted_chat.extend(
-                    app.formatter
-                        .format(format!("👤: {}\n", user_input).as_str()),
-                );
+                if app.chat.formatted_chat.width() == 0 {
+                    app.chat.formatted_chat = app
+                        .formatter
+                        .format(format!("👤: {}\n", user_input).as_str());
+                } else {
+                    app.chat.formatted_chat.extend(
+                        app.formatter
+                            .format(format!("👤: {}\n", user_input).as_str()),
+                    );
+                }
 
                 let conv = HashMap::from([
                     ("role".into(), "user".into()),