import asyncio
import base64

class BrowserSingleton:
    _instance = None
    _lock = asyncio.Lock()

    @classmethod
    async def get_instance(cls):
        async with cls._lock:
            if cls._instance is None:
                cls._instance = Browser()  
            return cls._instance

    @classmethod
    async def close(cls):
        async with cls._lock:
            if cls._instance:
                await cls._instance.close()
                cls._instance = None

class AgentManager:
    def __init__(self, agent_id):
        self.id = agent_id
        self.browser = None
        self.context = None
        self.agent = None

    async def run_agent(self):
        self.browser = await BrowserSingleton.get_instance()
        self.context = await self.browser.new_context(
            BrowserContextConfig(
                highlight_elements=False,
                save_recording_path=f"videos/{self.id}",
            )
        )

        self.agent = Agent(
                task="Go to google.com",
                llm=self.llm, #o3-mini
                browser=self.browser,
                browser_context=self.context,
                use_vision=False,
                generate_gif=False,
            )

    async def get_screenshot(self):
        if not self.context or not self.context.session:
            return None

        try:
            page = await self.context.get_current_page()
            screenshot = await page.screenshot(full_page=False, animations="disabled")
            return base64.b64encode(screenshot).decode("utf-8")
        except Exception as e:
            print(f"Error taking screenshot: {e}")
            return None

async def main():
    manager = AgentManager(agent_id="agent_1")
    task_future = asyncio.create_task(manager.run_agent())

    while not task_future.done():
        screenshot = await manager.get_screenshot()
        if screenshot:
            print("Screenshot captured:", screenshot[:30] + "...")  
        await asyncio.sleep(5)

    await task_future 

if __name__ == "__main__":
    asyncio.run(main())