feat: conversation saved in file

5 files changed, 34 insertions, 14 deletions

diff --git a/.gitignore b/.gitignore
index 13d3555..2a35603 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,2 +1,4 @@
 /target
+log.txt
 debug.txt
+conv/*
diff --git a/Cargo.toml b/Cargo.toml
index 2069fb2..6d0adf7 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -4,9 +4,11 @@ version = "0.1.0"
 edition = "2021"
 
 [dependencies]
+chrono = "0.4.40"
 color-eyre = "0.6.3"
 ratatui = "0.29.0"
 reqwest = { version = "0.12.12", features = ["blocking", "json"] }
 serde = { version = "1.0.217", features = ["derive"] }
 serde_json = "1.0.138"
 tokio = "1.43.0"
+uuid = { version = "1.15.1", features = ["v4"] } 
diff --git a/src/app/init.rs b/src/app/init.rs
index aa74ae5..f62b2d0 100644
--- a/src/app/init.rs
+++ b/src/app/init.rs
@@ -1,9 +1,11 @@
 use crate::app::llm::{Message, MessageType, LLM};
-use crate::helper::init::print_in_file;
+use crate::helper::init::warn;
+use uuid::Uuid;
 use tokio;
 
 pub struct App {
     pub messages: Vec<Message>, // History of recorded message
+    conv_id: Uuid,
     chat_llm: LLM,
     resume_llm: LLM,
 }
@@ -16,6 +18,7 @@ impl App {
                 MessageType::SYSTEM,
                 chat_llm.system_prompt.clone(),
             )],
+            conv_id: Uuid::new_v4(),
             chat_llm,
             resume_llm: LLM::new("config/resume-LLM.json".to_string()).unwrap(),
         }
@@ -23,6 +26,7 @@ impl App {
 
     fn append_message(&mut self, msg: String, role: MessageType) {
         let message = Message::new(role, msg);
+        message.save_message(self.conv_id.to_string());
         self.messages.push(message);
     }
 
diff --git a/src/app/llm.rs b/src/app/llm.rs
index 8603395..59e045b 100644
--- a/src/app/llm.rs
+++ b/src/app/llm.rs
@@ -1,9 +1,10 @@
-use crate::helper::init::print_in_file;
+use crate::helper::init::warn;
 use reqwest::{header::CONTENT_TYPE, Client};
 use serde::{Deserialize, Serialize};
 use serde_json::Value;
 use std::fmt;
-use std::fs;
+use std::fs::{OpenOptions, create_dir_all};
+use std::io::Write;
 
 #[derive(Deserialize, Debug)]
 pub struct LLM {
@@ -40,7 +41,7 @@ impl LLM {
                 while let Some(chunk) = res.chunk().await? {
                     let answer: Value = serde_json::from_slice(chunk.as_ref())?;
 
-                    print_in_file(answer.to_string());
+                    warn(answer.to_string());
                     if answer["done"].as_bool().unwrap_or(false) {
                         break;
                     }
@@ -53,7 +54,7 @@ impl LLM {
             Err(e) => return Err(Box::new(e)),
         }
     
-        print_in_file(full_message.clone());
+        warn(full_message.clone());
         Ok(full_message)
     }
 }
@@ -85,6 +86,21 @@ impl Message {
     pub fn new(role: MessageType, content: String) -> Message {
         Message { role, content }
     }
+
+    pub fn save_message(&self, conv_id: String) {
+        // Create conv directory if doesn't exist
+        create_dir_all("conv").unwrap();
+
+        // Save message
+        let mut file = OpenOptions::new()
+            .write(true)
+            .append(true)
+            .create(true)
+            .open("conv/".to_string() + &conv_id)
+            .unwrap();
+
+        writeln!(file, "{}", serde_json::to_string(self).unwrap()).unwrap();
+    }
 }
 
 impl fmt::Display for Message {
diff --git a/src/helper/init.rs b/src/helper/init.rs
index 084caf5..abee146 100644
--- a/src/helper/init.rs
+++ b/src/helper/init.rs
@@ -1,18 +1,14 @@
+use chrono::prelude::*;
 use std::fs::OpenOptions;
 use std::io::{self, Write};
 
-pub fn print_in_file(content: String) -> io::Result<()> {
-    // Open the file (create it if it doesn't exist, or truncate it if it does)
+pub fn warn(content: String) {
     let mut file = OpenOptions::new()
         .write(true)
         .append(true)
         .create(true)
-        .open("debug.txt")
+        .open("log.txt")
         .unwrap();
-
-    if let Err(e) = writeln!(file, "{}", content) {
-        eprintln!("Couldn't write to file: {}", e);
-    }
-
-    Ok(())
+    let utc: DateTime<Local> = Local::now();
+    writeln!(file, "[{}] {}", utc, content);
 }