Update playwright_model.py

playwright_model.py

@@ -77,13 +77,87 @@ async def capture_workflows(public_url: str, pdf_filename: str = "workflow_scree
             except Exception as e:
                 print(f"[WARN] Sidebar fix failed: {e}")
 
+        # Highlight sidebar item for current screen
+        async def highlight_sidebar(screen_name):
+            try:
+                await page.evaluate(f"""
+                    (() => {{
+                        const name = `{screen_name}`.toLowerCase().trim();
+                        const sidebarItems = document.querySelectorAll('aside a, nav a, .sidebar a, li a');
+                        
+                        // Remove any previous highlights
+                        sidebarItems.forEach(item => {{
+                            item.style.backgroundColor = '';
+                            item.style.color = '';
+                            item.style.fontWeight = '';
+                        }});
+                        
+                        let matched = null;
+                        for (const item of sidebarItems) {{
+                            const text = (item.textContent || '').toLowerCase().trim();
+                            if (name.includes(text) || text.includes(name)) {{
+                                matched = item;
+                                break;
+                            }}
+                        }}
+                        
+                        // Fallback mapping for alternate names
+                        if (!matched) {{
+                            const mapping = {{
+                                'my dashboard': 'dashboard',
+                                'dashboard': 'dashboard',
+                                'my tasks': 'activity',
+                                'tasks': 'activity',
+                                'activities': 'activity',
+                                'requisitions': 'requisition',
+                                'new requisition': 'requisition'
+                            }};
+                            const keyword = mapping[name];
+                            if (keyword) {{
+                                for (const item of sidebarItems) {{
+                                    const text = (item.textContent || '').toLowerCase().trim();
+                                    if (text.includes(keyword)) {{
+                                        matched = item;
+                                        break;
+                                    }}
+                                }}
+                            }}
+                        }}
+                        
+                        if (matched) {{
+                            matched.style.backgroundColor = '#0078D4';
+                            matched.style.color = 'white';
+                            matched.style.fontWeight = 'bold';
+                            matched.scrollIntoView({{ behavior: 'smooth', block: 'center' }});
+                        }}
+                    }})();
+                """)
+            except Exception as e:
+                print(f"[WARN] Sidebar highlight failed for {screen_name}: {e}")
+
+        # Reset sidebar highlights
+        async def reset_sidebar_highlight():
+            try:
+                await page.evaluate("""
+                    (() => {
+                        const sidebarItems = document.querySelectorAll('aside a, nav a, .sidebar a, li a');
+                        sidebarItems.forEach(item => {
+                            item.style.backgroundColor = '';
+                            item.style.color = '';
+                            item.style.fontWeight = '';
+                        });
+                    })();
+                """)
+            except Exception as e:
+                print(f"[WARN] Sidebar reset failed: {e}")
+
         # Wait for sidebar to appear
         await page.wait_for_selector("aside, .sidebar, nav", timeout=5000)
         await wait_for_layout_stable()
         await fix_sidebar_permanently()
         await asyncio.sleep(1)
 
-        # JS to enumerate workflows and highlight active menu item
+        # JS to enumerate workflows and tabs
         js_logic = """
         (function(){
           const menus=[...document.querySelectorAll('.menu-item')];
@@ -109,23 +183,6 @@ async def capture_workflows(public_url: str, pdf_filename: str = "workflow_scree
           window.__currentIndex=0;
           window.__done=false;
 
-          // Add CSS for active highlight
-          const highlightCSS = document.createElement('style');
-          highlightCSS.innerHTML = `
-            .menu-item.active {
-              background-color: #007bff !important;
-              color: #fff !important;
-              border-radius: 4px;
-              transition: background-color 0.3s ease;
-            }
-            .tab.active, .nav-link.active, [role="tab"].active, .sub-tab.active, .tab-item.active {
-              background-color: #007bff !important;
-              color: #fff !important;
-              border-radius: 4px;
-            }
-          `;
-          document.head.appendChild(highlightCSS);
-
           window.__getSubScreens = function(screen){
             const tabs=[...screen.querySelectorAll('.tab, .nav-link, [role="tab"], [data-tab], .sub-tab, .tab-item')];
             const list=[];
@@ -187,10 +244,12 @@ async def capture_workflows(public_url: str, pdf_filename: str = "workflow_scree
             screenshot_path = os.path.join(OUTPUT_DIR, f"{screen_name}.png")
             print(f"📸 Capturing main screen: {screen_name}")
 
+            await highlight_sidebar(screen_name)
             await wait_for_layout_stable()
             await fix_sidebar_permanently()
             await asyncio.sleep(0.8)
             await page.screenshot(path=screenshot_path, full_page=True)
+            await reset_sidebar_highlight()
             screenshots.append(screenshot_path)
 
             first_active_skipped = False
@@ -213,11 +272,13 @@ async def capture_workflows(public_url: str, pdf_filename: str = "workflow_scree
                 await page.evaluate(f"window.__clickSubScreen('{sub}')")
                 await wait_for_layout_stable()
                 await fix_sidebar_permanently()
+                await highlight_sidebar(f"{screen_name} - {sub}")
                 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")
                 await page.screenshot(path=sub_path, full_page=True)
+                await reset_sidebar_highlight()
                 screenshots.append(sub_path)
 
             index += 1