diff --git a/src/main.rs b/src/main.rs index 424a049..e9b0f97 100644 --- a/src/main.rs +++ b/src/main.rs @@ -168,19 +168,19 @@ fn main() -> anyhow::Result<()> { let formatted_day_before = day_before_at_midnight.format("%Y-%m-%d").to_string(); // Create a folder named with the day_before - if !std::fs::exists(format!("./evaluations/{formatted_day_before}")).unwrap() { - std::fs::create_dir(format!("./evaluations/{formatted_day_before}")) + if !std::fs::exists(format!("./evaluations/evaluations_noc-cli/{formatted_day_before}")).unwrap() { + std::fs::create_dir(format!("./evaluations/evaluations_noc-cli/{formatted_day_before}")) .expect("Failed to create directory") } // Create the response time folder if !std::fs::exists(format!( - "./evaluations/{formatted_day_before}/response_time.csv" + "./evaluations/evaluations_noc-cli/{formatted_day_before}/response_time.csv" )) .unwrap() { let mut response_time_file = std::fs::File::create_new(format!( - "./evaluations/{formatted_day_before}/response_time.csv" + "./evaluations/evaluations_noc-cli/{formatted_day_before}/response_time.csv" )) .expect("Failed to response_time.csv"); } @@ -219,15 +219,15 @@ fn main() -> anyhow::Result<()> { println!("Fim do dia: {}", formatted_day_end); // Criar pasta com o nome do dia processado - if !std::fs::exists(format!("./evaluations/{formatted_day}")).unwrap() { - std::fs::create_dir(format!("./evaluations/{formatted_day}")) + if !std::fs::exists(format!("./evaluations/evaluations_noc-cli/{formatted_day}")).unwrap() { + std::fs::create_dir(format!("./evaluations/evaluations_noc-cli/{formatted_day}")) .expect("Failed to create directory"); } // Criar arquivo response_time.csv se não existir - if !std::fs::exists(format!("./evaluations/{formatted_day}/response_time.csv")).unwrap() { + if !std::fs::exists(format!("./evaluations/evaluations_noc-cli/{formatted_day}/response_time.csv")).unwrap() { let _ = std::fs::File::create_new(format!( - "./evaluations/{formatted_day}/response_time.csv" + "./evaluations/evaluations_noc-cli/{formatted_day}/response_time.csv" )) .expect("Failed to create response_time.csv"); } @@ -235,8 +235,8 @@ fn main() -> anyhow::Result<()> { // Read system prompt - let prompt = std::fs::read_to_string("PROMPT.txt").unwrap(); - let filter_file_contents = std::fs::read_to_string("FILTER.txt").unwrap_or(String::new()); + let prompt = std::fs::read_to_string("./prompts/PROMPT_NOC-CLI.txt").unwrap(); + let filter_file_contents = std::fs::read_to_string("./filters/FILTER_NOC-CLI.txt").unwrap_or(String::new()); let filter_keywords = filter_file_contents .split("\n") .filter(|keyword| !keyword.is_empty()) @@ -484,8 +484,8 @@ fn main() -> anyhow::Result<()> { let mut response_time_file = std::fs::OpenOptions::new() .write(true) .open(format!( - "./evaluations/{formatted_day_before}/response_time.csv" - //"./evaluations/{formatted_day}/response_time.csv" + "./evaluations/evaluations_noc-cli/{formatted_day_before}/response_time.csv" + //"./evaluations/evaluations_noc-cli{formatted_day}/response_time.csv" )) .expect("Failed to open response time file for write"); response_time_file @@ -502,6 +502,14 @@ fn main() -> anyhow::Result<()> { .expect("Wrong message type received from talk histories") .iter() .rev() +//filtro para escrever somente as mensagens out no .txt + .filter(|(message_object)| { + let message_type = message_object["type"] + .as_str() + .expect("Failed to decode message as string"); + message_type == "out".to_string() + }) +//fim do filtro .map(|message_object| { let new_json_filtered = format!( "{{ @@ -560,7 +568,7 @@ fn main() -> anyhow::Result<()> { let tracking_number = &json["tracking_number"].as_str().unwrap_or(""); std::fs::write( format!( - "./evaluations/{}/{} - {} - {}.csv", + "./evaluations/evaluations_noc-cli/{}/{} - {} - {}.csv", formatted_day_before, user_name, talk_id, tracking_number //formatted_day, user_name, talk_id, tracking_number ), @@ -569,7 +577,7 @@ fn main() -> anyhow::Result<()> { .expect("Unable to write file"); std::fs::write( format!( - "./evaluations/{}/{} - {} - {} - prompt.txt", + "./evaluations/evaluations_noc-cli/{}/{} - {} - {} - prompt.txt", formatted_day_before, user_name, talk_id, tracking_number //formatted_day, user_name, talk_id, tracking_number ), @@ -584,10 +592,10 @@ fn main() -> anyhow::Result<()> { }); // Compress folder into zip - let source_dir_str = format!("./evaluations/{formatted_day_before}"); - //let source_dir_str = format!("./evaluations/{formatted_day}"); - let output_zip_file_str = format!("./evaluations/{formatted_day_before}.zip"); - //let output_zip_file_str = format!("./evaluations/{formatted_day}.zip"); + let source_dir_str = format!("./evaluations/evaluations_noc-cli/{formatted_day_before}"); + //let source_dir_str = format!("./evaluations/evaluations_noc-cli/{formatted_day}"); + let output_zip_file_str = format!("./evaluations/evaluations_noc-cli/{formatted_day_before}.zip"); + //let output_zip_file_str = format!("./evaluations/evaluations_noc-cli/{formatted_day}.zip"); let source_dir = std::path::Path::new(source_dir_str.as_str()); let output_zip_file = std::path::Path::new(output_zip_file_str.as_str()); zip_directory_util::zip_directory_util::zip_source_dir_to_dst_file(source_dir, output_zip_file);