secure the api calling for the phonemizer service

This commit is contained in:
matthieu 2025-01-30 19:23:03 +04:00
parent 3ba26023d0
commit f14f90c141
3 changed files with 12 additions and 12 deletions

View file

@ -4,7 +4,6 @@ use autofeur::save::Save;
use kdam::tqdm;
#[tokio::main]
/// Generates the DB file foe easy usage.
async fn main() {
let mut save = Save::default();

View file

@ -1,4 +1,7 @@
use std::{env, ops::Add};
use std::{
env::{self, VarError},
ops::Add,
};
use anyhow::anyhow;
use hypher::hyphenate;
@ -7,14 +10,13 @@ use itertools::Itertools;
use crate::save::Save;
async fn call_inference_service(word: &str) -> anyhow::Result<String> {
let server: Result<String, anyhow::Error> =
env::var("PHONEMIZER").or_else(|_| Ok("http://localhost:8000/".to_string()));
Ok(
reqwest::get(format!("{}?grapheme={}", server.unwrap(), word))
.await?
.text()
.await?,
)
let server: String = env::var("PHONEMIZER")
.or_else(|_| Ok::<String, VarError>("http://localhost:8000/".to_string()))
.unwrap();
let url = reqwest::Url::parse_with_params(&server, &[("grapheme", word)])?;
Ok(reqwest::get(url).await?.text().await?)
}
impl Save<'_> {
@ -77,7 +79,7 @@ impl Save<'_> {
}
// we finally just need to compute the end of the word which matches the sound
let found = completed_syllabes.drain(i+1..).join("");
let found = completed_syllabes.drain(i + 1..).join("");
println!("{} is equivalent to {}", completion, found);
Ok(format!("{} ({})", found, word))

View file

@ -13,7 +13,6 @@ pub struct TrieNode<'a> {
}
impl<'a> TrieNode<'a> {
// Create new node
pub fn new<'b>(is_final: bool) -> TrieNode<'b> {
TrieNode {
is_final,