cudag 0.3.10__py3-none-any.whl

cudag/__init__.py

@@ -0,0 +1,334 @@
+# Copyright (c) 2025 Tylt LLC. All rights reserved.
+# CONFIDENTIAL AND PROPRIETARY. Unauthorized use, copying, or distribution
+# is strictly prohibited. For licensing inquiries: hello@claimhawk.app
+
+"""CUDAG - ComputerUseDataAugmentedGeneration framework.
+
+A Rails-like DSL for generating VLM training data.
+
+Example model:
+    class Patient(Model):
+        first_name = string(faker="first_name")
+        last_name = string(faker="last_name")
+        dob = date_field(min_year=1940, max_year=2010)
+
+        full_name = computed("first_name", "last_name")
+        appointments = has_many("Appointment")
+
+Example screen:
+    class CalendarScreen(Screen):
+        name = "calendar"
+        base_image = "calendar.png"
+        size = (224, 208)
+
+        day_grid = grid((10, 50, 210, 150), rows=6, cols=7)
+        back_month = button((7, 192, 20, 12), label="Back")
+
+CLI:
+    cudag new <project-name>
+    cudag generate --config config/dataset.yaml
+"""
+
+__version__ = "0.1.0"
+
+# Core classes and DSL functions
+from cudag.core import (
+    # Coordinates
+    RU_MAX,
+    bounds_to_tolerance,
+    calculate_tolerance_ru,
+    tolerance_to_ru,
+    # Model DSL - classes
+    Attachment,
+    # Renderer
+    BaseRenderer,
+    # State
+    BaseState,
+    # Task
+    BaseTask,
+    # Utils
+    check_script_invocation,
+    get_researcher_name,
+    # Generator
+    run_generator,
+    # Fonts
+    load_font,
+    load_font_family,
+    SYSTEM_FONTS,
+    # Random utilities
+    choose,
+    date_in_range,
+    amount,
+    weighted_choice,
+    # Text utilities
+    measure_text,
+    center_text_position,
+    draw_centered_text,
+    wrap_text,
+    truncate_text,
+    ordinal_suffix,
+    # Drawing utilities
+    render_scrollbar,
+    # Config utilities
+    load_yaml_config,
+    get_config_path,
+    BelongsToRel,
+    BoolField,
+    # Screen DSL - classes
+    Bounds,
+    ButtonRegion,
+    ChoiceField,
+    Claim,
+    ClickRegion,
+    ComputedField,
+    # Dataset
+    DatasetBuilder,
+    DatasetConfig,
+    # Distribution
+    DistributionSampler,
+    # Scroll Tasks
+    ScrollTaskBase,
+    ScrollTaskConfig,
+    # Grounding Tasks
+    GroundingTaskBase,
+    bbox_to_ru,
+    scale_bbox,
+    DateField,
+    DropdownRegion,
+    TestCase,
+    Field,
+    FloatField,
+    GridRegion,
+    HasManyRel,
+    HasOneRel,
+    IntField,
+    ListField,
+    Model,
+    ModelGenerator,
+    MoneyField,
+    Patient,
+    Procedure,
+    Provider,
+    Region,
+    Relationship,
+    Screen,
+    ScreenBase,
+    ScreenMeta,
+    ScrollRegion,
+    ScrollState,
+    StringField,
+    TaskContext,
+    TaskSample,
+    TimeField,
+    # Model DSL - functions
+    attribute,
+    belongs_to,
+    boolean,
+    # Screen DSL - functions
+    button,
+    choice,
+    clamp_coord,
+    computed,
+    coord_distance,
+    coord_within_tolerance,
+    date_field,
+    decimal,
+    dropdown,
+    grid,
+    has_many,
+    has_one,
+    integer,
+    list_of,
+    money,
+    normalize_coord,
+    pixel_from_normalized,
+    region,
+    scrollable,
+    string,
+    time_field,
+    years_since,
+    # Semantic field types
+    City,
+    ClaimNumber,
+    ClaimStatus,
+    DOB,
+    Email,
+    Fee,
+    FirstName,
+    FullName,
+    LastName,
+    LicenseNumber,
+    MemberID,
+    NPI,
+    Phone,
+    ProcedureCode,
+    SSN,
+    Specialty,
+    State,
+    Street,
+    ZipCode,
+)
+
+# Prompts
+from cudag.prompts import (
+    COMPUTER_USE_TOOL,
+    CUA_SYSTEM_PROMPT,
+    TOOL_ACTIONS,
+    BboxCall,
+    ToolCall,
+    format_tool_call,
+    get_system_prompt,
+    parse_tool_call,
+    validate_tool_call,
+)
+
+__all__ = [
+    # Version
+    "__version__",
+    # Coordinates
+    "RU_MAX",
+    "normalize_coord",
+    "pixel_from_normalized",
+    "clamp_coord",
+    "coord_distance",
+    "coord_within_tolerance",
+    "tolerance_to_ru",
+    "bounds_to_tolerance",
+    "calculate_tolerance_ru",
+    # Screen DSL - classes
+    "Screen",
+    "ScreenBase",
+    "ScreenMeta",
+    "Region",
+    "Bounds",
+    "ClickRegion",
+    "ButtonRegion",
+    "GridRegion",
+    "ScrollRegion",
+    "DropdownRegion",
+    # Screen DSL - functions
+    "region",
+    "button",
+    "grid",
+    "scrollable",
+    "dropdown",
+    # State
+    "BaseState",
+    "ScrollState",
+    # Renderer
+    "BaseRenderer",
+    # Task
+    "BaseTask",
+    "TaskSample",
+    "TaskContext",
+    "TestCase",
+    # Dataset
+    "DatasetBuilder",
+    "DatasetConfig",
+    # Distribution
+    "DistributionSampler",
+    # Scroll Tasks
+    "ScrollTaskBase",
+    "ScrollTaskConfig",
+    # Grounding Tasks
+    "GroundingTaskBase",
+    "bbox_to_ru",
+    "scale_bbox",
+    # Model DSL - classes
+    "Model",
+    "ModelGenerator",
+    "Field",
+    "StringField",
+    "IntField",
+    "FloatField",
+    "BoolField",
+    "DateField",
+    "TimeField",
+    "ChoiceField",
+    "ListField",
+    "MoneyField",
+    "ComputedField",
+    # Model DSL - functions
+    "string",
+    "integer",
+    "decimal",
+    "money",
+    "date_field",
+    "time_field",
+    "boolean",
+    "choice",
+    "list_of",
+    "computed",
+    "years_since",
+    # Relationship DSL - classes
+    "Relationship",
+    "HasManyRel",
+    "BelongsToRel",
+    "HasOneRel",
+    # Relationship DSL - functions
+    "has_many",
+    "belongs_to",
+    "has_one",
+    # Common healthcare models
+    "Patient",
+    "Provider",
+    "Procedure",
+    "Claim",
+    "Attachment",
+    # Semantic field types
+    "FirstName",
+    "LastName",
+    "FullName",
+    "DOB",
+    "NPI",
+    "SSN",
+    "Phone",
+    "Email",
+    "Street",
+    "City",
+    "State",
+    "ZipCode",
+    "MemberID",
+    "ClaimNumber",
+    "ProcedureCode",
+    "LicenseNumber",
+    "Specialty",
+    "ClaimStatus",
+    "Fee",
+    # Prompts
+    "COMPUTER_USE_TOOL",
+    "TOOL_ACTIONS",
+    "BboxCall",
+    "ToolCall",
+    "format_tool_call",
+    "parse_tool_call",
+    "validate_tool_call",
+    "CUA_SYSTEM_PROMPT",
+    "get_system_prompt",
+    # Utils
+    "check_script_invocation",
+    "get_researcher_name",
+    # Generator
+    "run_generator",
+    # Fonts
+    "load_font",
+    "load_font_family",
+    "SYSTEM_FONTS",
+    # Random utilities
+    "choose",
+    "date_in_range",
+    "amount",
+    "weighted_choice",
+    # Text utilities
+    "measure_text",
+    "center_text_position",
+    "draw_centered_text",
+    "wrap_text",
+    "truncate_text",
+    "ordinal_suffix",
+    # Drawing utilities
+    "render_scrollbar",
+    # Config utilities
+    "load_yaml_config",
+    "get_config_path",
+]

