feat: remove the first two newlines

1 files changed, 10 insertions, 1 deletions

diff --git a/src/main.rs b/src/main.rs
index eb88226..b7e3f23 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -261,6 +261,7 @@ async fn handle_input(client: &Client, config: &Config, text: &str) -> Result<()
         return Ok(());
     }
     let mut stream = acquire_stream(client, config, text).await?;
+    let mut virgin = true;
     while let Some(part) = stream.next().await {
         let chunk = part?.data;
         if chunk == "[DONE]" {
@@ -272,7 +273,15 @@ async fn handle_input(client: &Client, config: &Config, text: &str) -> Result<()
             let text = data["choices"][0]["delta"]["content"]
                 .as_str()
                 .unwrap_or_default();
-
+            if text.is_empty() {
+                continue;
+            }
+            if virgin {
+                virgin = false;
+                if text == "\n\n" {
+                    continue;
+                }
+            }
             print!("{text}");
             stdout().flush().unwrap();
         }