bug:txt2txt response

api/core/controllers/text2text.py

@@ -2,22 +2,42 @@ 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')
+        try:
+            _traductor = Pipe("text2text-generation", model="Helsinki-NLP/opus-mt-en-es")
+            _traduccion = _traductor(texto)[0]
+            _response = _traduccion.get('generated_text')
+        except Exception as e:
+            _response = str(e)
+        finally:
+            return _response
     
     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')
+        try:
+            _traductor = Pipe("text2text-generation", model="Helsinki-NLP/opus-mt-es-en")
+            _traduccion = _traductor(texto)[0]
+            _response = _traduccion.get('generated_text')
+        except Exception as e:
+            _response = str(e)
+        finally:
+            return _response
     
     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')
+        try:
+            _traductor = Pipe("text2text-generation", model="Helsinki-NLP/opus-mt-ar-es")
+            _traduccion = _traductor(texto)[0]
+            _response = _traduccion.get('generated_text')
+        except Exception as e:
+            _response = str(e)
+        finally:
+            return _response
 
 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')
+        try:
+            _abstractor = Pipe("text2text-generation", model="facebook/bart-large-cnn")
+            _resumen = _abstractor(texto)[0]
+            _response = _resumen.get('generated_text')
+        except Exception as e:
+            _response = str(e)
+        finally:
+            return _response

api/main.py

@@ -56,7 +56,7 @@ def get_traduccion(data:dict) -> dict:
         if data:
             __response['original']= data.get('texto')
             __response['traduccion']= __main.traducir(texto=data.get('texto'),
-                                                      idioma="EN/ES")
+                                                      idioma=data.get('idioma'))
         else:
             raise __main.exception(status_code = 401, datail=f"Datos mal formados:\n{data}")
     except Exception as e: