from transformers import pipeline as Pipe

class Traductor:
    def EN_ES(texto:str)->str:
        _traductor = Pipe("text2text-generation", model="Helsinki-NLP/opus-mt-en-es")
        _traduccion = _traductor(texto)[0]
        return _traduccion.get('generated_text')
    
    def ES_EN(texto:str)->str:
        _traductor = Pipe("text2text-generation", model="Helsinki-NLP/opus-mt-es-en")
        _traduccion = _traductor(texto)[0]
        return _traduccion.get('generated_text')
    
    def AR_ES(texto:str)->str:
        _traductor = Pipe("text2text-generation", model="Helsinki-NLP/opus-mt-ar-es")
        _traduccion = _traductor(texto)[0]
        return _traduccion.get('generated_text')

class Abstractor:
    def resumen(texto:str)->str:
        _abstractor = Pipe("text2text-generation", model="facebook/bart-large-cnn")
        _resumen = _abstractor(texto)[0]
        return _resumen.get('generated_text')