coder160's picture
update file input
786049d
from api.core.app import Demos
__main = Demos()
api = __main.api()
@api.post("/texto_a_imagen/", status_code=201, responses = {201: {"content": {"image/png": {}}}} ,response_class=__main.api_response)
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')
__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:
print(e)
#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=__image, media_type="image/png")
@api.post("/texto_a_video/", status_code=201)
def get_text2video(data:dict) -> dict:
__response=dict({"request_data":data})
try:
if data:
__response['original']= data.get('texto')
__response['video']= __main.text_to_video(texto=data.get('texto'))
else:
raise __main.exception(status_code = 401, datail=f"Datos mal formados:\n{data}")
except Exception as e:
print(e)
#To-do ->agregar mas información en el error fecha, usuario, reqs
raise __main.exception(status_code = 403, datail=e)
finally:
return __response
@api.post("/texto_a_audio/", status_code=201)
def get_text2speach(data:dict) -> dict:
__response=dict({"request_data":data})
try:
if data:
__response['original']= data.get('texto')
__response['audio']= __main.text_to_speach(texto=data.get('texto'))
else:
raise __main.exception(status_code = 401, datail=f"Datos mal formados:\n{data}")
except Exception as e:
print(e)
#To-do ->agregar mas información en el error fecha, usuario, reqs
raise __main.exception(status_code = 403, datail=e)
finally:
return __response
@api.post("/imagen_a_imagen/", status_code=201, responses = {201: {"content": {"image/png": {}}}} ,response_class=__main.api_response)
def get_img2img(data:dict, file: bytes = __main.fileupload(...)):
__response=dict({"request_data":data})
try:
if data and 'texto' in data and 'modelo' in data and file is not None:
__response['original']= data.get('texto')
__image = file
else:
raise __main.exception(status_code = 401, datail=f"Datos mal formados:\n{data}")
except Exception as e:
print(e)
#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=__image, media_type="image/png")