Update playwright_model.py

playwright_model.py

@@ -17,22 +17,45 @@ 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")
 
-        # ✅ Improved sidebar + layout stability detector
+        # ✅ Detect and freeze sidebar layout to fixed position
+        async def fix_sidebar_position():
+            try:
+                await page.evaluate("""
+                    const sidebar = document.querySelector('aside, .sidebar, nav');
+                    if (sidebar) {
+                        // Lock the sidebar in a consistent fixed position
+                        const rect = sidebar.getBoundingClientRect();
+                        sidebar.style.position = 'fixed';
+                        sidebar.style.top = '0px';
+                        sidebar.style.left = rect.left + 'px';
+                        sidebar.style.width = rect.width + 'px';
+                        sidebar.style.height = '100vh';
+                        sidebar.style.zIndex = '1000';
+                        sidebar.style.boxShadow = '2px 0 6px rgba(0,0,0,0.05)';
+                        sidebar.style.overflow = 'hidden';
+                    }
+
+                    // Shift main content to align with fixed sidebar
+                    const main = document.querySelector('main, .main-content, .app-body, .content');
+                    if (main && sidebar) {
+                        main.style.marginLeft = sidebar.offsetWidth + 'px';
+                    }
+                """)
+            except Exception as e:
+                print(f"[WARN] Failed to fix sidebar: {e}")
+
         async def wait_for_sidebar_stable():
             try:
                 await page.evaluate("""
                     (async () => {
                         const sidebar = document.querySelector('aside, .sidebar, nav');
                         if (!sidebar) return;
-
                         function getMetrics() {
                             const rect = sidebar.getBoundingClientRect();
                             return { left: rect.left, width: rect.width };
                         }
-
                         let stableCount = 0;
                         let last = getMetrics();
-
                         await new Promise((resolve) => {
                             const check = setInterval(() => {
                                 const now = getMetrics();
@@ -43,13 +66,13 @@ async def capture_workflows(public_url: str, pdf_filename: str = "workflow_scree
                                     stableCount++;
                                     if (stableCount >= 5) {
                                         clearInterval(check);
-                                        setTimeout(resolve, 400); // safety delay
+                                        setTimeout(resolve, 300);
                                     }
                                 } else {
                                     stableCount = 0;
                                     last = now;
                                 }
-                            }, 120);
+                            }, 100);
                         });
                     })();
                 """)
@@ -79,7 +102,7 @@ async def capture_workflows(public_url: str, pdf_filename: str = "workflow_scree
                 })();
             """)
 
-        # ✅ Inject traversal logic
+        # Inject traversal logic
         js_logic = """
         (function(){
           const menus=[...document.querySelectorAll('.menu-item')];
@@ -146,7 +169,6 @@ async def capture_workflows(public_url: str, pdf_filename: str = "workflow_scree
         screenshots = []
         index = 0
 
-        # ✅ Loop through workflows
         while True:
             result = await page.evaluate("window.__captureNext()")
             if not result:
@@ -168,27 +190,13 @@ async def capture_workflows(public_url: str, pdf_filename: str = "workflow_scree
 
             await wait_for_sidebar_stable()
             await wait_for_layout_stable()
+            await fix_sidebar_position()
+            await asyncio.sleep(0.8)
 
-            # ✅ Fix sidebar height mismatch before capture
-            await page.evaluate("""
-                const sidebar = document.querySelector('aside, .sidebar, nav');
-                const main = document.querySelector('main, .main-content, .app-body, .content');
-                if (sidebar) {
-                    const fullHeight = Math.max(
-                        document.body.scrollHeight,
-                        document.documentElement.scrollHeight,
-                        main ? main.scrollHeight : 0
-                    );
-                    sidebar.style.minHeight = fullHeight + 'px';
-                    sidebar.style.height = fullHeight + 'px';
-                }
-            """)
-
-            await asyncio.sleep(1.0)
             await page.screenshot(path=screenshot_path, full_page=True)
             screenshots.append(screenshot_path)
 
-            # ✅ Capture sub-screens
+            # Capture sub-screens
             first_active_skipped = False
             for sub in sub_screens:
                 is_active = await page.evaluate(
@@ -209,23 +217,8 @@ async def capture_workflows(public_url: str, pdf_filename: str = "workflow_scree
                 await page.evaluate(f"window.__clickSubScreen('{sub}')")
                 await wait_for_sidebar_stable()
                 await wait_for_layout_stable()
-
-                # ✅ Sidebar height fix again for sub-screens
-                await page.evaluate("""
-                    const sidebar = document.querySelector('aside, .sidebar, nav');
-                    const main = document.querySelector('main, .main-content, .app-body, .content');
-                    if (sidebar) {
-                        const fullHeight = Math.max(
-                            document.body.scrollHeight,
-                            document.documentElement.scrollHeight,
-                            main ? main.scrollHeight : 0
-                        );
-                        sidebar.style.minHeight = fullHeight + 'px';
-                        sidebar.style.height = fullHeight + 'px';
-                    }
-                """)
-
-                await asyncio.sleep(1.0)
+                await fix_sidebar_position()
+                await asyncio.sleep(0.8)
 
                 sub_name_clean = sub.replace(" ", "_").lower()
                 sub_path = os.path.join(OUTPUT_DIR, f"{screen_name}_{sub_name_clean}.png")
@@ -267,4 +260,4 @@ async def capture_workflows(public_url: str, pdf_filename: str = "workflow_scree
         return pdf_path
 
 
-generate_ui_report = capture_workflows
+generate_ui_report = capture_workflows