Update playwright_model.py

playwright_model.py

@@ -68,34 +68,22 @@ async def capture_workflows(public_url: str, pdf_filename: str = "workflow_scree
                 break
 
             screen_name = result.get("screenName", f"screen_{index}")
+            screenshot_path = os.path.join(OUTPUT_DIR, f"{screen_name}.png")
+            print(f"📸 Capturing {screen_name} -> {screenshot_path}")
 
-            # 🧠 Dynamically detect readable title name
-            readable_name = await page.evaluate(
-                """(defaultName) => {
-                    let title = null;
-                    const activeMenu = document.querySelector('.menu-item.active');
-                    const visibleTitle = document.querySelector('.page-title, .header-title, main h2');
-                    if (activeMenu && activeMenu.textContent.trim()) title = activeMenu.textContent.trim();
-                    else if (visibleTitle && visibleTitle.textContent.trim()) title = visibleTitle.textContent.trim();
-                    else title = defaultName;
-                    return title;
-                }""",
-                screen_name
-            )
-
-            # 🪄 Update visible + browser tab title dynamically
+            # ✅ Dynamically update visible and tab title before screenshot
             await page.evaluate(
-                """(titleName) => {
+                """(name) => {
+                    // Update visible title text
                     const titleEl = document.querySelector('.page-title, .header-title, main h2');
-                    if (titleEl) titleEl.textContent = titleName;
-                    document.title = titleName;
+                    if (titleEl) titleEl.textContent = name;
+
+                    // Update browser tab title
+                    document.title = name;
                 }""",
-                readable_name
+                screen_name
             )
 
-            screenshot_path = os.path.join(OUTPUT_DIR, f"{readable_name}.png")
-            print(f"📸 Capturing {readable_name} -> {screenshot_path}")
-
             await asyncio.sleep(1.2)  # wait for animations/UI updates
             await page.screenshot(path=screenshot_path, full_page=True)
             screenshots.append(screenshot_path)
@@ -103,7 +91,7 @@ async def capture_workflows(public_url: str, pdf_filename: str = "workflow_scree
 
         await browser.close()
 
-        # ✅ Combine into PDF
+        # ✅ Combine all screenshots into a single PDF
         if not screenshots:
             raise RuntimeError("No screenshots captured — check if .screen elements exist!")
 
@@ -122,4 +110,5 @@ async def capture_workflows(public_url: str, pdf_filename: str = "workflow_scree
         print(f"✅ PDF generated successfully: {pdf_path}")
         return pdf_path
 
+
 generate_ui_report = capture_workflows