captchakraken 2.0.0__py3-none-any.whl

captchakraken/__init__.py

@@ -0,0 +1,61 @@
+"""
+CaptchaKraken — OpenCV grid detection + a fine-tuned Qwen3.5-9B vision LoRA
+served on vLLM.
+
+Usage:
+    from captchakraken import CaptchaSolver
+    solver = CaptchaSolver()   # auto-starts / connects to a local vLLM server
+    actions = solver.solve("captcha.png")
+
+Model/endpoint defaults live in `captchakraken.config` and are fully
+env-overridable (VLLM_BASE_URL, CAPTCHA_LORA_ADAPTER, …); the solver itself is
+model-agnostic. The legacy v1 stack (SAM3 grounding, multi-provider planner,
+detect/segment/drag-refine) lives on the `v1-old-architecture` branch.
+"""
+
+from pathlib import Path
+
+try:  # pragma: no cover
+    from dotenv import load_dotenv
+
+    project_root = Path(__file__).resolve().parent.parent
+    load_dotenv(project_root / ".env")
+except Exception:
+    pass
+
+from .action_types import (
+    CaptchaAction,
+    ClickAction,
+    DragAction,
+    TypeAction,
+    WaitAction,
+)
+from .image_processor import ImageProcessor
+from .overlay import add_overlays_to_image
+
+# The planner (requests) and solver (torch/vllm/transformers) pull in the heavy
+# serving stack. Keep them optional so leaf modules — e.g. tool_calls.find_grid,
+# which needs only cv2 + numpy + pillow — can be imported in a minimal env (CI's
+# hermetic grid-detection test) without the full GPU dependency set installed.
+try:  # pragma: no cover - exercised only when the serving stack is installed
+    from .planner import ActionPlanner
+    from .solver import CaptchaSolver, solve_captcha
+except ModuleNotFoundError:
+    ActionPlanner = None  # type: ignore[assignment,misc]
+    CaptchaSolver = None  # type: ignore[assignment,misc]
+    solve_captcha = None  # type: ignore[assignment]
+
+__all__ = [
+    "CaptchaSolver",
+    "solve_captcha",
+    "ActionPlanner",
+    "ImageProcessor",
+    "CaptchaAction",
+    "ClickAction",
+    "DragAction",
+    "TypeAction",
+    "WaitAction",
+    "add_overlays_to_image",
+]
+
+__version__ = "2.0.0"

captchakraken/action_types.py

@@ -0,0 +1,56 @@
+from typing import List, Literal, Optional, Union, Tuple
+
+from pydantic import BaseModel, RootModel
+
+
+class BoundingBox(RootModel):
+    """
+    Strongly typed bounding box: [x1, y1, x2, y2] in percentages (0.0 to 1.0).
+    Acts like a list for convenience but ensures exactly 4 float elements.
+    """
+    root: Tuple[float, float, float, float]
+
+    def __iter__(self):
+        return iter(self.root)
+
+    def __getitem__(self, item):
+        return self.root[item]
+
+    def __len__(self):
+        return len(self.root)
+
+
+class Action(BaseModel):
+    action: str
+
+
+class ClickAction(Action):
+    action: Literal["click"]
+    target_bounding_boxes: List[BoundingBox] # List of [x1, y1, x2, y2] in percentages
+
+
+class DragAction(Action):
+    action: Literal["drag"]
+    source_bounding_box: BoundingBox = None  # [x1, y1, x2, y2] in percentages
+    target_bounding_box: BoundingBox = None  # [x1, y1, x2, y2] in percentages
+
+
+class TypeAction(Action):
+    """Type text into an input."""
+    action: Literal["type"]
+    text: str
+    target_bounding_box: BoundingBox = None  # [x1, y1, x2, y2] in percentages
+
+
+class WaitAction(Action):
+    """Wait for a specified duration."""
+    action: Literal["wait"]
+    duration_ms: int
+
+
+class DoneAction(Action):
+    """Signal that the captcha is solved or no further actions are needed."""
+    action: Literal["done"]
+
+
+CaptchaAction = Union[ClickAction, DragAction, TypeAction, WaitAction, DoneAction]