# machine translation it -> de
# see https://www.1006.org/blog/2024-07-24_local_machine_translation_german_italian/

from transformers import MarianMTModel, MarianTokenizer

# model: https://huggingface.co/Helsinki-NLP/opus-mt-it-de
model_name = "Helsinki-NLP/opus-mt-it-de" 

tokenizer = MarianTokenizer.from_pretrained(model_name)
model = MarianMTModel.from_pretrained(model_name)

def translate(text):
    tokenized_text = tokenizer(text, return_tensors="pt", padding=True)
    translated_text = model.generate(**tokenized_text)
    return tokenizer.decode(translated_text[0], skip_special_tokens=True)


# original italian text (a random paragraph from a random website):
# https://www.comune.merano.bz.it/system/web/news.aspx?menuonr=224996926&detailonr=225677044-2418&sprache=3

print(translate(
"""
Nell’ambito di un progetto realizzato dal Planetarium Alto Adige e sostenuto
dal Servizio gestione del verde e ambiente del Comune di Merano, è stato
installato pochi giorni fa, sulla torre del palazzo municipale di via Portici,
un sensore in grado di misurare il livello di inquinamento luminoso sopra il
centro storico della città.
"""
))

# german translation:

# Im Rahmen eines Projekts, das vom Planetarium Südtirol durchgeführt und vom
# Grün- und Umweltdienst der Stadt Meran unterstützt wurde, wurde vor wenigen
# Tagen auf dem Turm des Stadtpalastes der Portici-Straße ein Sensor installiert,
# der den Grad der Lichtverschmutzung über der Altstadt messen kann.