diff --git a/src/main.rs b/src/main.rs index 1666581..8181821 100644 --- a/src/main.rs +++ b/src/main.rs @@ -7,7 +7,7 @@ use chrono::{self, Timelike}; use dotenv; use ipaddress; use itertools::{self, Itertools}; -use lettre; +use lettre::{self, message}; use reqwest; use serde_json::{self, json}; use zip; @@ -170,6 +170,14 @@ fn main() -> anyhow::Result<()> { std::fs::create_dir(format!("./evaluations/{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")).unwrap() { + let mut response_time_file = std::fs::File::create_new(format!("./evaluations/{formatted_day_before}/response_time.csv")).expect("Failed to response_time.csv"); + let header = "NOME;ID_TALK;TEMPO DE RESPOSTA;DATA"; + response_time_file.write_all(header.as_bytes()).expect("Failed to write header to file"); + drop(response_time_file) + } + // Read system prompt let prompt = std::fs::read_to_string("PROMPT.txt").unwrap(); @@ -190,6 +198,163 @@ fn main() -> anyhow::Result<()> { println!("IDS {:?}", talk_ids); + // Calculate the response time in seconds + let response_time = talk_ids + .iter() + .cloned() + .map(|talk_id| { + let talk_id_get_request = client + .get(format!("https://{}/api/talk_histories", PIPERUN_API_URL)) + .bearer_auth(&access_token) + .header("Content-Type", "application/json") + .header("Accept", "application/json") + .query(&[ + ("talk_id", talk_id), + ("type", "a".to_string()), + ("only_view", "1".to_string()), + ]); + + let talk_id_get_result = talk_id_get_request.send(); + + return talk_id_get_result; + }).filter_map_ok(|result| { + let json = result.json::().expect("Failed to deserialize response to JSON").to_owned(); + let talk_histories = &json["talk_histories"]; + let data = &talk_histories["data"]; + + // Filter chats that have very few messages + let talk_lenght = talk_histories.as_array().expect("Wrong message type received from talk histories").len(); + if talk_lenght < MINIMUM_NUMBER_OF_MESSAGES_TO_EVALUATE {return None;} + + // Filter chats that have less that specified ammount of talks with support agent form the last queue transfer + let found = talk_histories.as_array().expect("Wrong message type received from talk histories").into_iter().enumerate().find(|(pos, message_object)|{ + let message = message_object["message"].as_str().expect("Failed to decode message as string"); + let found = message.find("Atendimento transferido para a fila [NovaNet -> Atendimento -> Suporte]"); + found.is_some() + }); + + match found { + None => {return None;}, + Some(pos) => { + let pos_found = pos.0; + if pos_found < MINIMUM_NUMBER_OF_MESSAGES_WITH_AGENT_TO_EVALUATE {return None;} + } + }; + + // Filter chats that pass by multiple agents + let number_of_agents_except_pipebot = talk_histories.as_array().expect("Wrong message type received from talk histories").into_iter() + .map(|message| &message["user"]["name"]) + .filter(|name| name.as_str().unwrap() != "PipeBot".to_string()) + .unique() + .count(); + if number_of_agents_except_pipebot > 1 {return None;} + + // Filter stop notification active + let found_stop_notification = talk_histories.as_array().expect("Wrong message type received from talk histories").into_iter().enumerate().find(|(pos, message_object)|{ + let message = message_object["message"].as_str().expect("Failed to decode message as string"); + let found = message.find("ATENÇÃO CLIENTE COM NOTIFICAÇÃO DE PARADA ATIVA"); + found.is_some() + }); + + if found_stop_notification.is_some() {return None;} + + // Filter lack of dialogue + let lack_of_dialogue = talk_histories.as_array().expect("Wrong message type received from talk histories").into_iter().find(|message_object|{ + let message = message_object["message"].as_str().expect("Failed to decode message as string"); + let found = message.to_uppercase().find(&"Esse atendimento está sendo encerrado devido a inexistência de diálogo".to_uppercase()); + found.is_some() + }); + + if lack_of_dialogue.is_some() {return None;} + + // Filter is portal access + let portal_link_offer = talk_histories.as_array().expect("Wrong message type received from talk histories").into_iter().find(|message_object|{ + let message = message_object["message"].as_str().expect("Failed to decode message as string"); + let found = message.to_uppercase().find(&"https://novanet.portal.7az.com.br/login".to_uppercase()); + found.is_some() + }); + + if portal_link_offer.is_some() {return None;} + + // Filter no comercial agent available + let comercial_agents_unavailable = talk_histories.as_array().expect("Wrong message type received from talk histories").into_iter().find(|message_object|{ + let message = message_object["message"].as_str().expect("Failed to decode message as string"); + let found = message.to_uppercase().find(&"Transbordo automático para a fila [NovaNet -> Atendimento -> Suporte], pois não havia agentes disponíveis na fila [NovaNet -> Atendimento -> Comercial]".to_uppercase()); + found.is_some() + }); + + if comercial_agents_unavailable.is_some() {return None;} + + // Filter Bot finished chats + if json["agent"]["user"]["name"].as_str().unwrap_or("unknown_user") == "PipeBot" {return None;} + + return Some(json); + }) + // .clone() + .map(|messages| { + let json = messages.unwrap(); + let talk_histories = &json["talk_histories"]; + + // dbg!(&talk_histories); + + // talk_histories.as_array().unwrap().into_iter().enumerate().for_each(|(pos, message_obj)|{println!("{}: {}", pos, message_obj["message"])}); + + // find the bot transfer message + let bot_transfer_message = talk_histories + .as_array().expect("Wrong message type received from talk histories").into_iter() + .enumerate() + .filter(|(pos, message_object)| { + let user_name = message_object["user"]["name"].as_str().expect("Failed to decode message as string"); + user_name == "PipeBot".to_string() + }).find(|(pos, message_object)|{ + let message = message_object["message"].as_str().expect("Failed to decode message as string"); + // let found = message.find("Atendimento transferido para a fila [NovaNet -> Atendimento -> Suporte]"); + let found = message.find("Atendimento entregue da fila de espera para o agente"); + found.is_some() + }); + + // Find first agent message sent after the last bot message + let (pos, transfer_message) = bot_transfer_message.expect("Failed to get the transfer bot message position"); + + let msg = talk_histories.as_array().expect("Wrong message type received from talk histories").into_iter() + .take(pos) + .rev() + .filter(|message| { + message["type"] == "out".to_string() && message["user"]["name"] != "PipeBot".to_string() + }) + .take(1).collect_vec(); + + let agent_first_message = msg[0]; + + // Calculate time difference between bot message and agent message + let date_user_message_sent = agent_first_message["sent_at"].as_str().unwrap(); + + let format = "%Y-%m-%d %H:%M:%S"; + + let date_user_message_sent_parsed = match chrono::NaiveDateTime::parse_from_str(date_user_message_sent, format) { + Ok(dt) => dt, + Err(e) => {println!("Error parsing DateTime: {}", e); panic!("Failed parsing date")}, + }; + + let date_transfer_message_sent_parsed = match chrono::NaiveDateTime::parse_from_str(transfer_message["sent_at"].as_str().unwrap(), format) { + Ok(dt) => dt, + Err(e) => {println!("Error parsing DateTime: {}", e); panic!("Failed parsing date")}, + }; + + let response_time = (date_user_message_sent_parsed - date_transfer_message_sent_parsed).as_seconds_f32(); + let name = agent_first_message["user"]["name"].as_str().unwrap().to_owned(); + let id = json["tracking_number"].as_str().unwrap_or("").to_owned(); + let date = date_user_message_sent.to_owned(); + println!("response_time: {}s", (date_user_message_sent_parsed - date_transfer_message_sent_parsed).as_seconds_f32()); + + format!("{};{};{};{}", name, id, response_time, date) + }).reduce(|acc, e|{format!("{}\n{}",acc,e)}) + .unwrap_or("".to_string()); + + // Open file and append to it + let mut response_time_file = std::fs::OpenOptions::new().write(true).open(format!("./evaluations/{formatted_day_before}/response_time.csv")).expect("Failed to open response time file for write"); + response_time_file.write_all(response_time.as_bytes()).expect("Failed to write to file!"); + talk_ids .iter() .cloned()