scitex 2.5.0__py3-none-any.whl → 2.7.0__py3-none-any.whl

scitex/capture/__init__.py

@@ -27,16 +27,12 @@ Usage:
     capture.stop()
 """
 
-from .utils import (
-    capture,
-    start_monitor,
-    stop_monitor
-)
+from .utils import capture, start_monitor, stop_monitor
 from .gif import (
     create_gif_from_session,
     create_gif_from_files,
     create_gif_from_pattern,
-    create_gif_from_latest_session
+    create_gif_from_latest_session,
 )
 from .session import session
 from .capture import CaptureManager
@@ -44,15 +40,18 @@ from .capture import CaptureManager
 # Global manager for monitor enumeration
 _manager = CaptureManager()
 
+
 def get_info():
     """Get comprehensive display info (monitors, windows, virtual desktops)."""
     return _manager.get_info()
 
+
 # Simpler, clearer aliases
-get_info = get_info      # Primary: simple and clear
+get_info = get_info  # Primary: simple and clear
 list_windows = get_info  # Alternative: focus on windows
 get_display_info = get_info  # Legacy
 
+
 def capture_window(window_handle: int, output_path: str = None):
     """
     Capture a specific window by its handle.
@@ -74,10 +73,11 @@ def capture_window(window_handle: int, output_path: str = None):
     """
     return _manager.capture_window(window_handle, output_path)
 
+
 # Convenience aliases - these are the main public API
-snap = capture     # Primary: natural camera action
-take = capture     # Alternative: "take a picture"
-cpt = capture      # Legacy: backwards compatibility
+snap = capture  # Primary: natural camera action
+take = capture  # Alternative: "take a picture"
+cpt = capture  # Legacy: backwards compatibility
 start = start_monitor
 stop = stop_monitor
 
@@ -92,19 +92,19 @@ __email__ = "Yusuke.Watanabe@scitex.ai"
 # Only expose the essential functions
 __all__ = [
     "capture",
-    "snap",        # Primary API
-    "take",        # Alternative API
-    "cpt",         # Legacy
+    "snap",  # Primary API
+    "take",  # Alternative API
+    "cpt",  # Legacy
     "start",
     "stop",
     "session",
-    "get_info",          # Primary: get all display info
-    "list_windows",      # Alternative: focus on windows
-    "get_info", # Legacy
+    "get_info",  # Primary: get all display info
+    "list_windows",  # Alternative: focus on windows
+    "get_info",  # Legacy
     "get_display_info",  # Legacy
     "capture_window",
     "create_gif_from_session",
     "create_gif_from_files",
     "create_gif_from_pattern",
-    "create_gif_from_latest_session"
+    "create_gif_from_latest_session",
 ]

scitex/capture/__main__.py

@@ -5,9 +5,8 @@
 # ----------------------------------------
 from __future__ import annotations
 import os
-__FILE__ = (
-    "./src/scitex/capture/__main__.py"
-)
+
+__FILE__ = "./src/scitex/capture/__main__.py"
 __DIR__ = os.path.dirname(__FILE__)
 # ----------------------------------------
 

scitex/capture/capture.py

@@ -5,9 +5,8 @@
 # ----------------------------------------
 from __future__ import annotations
 import os
-__FILE__ = (
-    "./src/scitex/capture/capture.py"
-)
+
+__FILE__ = "./src/scitex/capture/capture.py"
 __DIR__ = os.path.dirname(__FILE__)
 # ----------------------------------------
 
@@ -91,9 +90,7 @@ class ScreenshotWorker:
 
         self.running = True
         self.screenshot_count = 0
-        self.session_id = session_id or datetime.now().strftime(
-            "%Y%m%d_%H%M%S"
-        )
+        self.session_id = session_id or datetime.now().strftime("%Y%m%d_%H%M%S")
 
         # Start worker thread
         self.worker_thread = threading.Thread(
@@ -173,7 +170,9 @@ class ScreenshotWorker:
             now = datetime.now()
             timestamp = now.strftime("%Y%m%d_%H%M%S_%f")[:-3]
             ext = "jpg" if self.use_jpeg else "png"
-            filename = f"{self.session_id}_{self.screenshot_count:04d}_{timestamp}.{ext}"
+            filename = (
+                f"{self.session_id}_{self.screenshot_count:04d}_{timestamp}.{ext}"
+            )
             filepath = self.output_dir / filename
 
             # Try Windows PowerShell method for WSL
@@ -200,10 +199,7 @@ class ScreenshotWorker:
 
     def _is_wsl(self) -> bool:
         """Check if running in WSL."""
-        return (
-            sys.platform == "linux"
-            and "microsoft" in os.uname().release.lower()
-        )
+        return sys.platform == "linux" and "microsoft" in os.uname().release.lower()
 
     def _capture_windows_screen(
         self, filepath: Path, monitor: int = 1, capture_all: bool = False
@@ -225,7 +221,6 @@ class ScreenshotWorker:
 
             # Check if script exists
             if script_path.exists():
-
                 # Find PowerShell executable
                 ps_paths = [
                     "powershell.exe",
@@ -388,9 +383,7 @@ class ScreenshotWorker:
                 try:
                     # Just check if the file exists and is executable
                     test_path = (
-                        Path(path)
-                        if not path.startswith("/mnt/")
-                        else Path(path)
+                        Path(path) if not path.startswith("/mnt/") else Path(path)
                     )
                     if path == "powershell.exe":
                         # In PATH - use it directly
@@ -421,18 +414,14 @@ class ScreenshotWorker:
                 print("🔄 Executing PowerShell script...")
 
             try:
-                result = subprocess.run(
-                    cmd, capture_output=True, text=True, timeout=10
-                )
+                result = subprocess.run(cmd, capture_output=True, text=True, timeout=10)
 
                 if self.verbose:
                     print(f"✓ PowerShell return code: {result.returncode}")
                     if result.stderr:
                         print(f"PowerShell stderr: {result.stderr[:500]}")
                     if result.stdout:
-                        print(
-                            f"✓ PowerShell stdout length: {len(result.stdout)} chars"
-                        )
+                        print(f"✓ PowerShell stdout length: {len(result.stdout)} chars")
             except subprocess.TimeoutExpired as e:
                 if self.verbose:
                     print(f"❌ PowerShell timeout after 10s")
@@ -455,9 +444,7 @@ class ScreenshotWorker:
                         img = Image.open(io.BytesIO(png_data))
                         # Convert RGBA to RGB for JPEG
                         if img.mode == "RGBA":
-                            rgb_img = Image.new(
-                                "RGB", img.size, (255, 255, 255)
-                            )
+                            rgb_img = Image.new("RGB", img.size, (255, 255, 255))
                             rgb_img.paste(img, mask=img.split()[3])
                             img = rgb_img
                         # Save as JPEG with quality
@@ -491,9 +478,7 @@ class ScreenshotWorker:
                 with mss.mss() as sct:
                     # Capture primary monitor
                     monitor = (
-                        sct.monitors[1]
-                        if len(sct.monitors) > 1
-                        else sct.monitors[0]
+                        sct.monitors[1] if len(sct.monitors) > 1 else sct.monitors[0]
                     )
                     screenshot = sct.grab(monitor)
 
@@ -508,9 +493,7 @@ class ScreenshotWorker:
                             "raw",
                             "BGRX",
                         )
-                        img.save(
-                            str(filepath), "JPEG", quality=self.jpeg_quality
-                        )
+                        img.save(str(filepath), "JPEG", quality=self.jpeg_quality)
                     else:
                         mss.tools.to_png(
                             screenshot.rgb,
@@ -695,9 +678,7 @@ class CaptureManager:
                 str(script_path),
             ]
 
-            result = subprocess.run(
-                cmd, capture_output=True, text=True, timeout=10
-            )
+            result = subprocess.run(cmd, capture_output=True, text=True, timeout=10)
 
             if result.returncode == 0 and result.stdout.strip():
                 # Parse JSON from output (skip non-JSON lines)
@@ -712,9 +693,7 @@ class CaptureManager:
                 return {"error": "No JSON in output"}
             else:
                 return {
-                    "error": (
-                        result.stderr if result.stderr else "Detection failed"
-                    )
+                    "error": (result.stderr if result.stderr else "Detection failed")
                 }
 
         except Exception as e:
@@ -781,9 +760,7 @@ class CaptureManager:
                 str(window_handle),
             ]
 
-            result = subprocess.run(
-                cmd, capture_output=True, text=True, timeout=10
-            )
+            result = subprocess.run(cmd, capture_output=True, text=True, timeout=10)
 
             if result.returncode == 0 and result.stdout.strip():
                 # Parse JSON from output
@@ -806,9 +783,7 @@ class CaptureManager:
                 if output_path is None:
                     timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
                     ext = "jpg" if jpeg else "png"
-                    output_path = (
-                        f"/tmp/window_{window_handle}_{timestamp}.{ext}"
-                    )
+                    output_path = f"/tmp/window_{window_handle}_{timestamp}.{ext}"
 
                 # Decode base64 image
                 img_data = base64.b64decode(data.get("Base64Data", ""))
@@ -822,18 +797,14 @@ class CaptureManager:
 
                         img = Image.open(io.BytesIO(img_data))
                         if img.mode == "RGBA":
-                            rgb_img = Image.new(
-                                "RGB", img.size, (255, 255, 255)
-                            )
+                            rgb_img = Image.new("RGB", img.size, (255, 255, 255))
                             rgb_img.paste(img, mask=img.split()[3])
                             img = rgb_img
-                        img.save(
-                            output_path, "JPEG", quality=quality, optimize=True
-                        )
+                        img.save(output_path, "JPEG", quality=quality, optimize=True)
                     except ImportError:
-                        output_path = output_path.replace(
-                            ".jpg", ".png"
-                        ).replace(".jpeg", ".png")
+                        output_path = output_path.replace(".jpg", ".png").replace(
+                            ".jpeg", ".png"
+                        )
                         with open(output_path, "wb") as f:
                             f.write(img_data)
                 else:
@@ -845,4 +816,5 @@ class CaptureManager:
         except Exception as e:
             return None
 
+
 # EOF

scitex/capture/cli.py

@@ -5,9 +5,8 @@
 # ----------------------------------------
 from __future__ import annotations
 import os
-__FILE__ = (
-    "./src/scitex/capture/cli.py"
-)
+
+__FILE__ = "./src/scitex/capture/cli.py"
 __DIR__ = os.path.dirname(__FILE__)
 # ----------------------------------------
 
@@ -41,36 +40,18 @@ Examples:
     )
 
     # Capture options
-    parser.add_argument(
-        "message", nargs="?", help="Optional message for filename"
-    )
-    parser.add_argument(
-        "--all", action="store_true", help="Capture all monitors"
-    )
-    parser.add_argument(
-        "--app", type=str, help="App name to capture (e.g., chrome)"
-    )
-    parser.add_argument(
-        "--url", type=str, help="URL to capture (e.g., 127.0.0.1:8000)"
-    )
-    parser.add_argument(
-        "--monitor", type=int, default=0, help="Monitor ID (0-based)"
-    )
-    parser.add_argument(
-        "--quality", type=int, default=85, help="JPEG quality (1-100)"
-    )
+    parser.add_argument("message", nargs="?", help="Optional message for filename")
+    parser.add_argument("--all", action="store_true", help="Capture all monitors")
+    parser.add_argument("--app", type=str, help="App name to capture (e.g., chrome)")
+    parser.add_argument("--url", type=str, help="URL to capture (e.g., 127.0.0.1:8000)")
+    parser.add_argument("--monitor", type=int, default=0, help="Monitor ID (0-based)")
+    parser.add_argument("--quality", type=int, default=85, help="JPEG quality (1-100)")
     parser.add_argument("-o", "--output", type=str, help="Output path")
 
     # Actions
-    parser.add_argument(
-        "--list", action="store_true", help="List available windows"
-    )
-    parser.add_argument(
-        "--info", action="store_true", help="Show display info"
-    )
-    parser.add_argument(
-        "--start", action="store_true", help="Start monitoring"
-    )
+    parser.add_argument("--list", action="store_true", help="List available windows")
+    parser.add_argument("--info", action="store_true", help="Show display info")
+    parser.add_argument("--start", action="store_true", help="Start monitoring")
     parser.add_argument("--stop", action="store_true", help="Stop monitoring")
     parser.add_argument(
         "--gif", action="store_true", help="Create GIF from latest session"
@@ -84,9 +65,7 @@ Examples:
         default=1.0,
         help="Monitoring interval in seconds",
     )
-    parser.add_argument(
-        "-q", "--quiet", action="store_true", help="Quiet mode"
-    )
+    parser.add_argument("-q", "--quiet", action="store_true", help="Quiet mode")
 
     args = parser.parse_args()
 
@@ -122,15 +101,11 @@ Examples:
                 bounds = mon.get("Bounds", {})
                 print(f"\n   Monitor {i}:")
                 print(f"     Device: {mon.get('DeviceName')}")
-                print(
-                    f"     Resolution: {bounds.get('Width')}x{bounds.get('Height')}"
-                )
+                print(f"     Resolution: {bounds.get('Width')}x{bounds.get('Height')}")
                 print(f"     Primary: {mon.get('IsPrimary')}")
 
             print(f"\n🪟 Windows: {windows.get('VisibleCount')}")
-            print(
-                f"   On current virtual desktop: {len(windows.get('Details', []))}"
-            )
+            print(f"   On current virtual desktop: {len(windows.get('Details', []))}")
 
             print(f"\n🖥️  Virtual Desktops:")
             print(f"   Supported: {vd.get('Supported')}")

scitex/capture/gif.py

@@ -5,9 +5,8 @@
 # ----------------------------------------
 from __future__ import annotations
 import os
-__FILE__ = (
-    "./src/scitex/capture/gif.py"
-)
+
+__FILE__ = "./src/scitex/capture/gif.py"
 __DIR__ = os.path.dirname(__FILE__)
 # ----------------------------------------
 
@@ -150,9 +149,7 @@ class GifCreator:
             target_size = images[0].size
             for i in range(1, len(images)):
                 if images[i].size != target_size:
-                    images[i] = images[i].resize(
-                        target_size, Image.Resampling.LANCZOS
-                    )
+                    images[i] = images[i].resize(target_size, Image.Resampling.LANCZOS)
 
             # Create output directory if it doesn't exist
             output_path = Path(output_path)
@@ -260,9 +257,7 @@ class GifCreator:
                 return []
 
             # Find all monitoring session files (format: SESSIONID_NNNN_timestamp.ext)
-            session_pattern = re.compile(
-                r"^(\d{8}_\d{6})_\d{4}_.*\.(jpg|png)$"
-            )
+            session_pattern = re.compile(r"^(\d{8}_\d{6})_\d{4}_.*\.(jpg|png)$")
 
             sessions = set()
             for file in screenshot_dir.iterdir():
@@ -341,4 +336,5 @@ def create_gif_from_latest_session(**kwargs) -> Optional[str]:
     creator = GifCreator()
     return creator.create_gif_from_recent_session(**kwargs)
 
+
 # EOF

scitex/capture/mcp_server.py

@@ -5,9 +5,8 @@
 # ----------------------------------------
 from __future__ import annotations
 import os
-__FILE__ = (
-    "./src/scitex/capture/mcp_server.py"
-)
+
+__FILE__ = "./src/scitex/capture/mcp_server.py"
 __DIR__ = os.path.dirname(__FILE__)
 # ----------------------------------------
 
@@ -382,11 +381,7 @@ class CaptureServer:
                 category = (
                     "stdout"
                     if "-stdout.jpg" in img_file.name
-                    else (
-                        "stderr"
-                        if "-stderr.jpg" in img_file.name
-                        else "unknown"
-                    )
+                    else ("stderr" if "-stderr.jpg" in img_file.name else "unknown")
                 )
 
                 mtime = datetime.fromtimestamp(img_file.stat().st_mtime)
@@ -614,13 +609,9 @@ class CaptureServer:
 
             # Filter by category if specified
             if category == "stdout":
-                screenshots = [
-                    s for s in screenshots if "-stdout.jpg" in s.name
-                ]
+                screenshots = [s for s in screenshots if "-stdout.jpg" in s.name]
             elif category == "stderr":
-                screenshots = [
-                    s for s in screenshots if "-stderr.jpg" in s.name
-                ]
+                screenshots = [s for s in screenshots if "-stderr.jpg" in s.name]
 
             # Sort by modification time (newest first)
             screenshots.sort(key=lambda p: p.stat().st_mtime, reverse=True)
@@ -631,9 +622,7 @@ class CaptureServer:
             # Build result
             result_list = []
             for screenshot in screenshots:
-                cat = (
-                    "stderr" if "-stderr.jpg" in screenshot.name else "stdout"
-                )
+                cat = "stderr" if "-stderr.jpg" in screenshot.name else "stdout"
                 result_list.append(
                     {
                         "filename": screenshot.name,
@@ -690,9 +679,7 @@ class CaptureServer:
                 )
 
                 # Get new cache size
-                total_size = sum(
-                    f.stat().st_size for f in cache_dir.glob("*.jpg")
-                )
+                total_size = sum(f.stat().st_size for f in cache_dir.glob("*.jpg"))
 
                 return {
                     "success": True,

scitex/capture/session.py

@@ -5,12 +5,12 @@
 # ----------------------------------------
 from __future__ import annotations
 import os
-__FILE__ = (
-    "./src/scitex/capture/session.py"
-)
+
+__FILE__ = "./src/scitex/capture/session.py"
 __DIR__ = os.path.dirname(__FILE__)
 # ----------------------------------------
 
+
 class Session:
     """Context manager for CAM session with automatic start/stop."""
 
@@ -67,4 +67,5 @@ def session(**kwargs):
     """Create a new session context manager."""
     return Session(**kwargs)
 
+
 # EOF

scitex/capture/utils.py

@@ -5,9 +5,8 @@
 # ----------------------------------------
 from __future__ import annotations
 import os
-__FILE__ = (
-    "./src/scitex/capture/utils.py"
-)
+
+__FILE__ = "./src/scitex/capture/utils.py"
 __DIR__ = os.path.dirname(__FILE__)
 # ----------------------------------------
 
@@ -149,9 +148,7 @@ def capture(
             if path is None:
                 timestamp_str = datetime.now().strftime("%Y%m%d_%H%M%S")
                 url_slug = (
-                    url.replace("://", "_")
-                    .replace("/", "_")
-                    .replace(":", "_")[:30]
+                    url.replace("://", "_").replace("/", "_").replace(":", "_")[:30]
                 )
                 path = f"~/.scitex/capture/{timestamp_str}-url-{url_slug}.jpg"
 
@@ -178,17 +175,13 @@ def capture(
                         "--disable-blink-features=AutomationControlled",
                         "--window-size=1920,1080",
                     ]
-                    browser = p.chromium.launch(
-                        headless=True, args=stealth_args
-                    )
+                    browser = p.chromium.launch(headless=True, args=stealth_args)
                     context = browser.new_context(
                         viewport={"width": url_width, "height": url_height}
                     )
                     page = context.new_page()
                     # Use domcontentloaded for faster capture, with longer timeout
-                    page.goto(
-                        url, wait_until="domcontentloaded", timeout=30000
-                    )
+                    page.goto(url, wait_until="domcontentloaded", timeout=30000)
                     # Wait additional time for rendering
                     page.wait_for_timeout(url_wait * 1000)
                     page.screenshot(
@@ -220,10 +213,7 @@ def capture(
             pass  # Try PowerShell fallback
 
         # For WSL: Fallback to Windows-side browser
-        if (
-            sys.platform == "linux"
-            and "microsoft" in os.uname().release.lower()
-        ):
+        if sys.platform == "linux" and "microsoft" in os.uname().release.lower():
             try:
                 import base64
                 import json
@@ -233,9 +223,7 @@ def capture(
                 if path is None:
                     timestamp_str = datetime.now().strftime("%Y%m%d_%H%M%S")
                     url_slug = (
-                        url.replace("://", "_")
-                        .replace("/", "_")
-                        .replace(":", "_")[:30]
+                        url.replace("://", "_").replace("/", "_").replace(":", "_")[:30]
                     )
                     path = f"~/.scitex/capture/{timestamp_str}-url-{url_slug}.jpg"
 
@@ -307,18 +295,14 @@ def capture(
 
                                             from PIL import Image
 
-                                            img = Image.open(
-                                                io.BytesIO(img_data)
-                                            )
+                                            img = Image.open(io.BytesIO(img_data))
                                             if img.mode == "RGBA":
                                                 rgb_img = Image.new(
                                                     "RGB",
                                                     img.size,
                                                     (255, 255, 255),
                                                 )
-                                                rgb_img.paste(
-                                                    img, mask=img.split()[3]
-                                                )
+                                                rgb_img.paste(img, mask=img.split()[3])
                                                 img = rgb_img
                                             img.save(
                                                 path,
@@ -373,9 +357,7 @@ def capture(
             result_path = _manager.capture_window(handle, path)
 
             if result_path and verbose:
-                print(
-                    f"📸 {matching_window.get('ProcessName')}: {result_path}"
-                )
+                print(f"📸 {matching_window.get('ProcessName')}: {result_path}")
 
             return result_path
         else:
@@ -394,9 +376,7 @@ def capture(
 
     # Take screenshot to temp location
     use_jpeg = (
-        True
-        if path is None or path.lower().endswith((".jpg", ".jpeg"))
-        else False
+        True if path is None or path.lower().endswith((".jpg", ".jpeg")) else False
     )
     worker = ScreenshotWorker(
         output_dir=temp_dir,
@@ -446,13 +426,9 @@ def capture(
         # Remove special chars, keep only alphanumeric and spaces
         import re
 
-        normalized = re.sub(
-            r"[^\w\s-]", "", message.split("\n")[0]
-        )  # First line only
+        normalized = re.sub(r"[^\w\s-]", "", message.split("\n")[0])  # First line only
         normalized = re.sub(r"[-\s]+", "-", normalized).strip("-")
-        normalized_msg = (
-            f"-{normalized[:50]}" if normalized else ""
-        )  # Limit length
+        normalized_msg = f"-{normalized[:50]}" if normalized else ""  # Limit length
 
     # Add category suffix
     category_suffix = f"-{category}"
@@ -460,9 +436,7 @@ def capture(
     # Handle path with category and message
     if path is None:
         # Include monitor/scope info in filename
-        path = (
-            f"~/.scitex/capture/<timestamp><scope><message><category_suffix>.jpg"
-        )
+        path = f"~/.scitex/capture/<timestamp><scope><message><category_suffix>.jpg"
 
     # Expand user home
     path = os.path.expanduser(path)
@@ -488,9 +462,7 @@ def capture(
     # Add message with category as metadata
     if message or category != "stdout":
         metadata = (
-            f"[{category.upper()}] {message}"
-            if message
-            else f"[{category.upper()}]"
+            f"[{category.upper()}] {message}" if message else f"[{category.upper()}]"
         )
         _add_message_metadata(str(final_path), metadata)
 
@@ -641,12 +613,8 @@ def _detect_category(filepath: str) -> str:
         # Red dominant = likely error
         # Yellow/orange dominant = likely warning
         pixels = img.convert("RGB").getdata()
-        red_count = sum(
-            1 for r, g, b in pixels if r > 200 and g < 100 and b < 100
-        )
-        yellow_count = sum(
-            1 for r, g, b in pixels if r > 200 and g > 150 and b < 100
-        )
+        red_count = sum(1 for r, g, b in pixels if r > 200 and g < 100 and b < 100)
+        yellow_count = sum(1 for r, g, b in pixels if r > 200 and g > 150 and b < 100)
 
         total_pixels = len(pixels)
         red_ratio = red_count / total_pixels if total_pixels > 0 else 0
@@ -663,10 +631,7 @@ def _detect_category(filepath: str) -> str:
 
     # Check filename for common error keywords
     filename_lower = str(filepath).lower()
-    if any(
-        word in filename_lower
-        for word in ["error", "fail", "exception", "crash"]
-    ):
+    if any(word in filename_lower for word in ["error", "fail", "exception", "crash"]):
         return "stderr"
     elif any(word in filename_lower for word in ["warn", "alert", "caution"]):
         return "stderr"  # Warnings also go to stderr

scitex/cli/__init__.py

@@ -14,4 +14,4 @@ Provides unified interface for:
 
 from .main import cli
 
-__all__ = ['cli']
+__all__ = ["cli"]