cudag/annotation/__init__.py

@@ -0,0 +1,77 @@
+# Copyright (c) 2025 Tylt LLC. All rights reserved.
+# CONFIDENTIAL AND PROPRIETARY. Unauthorized use, copying, or distribution
+# is strictly prohibited. For licensing inquiries: hello@claimhawk.app
+
+"""Annotation parsing and code generation for CUDAG.
+
+This module provides utilities for parsing Annotator exports and
+generating CUDAG project code from them.
+
+Example:
+    from cudag.annotation import AnnotationLoader, scaffold_generator
+
+    loader = AnnotationLoader()
+    parsed = loader.load("annotation.zip")
+
+    scaffold_generator(
+        name="my-generator",
+        annotation=parsed,
+        output_dir=Path("./projects"),
+    )
+
+For runtime data-driven generation:
+    from cudag.annotation import AnnotationConfig
+
+    config = AnnotationConfig.load(Path("assets/annotations"))
+    for icon in config.get_labeled_icons("desktop"):
+        print(f"{icon.label} at {icon.absolute_center}")
+
+For parsing grid transcriptions:
+    from cudag.annotation import parse_transcription
+
+    grid = config.get_element_by_label("patient-account")
+    if grid and grid.transcription:
+        for row in grid.transcription.rows:
+            print([cell.text for cell in row.cells])
+"""
+
+from cudag.annotation.config import (
+    AnnotatedElement,
+    AnnotatedIcon,
+    AnnotatedTask,
+    AnnotationConfig,
+)
+from cudag.annotation.loader import (
+    AnnotationLoader,
+    ParsedAnnotation,
+    ParsedElement,
+    ParsedTask,
+)
+from cudag.annotation.scaffold import scaffold_generator
+from cudag.annotation.transcription import (
+    ParsedTranscription,
+    TranscriptionCell,
+    TranscriptionRow,
+    parse_text_transcription,
+    parse_transcription,
+)
+
+__all__ = [
+    # Runtime config (for data-driven generators)
+    "AnnotationConfig",
+    "AnnotatedElement",
+    "AnnotatedIcon",
+    "AnnotatedTask",
+    # Transcription parsing (for grid data extraction)
+    "ParsedTranscription",
+    "TranscriptionCell",
+    "TranscriptionRow",
+    "parse_transcription",
+    "parse_text_transcription",
+    # Code generation (for scaffolding new generators)
+    "AnnotationLoader",
+    "ParsedAnnotation",
+    "ParsedElement",
+    "ParsedTask",
+    "scaffold_generator",
+]