import aiohttp, json


class RenderVideo:
    def __init__(self, url="https://yakova-rectifier.hf.space/create-video"):
        self.url = url

    @staticmethod
    def _parse_response(response: str) -> dict:
        return json.loads(response)

    async def post_request(self, data: dict) -> dict:
        headers = {"Accept": "application/json", "Content-Type": "application/json"}
        async with aiohttp.ClientSession() as session:
            async with session.post(
                self.url, data=json.dumps(data), headers=headers
            ) as resp:
                response = await resp.text()
                result = self._parse_response(response)
        return result

    async def render_video(self, data) -> dict:
        result = await self.post_request(data)
        return result