coder160/gpu_machine_learner

1from api.core.app import Demos2__main = Demos()3api = __main.api()4 5@api.post("/texto_a_imagen/", status_code=201, responses = {201: {"content": {"image/png": {}}}} ,response_class=__main.api_response)6def get_text2img(data:dict):7    __response=dict({"request_data":data})8    try:9        if data and 'texto' in data and 'modelo' in data:10            __response['original']= data.get('texto')11            __image =  __main.text_to_img(texto=data.get('texto'), 12                                          model=data.get('modelo'))13        else:14            raise __main.exception(status_code = 401, datail=f"Datos mal formados:\n{data}")15    except Exception as e:16        print(e)17        #To-do ->agregar mas información en el error fecha, usuario, reqs18        raise __main.exception(status_code = 403, datail=str(e))19    finally:20        return __main.api_response(content=__image, media_type="image/png")21 22@api.post("/texto_a_video/", status_code=201)23def get_text2video(data:dict) -> dict:24    __response=dict({"request_data":data})25    try:26        if data:27            __response['original']= data.get('texto')28            __response['video']= __main.text_to_video(texto=data.get('texto'))29        else:30            raise __main.exception(status_code = 401, datail=f"Datos mal formados:\n{data}")31    except Exception as e:32        print(e)33        #To-do ->agregar mas información en el error fecha, usuario, reqs34        raise __main.exception(status_code = 403, datail=e)35    finally:36        return __response37 38@api.post("/texto_a_audio/", status_code=201)39def get_text2speach(data:dict) -> dict:40    __response=dict({"request_data":data})41    try:42        if data:43            __response['original']= data.get('texto')44            __response['audio']= __main.text_to_speach(texto=data.get('texto'))45        else:46            raise __main.exception(status_code = 401, datail=f"Datos mal formados:\n{data}")47    except Exception as e:48        print(e)49        #To-do ->agregar mas información en el error fecha, usuario, reqs50        raise __main.exception(status_code = 403, datail=e)51    finally:52        return __response53 54 55 56@api.post("/imagen_a_imagen/", status_code=201, responses = {201: {"content": {"image/png": {}}}} ,response_class=__main.api_response)57def get_img2img(data:dict, file: bytes = __main.fileupload(...)):58    __response=dict({"request_data":data})59    try:60        if data and 'texto' in data and 'modelo' in data and file is not None:61            __response['original']= data.get('texto')62            __image =  file63        else:64            raise __main.exception(status_code = 401, datail=f"Datos mal formados:\n{data}")65    except Exception as e:66        print(e)67        #To-do ->agregar mas información en el error fecha, usuario, reqs68        raise __main.exception(status_code = 403, datail=str(e))69    finally:70        return __main.api_response(content=__image, media_type="image/png")71