Update playwright_model.py

playwright_model.py

@@ -16,7 +16,7 @@ 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")
 
-        # Wait until layout stabilizes
+        # --- Utility: wait for stable layout ---
         async def wait_for_layout_stable():
             await page.evaluate("""
                 (async () => {
@@ -40,55 +40,54 @@ async def capture_workflows(public_url: str, pdf_filename: str = "workflow_scree
                 })();
             """)
 
-        # Sidebar fixed; top bar remains dynamic
-        async def fix_sidebar_only():
+        # --- Auto-detect top bar and fix sidebar below it ---
+        async def fix_sidebar_permanently():
             try:
                 await page.evaluate("""
-                    // Unfix all headers / topbars
-                    document.querySelectorAll('header, .topbar, .top-bar, .header, .navbar').forEach(el => {
-                        el.style.position = 'relative';
-                        el.style.top = 'auto';
-                        el.style.zIndex = 'auto';
-                    });
-
-                    // Fix only sidebar
                     const sidebar = document.querySelector('aside, .sidebar, nav');
+                    const topbar = document.querySelector('.topbar, header, .header, .app-header, .navbar, .page-header');
+
+                    let topOffset = 0;
+                    if (topbar) {
+                        const rect = topbar.getBoundingClientRect();
+                        topOffset = rect.bottom || rect.height || 60;
+                    }
+
                     if (sidebar) {
                         const rect = sidebar.getBoundingClientRect();
                         const sidebarWidth = rect.width || sidebar.offsetWidth || 220;
                         sidebar.style.position = 'fixed';
-                        sidebar.style.top = '0';
+                        sidebar.style.top = topOffset + 'px';
                         sidebar.style.left = '0';
-                        sidebar.style.height = '100vh';
+                        sidebar.style.height = `calc(100vh - ${topOffset}px)`;
                         sidebar.style.width = sidebarWidth + 'px';
-                        sidebar.style.zIndex = '999';
-                        sidebar.style.overflowY = 'auto';
+                        sidebar.style.zIndex = '1000';
+                        sidebar.style.overflow = 'hidden';
                         sidebar.style.transition = 'none';
                         sidebar.dataset.locked = 'true';
                     }
 
-                    // Shift main content right to avoid overlap
                     const main = document.querySelector('main, .main-content, .app-body, .content');
                     if (main && sidebar) {
                         main.style.marginLeft = sidebar.offsetWidth + 'px';
                         main.style.position = 'relative';
                     }
 
-                    // Reset scrolling and page margins
+                    window.scrollTo(0, 0);
                     document.body.style.margin = '0';
                     document.body.style.padding = '0';
                     document.body.style.overflowX = 'hidden';
-                    window.scrollTo(0, 0);
                 """)
             except Exception as e:
                 print(f"[WARN] Sidebar fix failed: {e}")
 
+        # --- Wait until sidebar exists ---
         await page.wait_for_selector("aside, .sidebar, nav", timeout=5000)
         await wait_for_layout_stable()
-        await fix_sidebar_only()
+        await fix_sidebar_permanently()
         await asyncio.sleep(1)
 
-        # JS navigation logic
+        # --- JS Logic: detect workflows and tabs dynamically ---
         js_logic = """
         (function(){
           const menus=[...document.querySelectorAll('.menu-item')];
@@ -164,7 +163,7 @@ async def capture_workflows(public_url: str, pdf_filename: str = "workflow_scree
         screenshots = []
         index = 0
 
-        # Loop through screens
+        # --- Capture all screens ---
         while True:
             result = await page.evaluate("window.__captureNext()")
             if not result:
@@ -176,11 +175,12 @@ async def capture_workflows(public_url: str, pdf_filename: str = "workflow_scree
             print(f"📸 Capturing main screen: {screen_name}")
 
             await wait_for_layout_stable()
-            await fix_sidebar_only()
+            await fix_sidebar_permanently()
             await asyncio.sleep(1.0)
             await page.screenshot(path=screenshot_path, full_page=True)
             screenshots.append(screenshot_path)
 
+            # --- Sub-tabs ---
             first_active_skipped = False
             for sub in sub_screens:
                 is_active = await page.evaluate(
@@ -200,7 +200,7 @@ 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}')")
                 await wait_for_layout_stable()
-                await fix_sidebar_only()
+                await fix_sidebar_permanently()
                 await asyncio.sleep(1.0)
 
                 sub_name_clean = sub.replace(" ", "_").lower()
@@ -212,6 +212,7 @@ async def capture_workflows(public_url: str, pdf_filename: str = "workflow_scree
 
         await browser.close()
 
+        # --- Generate PDF ---
         if not screenshots:
             raise RuntimeError("No screenshots captured — check if .screen elements exist!")