DubAI / src /models /translation.py
pokung's picture
Upload 18 files
e848a0c verified
Raw
History Blame Contribute Delete
795 Bytes
import deepl
import os
from typing import Optional
from dotenv import load_dotenv
load_dotenv()
class TranslationService:
def __init__(self):
self.api_key = os.getenv("DEEPL_API_KEY")
if not self.api_key:
raise ValueError("Zmienna środowiskowa DEEPL_API_KEY nie jest ustawiona!")
self.translator = deepl.Translator(self.api_key)
def translate(
self,
text: str,
source_lang: str = "PL",
target_lang: str = "EN-US"
) -> str:
"""Tłumaczenie tekstu"""
result = self.translator.translate_text(
text,
source_lang=source_lang,
target_lang=target_lang
)
return result.text.replace(".", ",")
# Singleton pattern
TRANSLATOR = TranslationService()