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")
 
-        # ✅ Inject workflow traversal logic
+        # ✅ Inject workflow traversal + sub-screen logic
         js_logic = """
         (function(){
           const menus=[...document.querySelectorAll('.menu-item')];
@@ -36,6 +36,18 @@ async def capture_workflows(public_url: str, pdf_filename: str = "workflow_scree
           window.__visitedWorkflows=[];
           window.__currentIndex=0;
           window.__done=false;
+
+          // helper: get all visible sub-screen tabs dynamically
+          window.__getSubScreens = function(screen){
+            const tabs=[...screen.querySelectorAll('.tab, .nav-link, [role="tab"], [data-tab], .sub-tab, .tab-item')];
+            const list=[];
+            for(const t of tabs){
+              const sub=t.textContent?.trim();
+              if(sub && !list.includes(sub)) list.push(sub);
+            }
+            return list;
+          };
+
           window.__captureNext=async function(){
             if(window.__done) return false;
             if(window.__currentIndex>=window.__ordered.length){window.__done=true;return false;}
@@ -43,7 +55,10 @@ async def capture_workflows(public_url: str, pdf_filename: str = "workflow_scree
             const {menu,screen}=pair;
             if(!screen){window.__done=true;return false;}
             const wfName=screen.id || screen.getAttribute('data-name') || ('screen_'+window.__currentIndex);
-            if(window.__visitedWorkflows.includes(wfName)){window.__done=true;return false;}
+            if(window.__visitedWorkflows.includes(wfName)){
+              window.__currentIndex++;
+              return window.__captureNext();
+            }
             window.__visitedWorkflows.push(wfName);
             document.querySelectorAll('.screen').forEach(s=>s.classList.remove('active'));
             document.querySelectorAll('.menu-item').forEach(m=>m.classList.remove('active'));
@@ -51,7 +66,16 @@ async def capture_workflows(public_url: str, pdf_filename: str = "workflow_scree
             if(menu) menu.classList.add('active');
             screen.scrollIntoView({behavior:'smooth',block:'center'});
             window.__currentIndex++;
-            return {screenName:wfName};
+            const subs = window.__getSubScreens(screen);
+            return {screenName:wfName, subScreens:subs};
+          };
+
+          // click sub-screen tab by text
+          window.__clickSubScreen = async function(name){
+            const tabs=[...document.querySelectorAll('.tab, .nav-link, [role="tab"], [data-tab], .sub-tab, .tab-item')];
+            const t=tabs.find(x=>x.textContent.trim()===name);
+            if(t){t.click(); return true;}
+            return false;
           };
         })();
         """
@@ -68,10 +92,11 @@ async def capture_workflows(public_url: str, pdf_filename: str = "workflow_scree
                 break
 
             screen_name = result.get("screenName", f"screen_{index}")
+            sub_screens = result.get("subScreens", [])
             screenshot_path = os.path.join(OUTPUT_DIR, f"{screen_name}.png")
             print(f"📸 Capturing main screen: {screen_name}")
 
-            # ✅ Update visible title and browser tab
+            # Update visible + browser tab title dynamically
             await page.evaluate(
                 """(name) => {
                     const titleEl = document.querySelector('.page-title, .header-title, main h2');
@@ -85,41 +110,32 @@ async def capture_workflows(public_url: str, pdf_filename: str = "workflow_scree
             await page.screenshot(path=screenshot_path, full_page=True)
             screenshots.append(screenshot_path)
 
-            # ✅ Capture sub-screens if any
-            sub_screens = await page.evaluate(
-                """() => {
-                    const subs = [...document.querySelectorAll('.tab-button, .sub-screen-tab, [data-subscreen]')];
-                    return subs.map(s => s.textContent.trim()).filter(Boolean);
-                }"""
-            )
-
-            # Detect active sub-screen (default one)
-            default_sub = await page.evaluate(
-                """() => {
-                    const active = document.querySelector('.tab-button.active, .sub-screen-tab.active, [data-subscreen].active');
-                    return active ? active.textContent.trim() : null;
-                }"""
-            )
-
+            # ✅ Capture sub-screens dynamically (skip first active one)
+            first_active_skipped = False
             for sub in sub_screens:
-                # 🧠 Skip default sub-screen (already captured above)
-                if sub == default_sub:
-                    print(f"⏩ Skipping duplicate default sub-screen: {sub}")
-                    continue
-
-                print(f"   ↳ Capturing sub-screen: {sub}")
-                await page.evaluate(
-                    """(name) => {
-                        const btns = [...document.querySelectorAll('.tab-button, .sub-screen-tab, [data-subscreen]')];
-                        const btn = btns.find(b => b.textContent.trim() === name);
-                        if (btn) btn.click();
+                # Check if this sub tab is already active
+                is_active = await page.evaluate(
+                    """(subText) => {
+                        const tabs=[...document.querySelectorAll('.tab, .nav-link, [role="tab"], [data-tab], .sub-tab, .tab-item')];
+                        const t=tabs.find(x=>x.textContent.trim()===subText);
+                        if(!t) return false;
+                        const cls=t.getAttribute('class')||'';
+                        return cls.includes('active');
                     }""",
                     sub
                 )
 
-                await asyncio.sleep(1.2)
+                if is_active and not first_active_skipped:
+                    first_active_skipped = True
+                    continue  # skip first active sub-screen
+
+                print(f"   ↳ Capturing sub-screen: {sub}")
+                await page.evaluate(f"window.__clickSubScreen('{sub}')")
+                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")
 
-                # Update title for sub-screen
+                # Update tab title for sub-screen
                 await page.evaluate(
                     """([mainName, subName]) => {
                         const titleEl = document.querySelector('.page-title, .header-title, main h2');
@@ -130,7 +146,6 @@ async def capture_workflows(public_url: str, pdf_filename: str = "workflow_scree
                     [screen_name, sub]
                 )
 
-                sub_path = os.path.join(OUTPUT_DIR, f"{screen_name}_{sub}.png")
                 await page.screenshot(path=sub_path, full_page=True)
                 screenshots.append(sub_path)
 
@@ -138,7 +153,7 @@ async def capture_workflows(public_url: str, pdf_filename: str = "workflow_scree
 
         await browser.close()
 
-        # ✅ Combine into PDF
+        # ✅ Combine all screenshots into PDF
         if not screenshots:
             raise RuntimeError("No screenshots captured — check if .screen elements exist!")