Update playwright_model.py

playwright_model.py

@@ -16,6 +16,30 @@ async def capture_workflows(public_url: str, pdf_filename: str = "workflow_scree
         print(f"Opening page: {public_url}")
         await page.goto(public_url, wait_until="load")
 
+        # ✅ Helper JS to wait until sidebar finishes sliding
+        async def wait_for_sidebar_stable():
+            await page.evaluate("""
+                return new Promise(resolve => {
+                    const sidebar = document.querySelector('.sidebar, .side-nav, .menu-panel, .navigation, .drawer');
+                    if (!sidebar) return resolve();
+                    let lastPos = sidebar.getBoundingClientRect().left;
+                    let stableCount = 0;
+
+                    const check = () => {
+                        const newPos = sidebar.getBoundingClientRect().left;
+                        if (Math.abs(newPos - lastPos) < 0.5) {
+                            stableCount++;
+                        } else {
+                            stableCount = 0;
+                            lastPos = newPos;
+                        }
+                        if (stableCount > 5) return resolve();
+                        requestAnimationFrame(check);
+                    };
+                    check();
+                });
+            """)
+
         # ✅ Inject workflow traversal + sub-screen logic
         js_logic = """
         (function(){
@@ -106,6 +130,9 @@ async def capture_workflows(public_url: str, pdf_filename: str = "workflow_scree
                 screen_name
             )
 
+            # ✅ Wait until sidebar (if any) has finished sliding
+            await wait_for_sidebar_stable()
+
             await asyncio.sleep(1.2)
             await page.screenshot(path=screenshot_path, full_page=True)
             screenshots.append(screenshot_path)
@@ -131,7 +158,11 @@ async def capture_workflows(public_url: str, pdf_filename: str = "workflow_scree
 
                 print(f"   ↳ Capturing sub-screen: {sub}")
                 await page.evaluate(f"window.__clickSubScreen('{sub}')")
+
+                # ✅ Wait until sidebar (if any) is stable again
+                await wait_for_sidebar_stable()
                 await asyncio.sleep(1.0)
+
                 sub_name_clean = sub.replace(" ", "_").lower()
                 sub_path = os.path.join(OUTPUT_DIR, f"{screen_name}_{sub_name_clean}.png")