Spaces:
Running on Zero
Running on Zero
| """Keep BrowserGym's synchronous Playwright connection local to each invocation.""" | |
| from contextlib import contextmanager | |
| def browser_session(): | |
| """Create and close Playwright in the worker that actually uses the browser. | |
| BrowserGym caches a global connection. ZeroGPU workers must not reuse a | |
| connection inherited from another process or a previous Gradio thread. | |
| Callers serialize browser work because BrowserGym's cache is process-global. | |
| """ | |
| import browsergym.core | |
| from playwright.sync_api import sync_playwright | |
| # An inherited connection belongs to its original loop: do not call stop on it. | |
| browsergym.core._set_global_playwright(None) | |
| with sync_playwright() as playwright: | |
| browsergym.core._set_global_playwright(playwright) | |
| try: | |
| yield | |
| finally: | |
| browsergym.core._set_global_playwright(None) | |