Fix improve_formatting

Author: pierotofy

README.md

@@ -143,7 +143,6 @@ You can use the LTEngine API using the following bindings:
  - [ ] Better language detection for short texts (port [LexiLang](https://github.com/LibreTranslate/LexiLang) to Rust)
  - [ ] Test/add more LLM models aside from Gemma3
  - [ ] Create comparative benchmarks between LTEngine and proprietary software.
- - [ ] Normalize punctuation, uppercase/lowercase
  - [ ] Add support for command line inference (run `./ltengine translate` as a command line app separate from `./ltengine server`)
  - [ ] Make ltengine available as a library, possibly creating bindings for other languages like Python.
  - [ ] Automated builds / CI

ltengine/src/main.rs

@@ -142,17 +142,19 @@ fn check_params(body: &TranslateRequest, args: &Args, required_params: &[(&str,
 }
 
 fn improve_formatting(q: &String, translation: &String) -> String {
+    let t = translation.trim().to_string();
+
     if q.len() == 0 {
         return String::new();
     }
 
-    if translation.len() == 0 {
-        return q.clone()
+    if t.len() == 0 {
+        return q.clone();
     }
 
     let q_last_char = q.chars().rev().next().unwrap();
-    let translation_last_char = translation.chars().rev().next().unwrap();
-    let mut result = translation.clone();
+    let translation_last_char = t.chars().rev().next().unwrap();
+    let mut result = t.clone();
 
     const PUNCTUATION_CHARS: [char; 6] = ['!', '?', '.', ',', ';', '。'];
     if PUNCTUATION_CHARS.contains(&q_last_char){
@@ -177,9 +179,9 @@ fn improve_formatting(q: &String, translation: &String) -> String {
 
     if let (Some(q0), Some(r0)) = (q.chars().next(), result.chars().next()) {
         if q0.is_lowercase() && r0.is_uppercase() {
-            result.replace_range(0..r0.len_utf8(), &r0.to_uppercase().to_string());
-        }else if q0.is_uppercase() && r0.is_lowercase() {
             result.replace_range(0..r0.len_utf8(), &r0.to_lowercase().to_string());
+        }else if q0.is_uppercase() && r0.is_lowercase() {
+            result.replace_range(0..r0.len_utf8(), &r0.to_uppercase().to_string());
         }
     }