update main

api/core/controllers/text2image.py

@@ -44,6 +44,7 @@ class Generador:
             _imagen = _generador(prompt).images[0]
             _response = Generador.img_to_bytes(img=_imagen)
         except Exception as e:
+            print(e)
             _response = bytes(str(e), 'utf-8')
         finally:
             return _response

api/main.py

@@ -83,13 +83,13 @@ def get_resumen(data:dict) -> dict:
         return __response
     
 @api.post("/texto_a_imagen/", status_code=201, responses = {201: {"content": {"image/png": {}}}} ,response_class=__main.api_response)
-def get_text2img(data:dict) -> dict:
+def get_text2img(data:dict):
     __response=dict({"request_data":data})
     try:
         if data and 'texto' in data and 'modelo' in data:
             __response['original']= data.get('texto')
-            __response['image']= __main.text_to_img(texto=data.get('texto'), 
-                                                    model=data.get('modelo'))
+            __image =  __main.text_to_img(texto=data.get('texto'), 
+                                          model=data.get('modelo'))
         else:
             raise __main.exception(status_code = 401, datail=f"Datos mal formados:\n{data}")
     except Exception as e:
@@ -97,7 +97,7 @@ def get_text2img(data:dict) -> dict:
         #To-do ->agregar mas información en el error fecha, usuario, reqs
         raise __main.exception(status_code = 403, datail=str(e))
     finally:
-        return __main.api_response(content=__response, media_type="image/png")
+        return __main.api_response(content=__image, media_type="image/png")
 
 @api.post("/texto_a_video/", status_code=201)
 def get_text2video(data:dict) -> dict: