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")