Update playwright_model.py

playwright_model.py

@@ -13,17 +13,16 @@ async def capture_workflows(public_url: str, pdf_filename: str = "workflow_scree
     async with async_playwright() as p:
         browser = await p.chromium.launch(headless=True)
         page = await browser.new_page()
-        print(f"🌐 Opening page: {public_url}")
+        print(f"Opening page: {public_url}")
         await page.goto(public_url, wait_until="load")
 
-        # ✅ Inject enhanced logic to capture main and sub screens dynamically
+        # ✅ Inject workflow traversal + sub-screen logic
         js_logic = """
         (function(){
           const menus=[...document.querySelectorAll('.menu-item')];
           const screens=[...document.querySelectorAll('.screen')];
           window.__ordered=[];
           const seen=new Set();
-
           for(const m of menus){
             let id=m.dataset.screen||m.dataset.target;
             if(!id){
@@ -34,20 +33,20 @@ async def capture_workflows(public_url: str, pdf_filename: str = "workflow_scree
             if(s && !seen.has(s)){seen.add(s);window.__ordered.push({menu:m,screen:s});}
           }
           for(const s of screens) if(!seen.has(s)) window.__ordered.push({menu:null,screen:s});
-
           window.__visitedWorkflows=[];
           window.__currentIndex=0;
           window.__done=false;
 
-          async function getSubScreens(screen){
-            const tabs=[...screen.querySelectorAll('.tab, .nav-link, [role="tab"], [data-tab]')];
-            const results=[];
+          // 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 subName=t.textContent?.trim().replace(/\\s+/g,'-').toLowerCase();
-              if(subName) results.push({tab:t, name:subName});
+              const sub=t.textContent?.trim();
+              if(sub && !list.includes(sub)) list.push(sub);
             }
-            return results;
-          }
+            return list;
+          };
 
           window.__captureNext=async function(){
             if(window.__done) return false;
@@ -55,52 +54,38 @@ async def capture_workflows(public_url: str, pdf_filename: str = "workflow_scree
             const pair=window.__ordered[window.__currentIndex];
             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.__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'));
             screen.classList.add('active');
             if(menu) menu.classList.add('active');
             screen.scrollIntoView({behavior:'smooth',block:'center'});
-
-            const mainTitle = screen.querySelector('.page-title, .header-title, h2')?.textContent?.trim() || wfName;
-            document.title = mainTitle;
-
-            const subs = await getSubScreens(screen);
             window.__currentIndex++;
-            return {screenName: wfName, subScreens: subs.map(x=>x.name)};
+            const subs = window.__getSubScreens(screen);
+            return {screenName:wfName, subScreens:subs};
           };
 
-          window.__clickSubScreen = async function(mainName, subName){
-            const screen = document.getElementById(mainName);
-            if(!screen) return false;
-            const tab=[...screen.querySelectorAll('.tab, .nav-link, [role="tab"], [data-tab]')]
-              .find(t=>t.textContent.trim().toLowerCase().includes(subName));
-            if(tab){
-              tab.click();
-              await new Promise(r=>setTimeout(r,700));
-              const title = screen.querySelector('.page-title, .header-title, h2')?.textContent?.trim() || mainName;
-              document.title = title + ' - ' + subName;
-              return true;
-            }
+          // 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;
           };
         })();
         """
-
         await page.evaluate(js_logic)
         await asyncio.sleep(1)
 
         screenshots = []
         index = 0
 
-        # ✅ Loop through main workflows and their sub-screens dynamically
+        # ✅ Loop through each workflow
         while True:
             result = await page.evaluate("window.__captureNext()")
             if not result:
@@ -108,21 +93,42 @@ async def capture_workflows(public_url: str, pdf_filename: str = "workflow_scree
 
             screen_name = result.get("screenName", f"screen_{index}")
             sub_screens = result.get("subScreens", [])
-
-            # Capture main screen
             screenshot_path = os.path.join(OUTPUT_DIR, f"{screen_name}.png")
             print(f"📸 Capturing main screen: {screen_name}")
 
+            # Update visible + browser tab title dynamically
+            await page.evaluate(
+                """(name) => {
+                    const titleEl = document.querySelector('.page-title, .header-title, main h2');
+                    if (titleEl) titleEl.textContent = name;
+                    document.title = name;
+                }""",
+                screen_name
+            )
+
             await asyncio.sleep(1.2)
             await page.screenshot(path=screenshot_path, full_page=True)
             screenshots.append(screenshot_path)
 
-            # Capture sub-screens/tabs
+            # ✅ Capture sub-screens dynamically
             for sub in sub_screens:
-                print(f"  ↳ Capturing sub-screen: {sub}")
-                await page.evaluate(f"window.__clickSubScreen('{screen_name}','{sub}')")
+                print(f"   ↳ Capturing sub-screen: {sub}")
+                await page.evaluate(f"window.__clickSubScreen('{sub}')")
                 await asyncio.sleep(1.0)
-                sub_path = os.path.join(OUTPUT_DIR, f"{screen_name}_{sub}.png")
+                sub_name_clean = sub.replace(" ", "_").lower()
+                sub_path = os.path.join(OUTPUT_DIR, f"{screen_name}_{sub_name_clean}.png")
+
+                # update tab title for sub-screen
+                await page.evaluate(
+                    """(mainName, subName) => {
+                        const titleEl = document.querySelector('.page-title, .header-title, main h2');
+                        const newTitle = mainName + ' - ' + subName;
+                        if (titleEl) titleEl.textContent = newTitle;
+                        document.title = newTitle;
+                    }""",
+                    screen_name, sub
+                )
+
                 await page.screenshot(path=sub_path, full_page=True)
                 screenshots.append(sub_path)