coder160 commited on
Commit
377fd5f
·
1 Parent(s): 16e3929

update response text2img

Browse files
Files changed (2) hide show
  1. api/core/app.py +1 -1
  2. api/main.py +2 -1
api/core/app.py CHANGED
@@ -73,7 +73,7 @@ class Demos:
73
  _resumen = Abstractor.resumen(texto=texto)
74
  return _resumen
75
  @staticmethod
76
- def text_to_img(texto:str=None, model:str="PROMPTHERO")->list:
77
  """Genera un BITARRAY de la imagen con el texto dado.
78
 
79
  args:
 
73
  _resumen = Abstractor.resumen(texto=texto)
74
  return _resumen
75
  @staticmethod
76
+ def text_to_img(texto:str=None, model:str="PROMPTHERO")->bytes|None:
77
  """Genera un BITARRAY de la imagen con el texto dado.
78
 
79
  args:
api/main.py CHANGED
@@ -90,6 +90,7 @@ def get_text2img(data:dict) -> dict:
90
  __response['original']= data.get('texto')
91
  __response['image']= __main.text_to_img(texto=data.get('texto'),
92
  model=data.get('modelo'))
 
93
  else:
94
  raise __main.exception(status_code = 401, datail=f"Datos mal formados:\n{data}")
95
  except Exception as e:
@@ -97,7 +98,7 @@ def get_text2img(data:dict) -> dict:
97
  #To-do ->agregar mas información en el error fecha, usuario, reqs
98
  raise __main.exception(status_code = 403, datail=e)
99
  finally:
100
- return __main.api_response(content=__response, media_type="image/png")
101
 
102
  @api.post("/texto_a_video/", status_code=201)
103
  def get_text2video(data:dict) -> dict:
 
90
  __response['original']= data.get('texto')
91
  __response['image']= __main.text_to_img(texto=data.get('texto'),
92
  model=data.get('modelo'))
93
+ __r = __main.api_response(content=__response, media_type="image/png")
94
  else:
95
  raise __main.exception(status_code = 401, datail=f"Datos mal formados:\n{data}")
96
  except Exception as e:
 
98
  #To-do ->agregar mas información en el error fecha, usuario, reqs
99
  raise __main.exception(status_code = 403, datail=e)
100
  finally:
101
+ return __r
102
 
103
  @api.post("/texto_a_video/", status_code=201)
104
  def get_text2video(data:dict) -> dict: