Update routers/other_route.py

routers/other_route.py

@@ -8,17 +8,17 @@ router = APIRouter()
 @router.get("/translate/")
 def translate_text(
     text: str = Query(..., description="Texto a ser traduzido"),
-    target_language: str = Query("en", description="Idioma alvo para tradução"),
-    source_language: str = Query(None, description="Idioma de origem (opcional, se não fornecido será detectado automaticamente)")
+    target_lang: str = Query("en", description="Idioma alvo para tradução"),
+    source_lang: str = Query(None, description="Idioma de origem (opcional, se não fornecido será detectado automaticamente)")
 ):
     """
     Traduz o texto para o idioma alvo e retorna detalhes, incluindo a detecção de idioma e o tempo de tradução.
     """
     # Verifica se o idioma de destino é válido
-    if target_language not in LANGUAGES:
+    if target_lang not in LANGUAGES:
         raise HTTPException(
             status_code=400,
-            detail=f"Invalid target language: {target_language}. Supported languages: {', '.join(LANGUAGES.keys())}"
+            detail=f"Invalid target language: {target_lang}. Supported languages: {', '.join(LANGUAGES.keys())}"
         )
     
     try:
@@ -27,26 +27,30 @@ def translate_text(
 
         # Se o idioma de origem não for fornecido, será detectado automaticamente
         start_time = time.time()  # Início do tempo de tradução
-        if source_language:
+        if source_lang:
             # Traduz com idioma de origem especificado
-            translated = translator.translate(text, src=source_language, dest=target_language)
+            translated = translator.translate(text, src=source_lang, dest=target_lang)
+            detected_lang = False  # Indica que o idioma foi fornecido
         else:
             # Se não for fornecido, detecta automaticamente o idioma de origem
-            translated = translator.translate(text, dest=target_language)
+            translated = translator.translate(text, dest=target_lang)
+            detected_lang = True  # Indica que o idioma foi detectado
 
         end_time = time.time()  # Fim do tempo de tradução
         translation_time = round(end_time - start_time, 4)  # Tempo em segundos (com 4 casas decimais)
 
         # Retorna a resposta com dados adicionais
         return {
-            "original_text": text,
-            "translated_text": translated.text,
-            "detected_source_language": translated.src,
-            "source_language": source_language or "auto-detected",
-            "target_language": target_language,
-            "translation_time_seconds": translation_time,
-            "original_language_name": LANGUAGES.get(translated.src, "Unknown"),
-            "target_language_name": LANGUAGES.get(target_language, "Unknown")
+            "orig_text": text,
+            "trans_text": translated.text,
+            "detected_lang": detected_lang,
+            "source_lang": translated.src if not source_lang else source_lang,
+            "target_lang": target_lang,
+            "trans_time_s": translation_time,
+            "orig_lang_name": LANGUAGES.get(translated.src, "Unknown"),
+            "target_lang_name": LANGUAGES.get(target_lang, "Unknown"),
+            "orig_word_count": len(text.split()),
+            "trans_word_count": len(translated.text.split())
         }
 
     except Exception as e: