edgeflowjs 0.1.0 → 0.3.0

package/dist/pipelines/image-segmentation.js

@@ -0,0 +1,535 @@
+/**
+ * edgeFlow.js - Image Segmentation Pipeline
+ *
+ * Interactive image segmentation using SAM (Segment Anything Model).
+ * Supports point prompts and bounding box prompts.
+ */
+import { EdgeFlowTensor } from '../core/tensor.js';
+import { BasePipeline, registerPipeline } from './base.js';
+import { loadModel, loadModelFromBuffer, runInference, runInferenceNamed } from '../core/runtime.js';
+// ============================================================================
+// Default Model URLs (SlimSAM - quantized for browser)
+// ============================================================================
+const DEFAULT_SAM_MODELS = {
+    encoder: 'https://huggingface.co/Xenova/slimsam-77-uniform/resolve/main/onnx/vision_encoder_quantized.onnx',
+    decoder: 'https://huggingface.co/Xenova/slimsam-77-uniform/resolve/main/onnx/prompt_encoder_mask_decoder_quantized.onnx',
+};
+// ============================================================================
+// Image Segmentation Pipeline
+// ============================================================================
+/**
+ * ImageSegmentationPipeline - Interactive image segmentation
+ *
+ * Uses SAM-style models for point/box prompted segmentation.
+ *
+ * @example
+ * ```typescript
+ * const segmenter = createImageSegmentationPipeline();
+ *
+ * // Load models with progress callback
+ * await segmenter.loadModels((progress) => {
+ *   console.log(`Loading ${progress.model}: ${progress.progress}%`);
+ * });
+ *
+ * // Set image and segment
+ * await segmenter.setImage(imageElement);
+ * const result = await segmenter.segment({
+ *   points: [{ x: 0.5, y: 0.5, label: 1 }]
+ * });
+ * ```
+ */
+export class ImageSegmentationPipeline extends BasePipeline {
+    encoderModel = null;
+    decoderModel = null;
+    imageEmbedding = null;
+    imagePositionalEmbedding = null;
+    currentImageSize = null;
+    resizedImageSize = null;
+    inputSize = 1024; // SAM default input size
+    modelsLoaded = false;
+    // Custom model URLs
+    encoderUrl;
+    decoderUrl;
+    constructor(config) {
+        super(config);
+        this.encoderUrl = DEFAULT_SAM_MODELS.encoder;
+        this.decoderUrl = DEFAULT_SAM_MODELS.decoder;
+    }
+    /**
+     * Check if models are loaded
+     */
+    get isModelsLoaded() {
+        return this.modelsLoaded;
+    }
+    /**
+     * Set custom model URLs
+     */
+    setModelUrls(encoder, decoder) {
+        this.encoderUrl = encoder;
+        this.decoderUrl = decoder;
+    }
+    /**
+     * Load both encoder and decoder models with progress callback
+     */
+    async loadModels(onProgress) {
+        if (this.modelsLoaded)
+            return;
+        // Load encoder
+        onProgress?.({ model: 'encoder', loaded: 0, total: 100, progress: 0 });
+        const encoderData = await this.fetchModelWithProgress(this.encoderUrl, (loaded, total) => {
+            onProgress?.({
+                model: 'encoder',
+                loaded,
+                total,
+                progress: Math.round((loaded / total) * 100),
+            });
+        });
+        this.encoderModel = await loadModelFromBuffer(encoderData, {
+            runtime: 'wasm', // Uses ONNXRuntime which auto-detects WebGPU internally
+        });
+        // Load decoder
+        onProgress?.({ model: 'decoder', loaded: 0, total: 100, progress: 0 });
+        const decoderData = await this.fetchModelWithProgress(this.decoderUrl, (loaded, total) => {
+            onProgress?.({
+                model: 'decoder',
+                loaded,
+                total,
+                progress: Math.round((loaded / total) * 100),
+            });
+        });
+        this.decoderModel = await loadModelFromBuffer(decoderData, {
+            runtime: 'wasm', // Uses ONNXRuntime which auto-detects WebGPU internally
+        });
+        this.modelsLoaded = true;
+    }
+    /**
+     * Fetch model with progress tracking
+     */
+    async fetchModelWithProgress(url, onProgress) {
+        const response = await fetch(url);
+        if (!response.ok) {
+            throw new Error(`Failed to fetch model: ${response.status} ${response.statusText}`);
+        }
+        const contentLength = response.headers.get('content-length');
+        const total = contentLength ? parseInt(contentLength, 10) : 0;
+        if (!response.body) {
+            // Fallback if no streaming support
+            const buffer = await response.arrayBuffer();
+            onProgress(buffer.byteLength, buffer.byteLength);
+            return buffer;
+        }
+        const reader = response.body.getReader();
+        const chunks = [];
+        let loaded = 0;
+        while (true) {
+            const { done, value } = await reader.read();
+            if (done)
+                break;
+            chunks.push(value);
+            loaded += value.length;
+            onProgress(loaded, total || loaded);
+        }
+        // Combine chunks into ArrayBuffer
+        const buffer = new Uint8Array(loaded);
+        let offset = 0;
+        for (const chunk of chunks) {
+            buffer.set(chunk, offset);
+            offset += chunk.length;
+        }
+        return buffer.buffer;
+    }
+    /**
+     * Initialize pipeline (override to skip default model loading)
+     */
+    async initialize() {
+        if (this.isReady)
+            return;
+        // Don't call super.initialize() - we handle model loading separately
+        this.isReady = true;
+    }
+    /**
+     * Load encoder model (processes the image once)
+     */
+    async loadEncoder(modelUrl) {
+        this.encoderModel = await loadModel(modelUrl, {
+            runtime: 'wasm',
+        });
+    }
+    /**
+     * Load decoder model (processes prompts to generate masks)
+     */
+    async loadDecoder(modelUrl) {
+        this.decoderModel = await loadModel(modelUrl, {
+            runtime: 'wasm',
+        });
+    }
+    /**
+     * Set and encode the image (call once per image)
+     */
+    async setImage(image) {
+        if (!this.modelsLoaded) {
+            throw new Error('Models not loaded. Call loadModels() first.');
+        }
+        // Get image data
+        const imageData = await this.loadImage(image);
+        this.currentImageSize = {
+            width: imageData.width,
+            height: imageData.height,
+        };
+        // Preprocess image for SAM
+        const { tensor: inputTensor, resizedSize } = this.preprocessImage(imageData);
+        this.resizedImageSize = resizedSize;
+        // Run encoder
+        if (this.encoderModel) {
+            const outputs = await runInference(this.encoderModel, [inputTensor]);
+            // SlimSAM encoder outputs: [image_embeddings, image_positional_embeddings]
+            this.imageEmbedding = outputs[0];
+            this.imagePositionalEmbedding = outputs[1];
+            console.log('[SAM] Encoder outputs:', outputs.length);
+            console.log('[SAM] image_embeddings shape:', this.imageEmbedding.shape);
+            if (this.imagePositionalEmbedding) {
+                console.log('[SAM] image_positional_embeddings shape:', this.imagePositionalEmbedding.shape);
+            }
+        }
+        else {
+            throw new Error('Encoder model not loaded');
+        }
+    }
+    /**
+     * Segment the image with given prompts
+     */
+    async segment(options = {}) {
+        if (!this.imageEmbedding || !this.currentImageSize || !this.resizedImageSize) {
+            throw new Error('No image set. Call setImage() first.');
+        }
+        if (!this.decoderModel) {
+            throw new Error('Decoder model not loaded');
+        }
+        const startTime = performance.now();
+        const { points = [], boxes = [], maskThreshold = 0.0, returnAllMasks = false } = options;
+        // Prepare inputs for decoder
+        const decoderInputs = this.prepareDecoderInputs(points, boxes);
+        // Add image embeddings to inputs
+        decoderInputs.set('image_embeddings', this.imageEmbedding);
+        // Add positional embeddings (required by SlimSAM)
+        if (this.imagePositionalEmbedding) {
+            decoderInputs.set('image_positional_embeddings', this.imagePositionalEmbedding);
+        }
+        else {
+            throw new Error('image_positional_embeddings not available from encoder');
+        }
+        // Run decoder model with named inputs
+        const outputs = await runInferenceNamed(this.decoderModel, decoderInputs);
+        // SAM decoder outputs: [masks, iou_predictions]
+        const masks = outputs[0];
+        const scores = outputs[1];
+        // Post-process masks
+        const result = this.postprocessMasks(masks, scores, maskThreshold, returnAllMasks);
+        result.processingTime = performance.now() - startTime;
+        return result;
+    }
+    /**
+     * Run segmentation (implements BasePipeline interface)
+     */
+    async run(input, options) {
+        await this.setImage(input);
+        return this.segment(options);
+    }
+    /**
+     * Load image from various sources
+     */
+    async loadImage(input) {
+        // Handle different input types
+        if (typeof input === 'string') {
+            // URL or base64
+            return this.loadImageFromUrl(input);
+        }
+        else if (input instanceof HTMLImageElement) {
+            return this.imageElementToImageData(input);
+        }
+        else if (input instanceof HTMLCanvasElement) {
+            return this.canvasToImageData(input);
+        }
+        else if (input instanceof ImageData) {
+            return input;
+        }
+        else if (typeof ImageBitmap !== 'undefined' && input instanceof ImageBitmap) {
+            return this.imageBitmapToImageData(input);
+        }
+        throw new Error('Unsupported image input type');
+    }
+    /**
+     * Load image from URL
+     */
+    async loadImageFromUrl(url) {
+        return new Promise((resolve, reject) => {
+            const img = new Image();
+            img.crossOrigin = 'anonymous';
+            img.onload = () => {
+                const canvas = document.createElement('canvas');
+                canvas.width = img.width;
+                canvas.height = img.height;
+                const ctx = canvas.getContext('2d');
+                ctx.drawImage(img, 0, 0);
+                resolve(ctx.getImageData(0, 0, img.width, img.height));
+            };
+            img.onerror = reject;
+            img.src = url;
+        });
+    }
+    /**
+     * Convert HTMLImageElement to ImageData
+     */
+    imageElementToImageData(img) {
+        const canvas = document.createElement('canvas');
+        canvas.width = img.naturalWidth || img.width;
+        canvas.height = img.naturalHeight || img.height;
+        const ctx = canvas.getContext('2d');
+        ctx.drawImage(img, 0, 0);
+        return ctx.getImageData(0, 0, canvas.width, canvas.height);
+    }
+    /**
+     * Convert canvas to ImageData
+     */
+    canvasToImageData(canvas) {
+        const ctx = canvas.getContext('2d');
+        return ctx.getImageData(0, 0, canvas.width, canvas.height);
+    }
+    /**
+     * Convert ImageBitmap to ImageData
+     */
+    imageBitmapToImageData(bitmap) {
+        const canvas = document.createElement('canvas');
+        canvas.width = bitmap.width;
+        canvas.height = bitmap.height;
+        const ctx = canvas.getContext('2d');
+        ctx.drawImage(bitmap, 0, 0);
+        return ctx.getImageData(0, 0, canvas.width, canvas.height);
+    }
+    /**
+     * Preprocess image for SAM
+     */
+    preprocessImage(imageData) {
+        const { width, height } = imageData;
+        // Calculate resize dimensions (longest side = inputSize)
+        const scale = this.inputSize / Math.max(width, height);
+        const newWidth = Math.round(width * scale);
+        const newHeight = Math.round(height * scale);
+        // Create resized canvas with padding
+        const canvas = document.createElement('canvas');
+        canvas.width = this.inputSize;
+        canvas.height = this.inputSize;
+        const ctx = canvas.getContext('2d');
+        // Fill with padding color (SAM uses pixel mean)
+        ctx.fillStyle = `rgb(123.675, 116.28, 103.53)`;
+        ctx.fillRect(0, 0, this.inputSize, this.inputSize);
+        // Draw resized image (top-left aligned)
+        const tempCanvas = document.createElement('canvas');
+        tempCanvas.width = width;
+        tempCanvas.height = height;
+        const tempCtx = tempCanvas.getContext('2d');
+        tempCtx.putImageData(imageData, 0, 0);
+        ctx.drawImage(tempCanvas, 0, 0, newWidth, newHeight);
+        // Get pixel data
+        const resizedData = ctx.getImageData(0, 0, this.inputSize, this.inputSize);
+        // Convert to tensor (NCHW format, normalized with ImageNet mean/std)
+        const tensorData = new Float32Array(3 * this.inputSize * this.inputSize);
+        const mean = [123.675, 116.28, 103.53];
+        const std = [58.395, 57.12, 57.375];
+        for (let i = 0; i < this.inputSize * this.inputSize; i++) {
+            const pixelIdx = i * 4;
+            tensorData[i] = (resizedData.data[pixelIdx] - mean[0]) / std[0]; // R
+            tensorData[this.inputSize * this.inputSize + i] =
+                (resizedData.data[pixelIdx + 1] - mean[1]) / std[1]; // G
+            tensorData[2 * this.inputSize * this.inputSize + i] =
+                (resizedData.data[pixelIdx + 2] - mean[2]) / std[2]; // B
+        }
+        return {
+            tensor: new EdgeFlowTensor(tensorData, [1, 3, this.inputSize, this.inputSize], 'float32'),
+            resizedSize: { width: newWidth, height: newHeight },
+        };
+    }
+    /**
+     * Prepare decoder inputs (prompts) for SlimSAM
+     *
+     * SlimSAM prompt_encoder_mask_decoder expects these named inputs:
+     * - image_embeddings: [1, 256, 64, 64]
+     * - point_coords: [batch, num_points, 2]
+     * - point_labels: [batch, num_points]
+     * - mask_input: [batch, 1, 256, 256]
+     * - has_mask_input: [batch, 1]
+     * - orig_im_size: [2]
+     * - position_ids: [batch, num_points]
+     */
+    prepareDecoderInputs(points, boxes) {
+        const { width: resizedW, height: resizedH } = this.resizedImageSize;
+        // Scale factors for converting normalized coords to resized image coords
+        const scaleX = resizedW;
+        const scaleY = resizedH;
+        const allPoints = [];
+        const allLabels = [];
+        // Add point prompts
+        for (const point of points) {
+            allPoints.push(point.x * scaleX, point.y * scaleY);
+            allLabels.push(point.label);
+        }
+        // Add box prompts (as two corner points)
+        for (const box of boxes) {
+            // Top-left corner (label 2)
+            allPoints.push(box.x1 * scaleX, box.y1 * scaleY);
+            allLabels.push(2);
+            // Bottom-right corner (label 3)
+            allPoints.push(box.x2 * scaleX, box.y2 * scaleY);
+            allLabels.push(3);
+        }
+        // Default point if no prompts (center of image)
+        if (allPoints.length === 0) {
+            allPoints.push(resizedW / 2, resizedH / 2);
+            allLabels.push(1);
+        }
+        const numPoints = allLabels.length;
+        const inputs = new Map();
+        // input_points: [1, 1, num_points, 2] - SlimSAM format (float32)
+        inputs.set('input_points', new EdgeFlowTensor(new Float32Array(allPoints), [1, 1, numPoints, 2], 'float32'));
+        // input_labels: [1, 1, num_points] - SlimSAM format (int64)
+        inputs.set('input_labels', new EdgeFlowTensor(BigInt64Array.from(allLabels.map(l => BigInt(l))), [1, 1, numPoints], 'int64'));
+        // Note: image_embeddings and image_positional_embeddings are added in segment()
+        // SlimSAM decoder only needs: image_embeddings, image_positional_embeddings, input_points, input_labels
+        return inputs;
+    }
+    /**
+     * Post-process masks from decoder output
+     */
+    postprocessMasks(masks, scores, threshold, returnAllMasks) {
+        const { width, height } = this.currentImageSize;
+        const scoresData = scores.toFloat32Array();
+        const masksData = masks.toFloat32Array();
+        // SAM outputs multiple masks (usually 3)
+        const numMasks = scoresData.length;
+        const maskShape = masks.shape; // [1, num_masks, H, W]
+        const maskH = maskShape[2] ?? height;
+        const maskW = maskShape[3] ?? width;
+        // Find best mask by score
+        let bestIdx = 0;
+        let bestScore = scoresData[0] ?? 0;
+        for (let i = 1; i < numMasks; i++) {
+            if ((scoresData[i] ?? 0) > bestScore) {
+                bestScore = scoresData[i] ?? 0;
+                bestIdx = i;
+            }
+        }
+        // Extract and resize the best mask to original image size
+        const outputMask = this.resizeMask(masksData, bestIdx, maskW, maskH, width, height, threshold);
+        const result = {
+            mask: outputMask,
+            width,
+            height,
+            score: bestScore,
+        };
+        if (returnAllMasks && numMasks > 1) {
+            result.allMasks = [];
+            for (let m = 0; m < numMasks; m++) {
+                const mask = this.resizeMask(masksData, m, maskW, maskH, width, height, threshold);
+                result.allMasks.push({
+                    mask,
+                    score: scoresData[m] ?? 0,
+                });
+            }
+        }
+        return result;
+    }
+    /**
+     * Resize mask from model output size to original image size
+     */
+    resizeMask(masksData, maskIdx, srcW, srcH, dstW, dstH, threshold) {
+        const outputMask = new Uint8Array(dstW * dstH);
+        const maskOffset = maskIdx * srcW * srcH;
+        // Bilinear interpolation for resizing
+        for (let y = 0; y < dstH; y++) {
+            for (let x = 0; x < dstW; x++) {
+                // Map to source coordinates
+                const srcX = (x / dstW) * srcW;
+                const srcY = (y / dstH) * srcH;
+                // Bilinear interpolation
+                const x0 = Math.floor(srcX);
+                const x1 = Math.min(x0 + 1, srcW - 1);
+                const y0 = Math.floor(srcY);
+                const y1 = Math.min(y0 + 1, srcH - 1);
+                const xFrac = srcX - x0;
+                const yFrac = srcY - y0;
+                const v00 = masksData[maskOffset + y0 * srcW + x0] ?? 0;
+                const v01 = masksData[maskOffset + y0 * srcW + x1] ?? 0;
+                const v10 = masksData[maskOffset + y1 * srcW + x0] ?? 0;
+                const v11 = masksData[maskOffset + y1 * srcW + x1] ?? 0;
+                const value = v00 * (1 - xFrac) * (1 - yFrac) +
+                    v01 * xFrac * (1 - yFrac) +
+                    v10 * (1 - xFrac) * yFrac +
+                    v11 * xFrac * yFrac;
+                // Apply sigmoid and threshold
+                const sigmoid = 1 / (1 + Math.exp(-value));
+                outputMask[y * dstW + x] = sigmoid > threshold ? 255 : 0;
+            }
+        }
+        return outputMask;
+    }
+    /**
+     * Clear the current image embedding
+     */
+    clearImage() {
+        this.imageEmbedding = null;
+        this.imagePositionalEmbedding = null;
+        this.currentImageSize = null;
+        this.resizedImageSize = null;
+    }
+    /**
+     * Preprocess (required by BasePipeline)
+     */
+    async preprocess(input) {
+        const imageData = await this.loadImage(input);
+        const { tensor } = this.preprocessImage(imageData);
+        return [tensor];
+    }
+    /**
+     * Postprocess (required by BasePipeline)
+     */
+    async postprocess(_outputs, _options) {
+        // This is handled in segment() method
+        return {
+            mask: new Uint8Array(0),
+            width: 0,
+            height: 0,
+            score: 0,
+        };
+    }
+    /**
+     * Dispose resources
+     */
+    dispose() {
+        super.dispose();
+        this.encoderModel?.dispose();
+        this.decoderModel?.dispose();
+        this.imageEmbedding = null;
+        this.imagePositionalEmbedding = null;
+        this.currentImageSize = null;
+        this.resizedImageSize = null;
+        this.modelsLoaded = false;
+    }
+}
+// ============================================================================
+// Factory Function
+// ============================================================================
+/**
+ * Create image segmentation pipeline
+ */
+export function createImageSegmentationPipeline(config = {}) {
+    return new ImageSegmentationPipeline({
+        task: 'image-segmentation',
+        model: config.model ?? 'slimsam',
+        runtime: config.runtime,
+        cache: config.cache ?? true,
+        quantization: config.quantization,
+    });
+}
+// Register pipeline
+registerPipeline('image-segmentation', (config) => new ImageSegmentationPipeline(config));
+//# sourceMappingURL=image-segmentation.js.map

package/dist/pipelines/image-segmentation.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"image-segmentation.js","sourceRoot":"","sources":["../../src/pipelines/image-segmentation.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAOH,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AACnD,OAAO,EAAE,YAAY,EAAkB,gBAAgB,EAAE,MAAM,WAAW,CAAC;AAC3E,OAAO,EAAE,SAAS,EAAE,mBAAmB,EAAE,YAAY,EAAE,iBAAiB,EAAE,MAAM,oBAAoB,CAAC;AAErG,+EAA+E;AAC/E,uDAAuD;AACvD,+EAA+E;AAE/E,MAAM,kBAAkB,GAAG;IACzB,OAAO,EAAE,kGAAkG;IAC3G,OAAO,EAAE,+GAA+G;CACzH,CAAC;AAsFF,+EAA+E;AAC/E,8BAA8B;AAC9B,+EAA+E;AAE/E;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,MAAM,OAAO,yBAA0B,SAAQ,YAG9C;IACS,YAAY,GAAuB,IAAI,CAAC;IACxC,YAAY,GAAuB,IAAI,CAAC;IACxC,cAAc,GAA0B,IAAI,CAAC;IAC7C,wBAAwB,GAA0B,IAAI,CAAC;IACvD,gBAAgB,GAA6C,IAAI,CAAC;IAClE,gBAAgB,GAA6C,IAAI,CAAC;IAClE,SAAS,GAAW,IAAI,CAAC,CAAC,yBAAyB;IACnD,YAAY,GAAY,KAAK,CAAC;IAEtC,oBAAoB;IACZ,UAAU,CAAS;IACnB,UAAU,CAAS;IAE3B,YAAY,MAAsB;QAChC,KAAK,CAAC,MAAM,CAAC,CAAC;QACd,IAAI,CAAC,UAAU,GAAG,kBAAkB,CAAC,OAAO,CAAC;QAC7C,IAAI,CAAC,UAAU,GAAG,kBAAkB,CAAC,OAAO,CAAC;IAC/C,CAAC;IAED;;OAEG;IACH,IAAI,cAAc;QAChB,OAAO,IAAI,CAAC,YAAY,CAAC;IAC3B,CAAC;IAED;;OAEG;IACH,YAAY,CAAC,OAAe,EAAE,OAAe;QAC3C,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC;QAC1B,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC;IAC5B,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU,CACd,UAAkD;QAElD,IAAI,IAAI,CAAC,YAAY;YAAE,OAAO;QAE9B,eAAe;QACf,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;QAEvE,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,sBAAsB,CACnD,IAAI,CAAC,UAAU,EACf,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE;YAChB,UAAU,EAAE,CAAC;gBACX,KAAK,EAAE,SAAS;gBAChB,MAAM;gBACN,KAAK;gBACL,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,GAAG,KAAK,CAAC,GAAG,GAAG,CAAC;aAC7C,CAAC,CAAC;QACL,CAAC,CACF,CAAC;QAEF,IAAI,CAAC,YAAY,GAAG,MAAM,mBAAmB,CAAC,WAAW,EAAE;YACzD,OAAO,EAAE,MAAM,EAAE,wDAAwD;SAC1E,CAAC,CAAC;QAEH,eAAe;QACf,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;QAEvE,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,sBAAsB,CACnD,IAAI,CAAC,UAAU,EACf,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE;YAChB,UAAU,EAAE,CAAC;gBACX,KAAK,EAAE,SAAS;gBAChB,MAAM;gBACN,KAAK;gBACL,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,GAAG,KAAK,CAAC,GAAG,GAAG,CAAC;aAC7C,CAAC,CAAC;QACL,CAAC,CACF,CAAC;QAEF,IAAI,CAAC,YAAY,GAAG,MAAM,mBAAmB,CAAC,WAAW,EAAE;YACzD,OAAO,EAAE,MAAM,EAAE,wDAAwD;SAC1E,CAAC,CAAC;QAEH,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC;IAC3B,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,sBAAsB,CAClC,GAAW,EACX,UAAmD;QAEnD,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,CAAC;QAElC,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,0BAA0B,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;QACtF,CAAC;QAED,MAAM,aAAa,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC;QAC7D,MAAM,KAAK,GAAG,aAAa,CAAC,CAAC,CAAC,QAAQ,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAE9D,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;YACnB,mCAAmC;YACnC,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,WAAW,EAAE,CAAC;YAC5C,UAAU,CAAC,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,UAAU,CAAC,CAAC;YACjD,OAAO,MAAM,CAAC;QAChB,CAAC;QAED,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;QACzC,MAAM,MAAM,GAAiB,EAAE,CAAC;QAChC,IAAI,MAAM,GAAG,CAAC,CAAC;QAEf,OAAO,IAAI,EAAE,CAAC;YACZ,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;YAE5C,IAAI,IAAI;gBAAE,MAAM;YAEhB,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YACnB,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC;YACvB,UAAU,CAAC,MAAM,EAAE,KAAK,IAAI,MAAM,CAAC,CAAC;QACtC,CAAC;QAED,kCAAkC;QAClC,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,MAAM,CAAC,CAAC;QACtC,IAAI,MAAM,GAAG,CAAC,CAAC;QACf,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YAC1B,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC;QACzB,CAAC;QAED,OAAO,MAAM,CAAC,MAAM,CAAC;IACvB,CAAC;IAED;;OAEG;IACM,KAAK,CAAC,UAAU;QACvB,IAAI,IAAI,CAAC,OAAO;YAAE,OAAO;QACzB,qEAAqE;QACrE,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;IACtB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,WAAW,CAAC,QAAgB;QAChC,IAAI,CAAC,YAAY,GAAG,MAAM,SAAS,CAAC,QAAQ,EAAE;YAC5C,OAAO,EAAE,MAAM;SAChB,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,WAAW,CAAC,QAAgB;QAChC,IAAI,CAAC,YAAY,GAAG,MAAM,SAAS,CAAC,QAAQ,EAAE;YAC5C,OAAO,EAAE,MAAM;SAChB,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,QAAQ,CAAC,KAAiB;QAC9B,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;YACvB,MAAM,IAAI,KAAK,CAAC,6CAA6C,CAAC,CAAC;QACjE,CAAC;QAED,iBAAiB;QACjB,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAC9C,IAAI,CAAC,gBAAgB,GAAG;YACtB,KAAK,EAAE,SAAS,CAAC,KAAK;YACtB,MAAM,EAAE,SAAS,CAAC,MAAM;SACzB,CAAC;QAEF,2BAA2B;QAC3B,MAAM,EAAE,MAAM,EAAE,WAAW,EAAE,WAAW,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;QAC7E,IAAI,CAAC,gBAAgB,GAAG,WAAW,CAAC;QAEpC,cAAc;QACd,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACtB,MAAM,OAAO,GAAG,MAAM,YAAY,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC,WAAW,CAAC,CAAC,CAAC;YACrE,2EAA2E;YAC3E,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC,CAAC,CAAmB,CAAC;YACnD,IAAI,CAAC,wBAAwB,GAAG,OAAO,CAAC,CAAC,CAAmB,CAAC;YAC7D,OAAO,CAAC,GAAG,CAAC,wBAAwB,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;YACtD,OAAO,CAAC,GAAG,CAAC,+BAA+B,EAAE,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;YACxE,IAAI,IAAI,CAAC,wBAAwB,EAAE,CAAC;gBAClC,OAAO,CAAC,GAAG,CAAC,0CAA0C,EAAE,IAAI,CAAC,wBAAwB,CAAC,KAAK,CAAC,CAAC;YAC/F,CAAC;QACH,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO,CAAC,UAAoC,EAAE;QAClD,IAAI,CAAC,IAAI,CAAC,cAAc,IAAI,CAAC,IAAI,CAAC,gBAAgB,IAAI,CAAC,IAAI,CAAC,gBAAgB,EAAE,CAAC;YAC7E,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;QAC1D,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC;YACvB,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;QAC9C,CAAC;QAED,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QACpC,MAAM,EAAE,MAAM,GAAG,EAAE,EAAE,KAAK,GAAG,EAAE,EAAE,aAAa,GAAG,GAAG,EAAE,cAAc,GAAG,KAAK,EAAE,GAAG,OAAO,CAAC;QAEzF,6BAA6B;QAC7B,MAAM,aAAa,GAAG,IAAI,CAAC,oBAAoB,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QAE/D,iCAAiC;QACjC,aAAa,CAAC,GAAG,CAAC,kBAAkB,EAAE,IAAI,CAAC,cAAe,CAAC,CAAC;QAE5D,kDAAkD;QAClD,IAAI,IAAI,CAAC,wBAAwB,EAAE,CAAC;YAClC,aAAa,CAAC,GAAG,CAAC,6BAA6B,EAAE,IAAI,CAAC,wBAAwB,CAAC,CAAC;QAClF,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,KAAK,CAAC,wDAAwD,CAAC,CAAC;QAC5E,CAAC;QAED,sCAAsC;QACtC,MAAM,OAAO,GAAG,MAAM,iBAAiB,CAAC,IAAI,CAAC,YAAY,EAAE,aAAa,CAAC,CAAC;QAE1E,gDAAgD;QAChD,MAAM,KAAK,GAAG,OAAO,CAAC,CAAC,CAAmB,CAAC;QAC3C,MAAM,MAAM,GAAG,OAAO,CAAC,CAAC,CAAmB,CAAC;QAE5C,qBAAqB;QACrB,MAAM,MAAM,GAAG,IAAI,CAAC,gBAAgB,CAAC,KAAK,EAAE,MAAM,EAAE,aAAa,EAAE,cAAc,CAAC,CAAC;QACnF,MAAM,CAAC,cAAc,GAAG,WAAW,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;QAEtD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACM,KAAK,CAAC,GAAG,CAChB,KAAiB,EACjB,OAAkC;QAElC,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QAC3B,OAAO,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IAC/B,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,SAAS,CAAC,KAAiB;QACvC,+BAA+B;QAC/B,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YAC9B,gBAAgB;YAChB,OAAO,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,CAAC;QACtC,CAAC;aAAM,IAAI,KAAK,YAAY,gBAAgB,EAAE,CAAC;YAC7C,OAAO,IAAI,CAAC,uBAAuB,CAAC,KAAK,CAAC,CAAC;QAC7C,CAAC;aAAM,IAAI,KAAK,YAAY,iBAAiB,EAAE,CAAC;YAC9C,OAAO,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;QACvC,CAAC;aAAM,IAAI,KAAK,YAAY,SAAS,EAAE,CAAC;YACtC,OAAO,KAAK,CAAC;QACf,CAAC;aAAM,IAAI,OAAO,WAAW,KAAK,WAAW,IAAI,KAAK,YAAY,WAAW,EAAE,CAAC;YAC9E,OAAO,IAAI,CAAC,sBAAsB,CAAC,KAAK,CAAC,CAAC;QAC5C,CAAC;QACD,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;IAClD,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,gBAAgB,CAAC,GAAW;QACxC,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACrC,MAAM,GAAG,GAAG,IAAI,KAAK,EAAE,CAAC;YACxB,GAAG,CAAC,WAAW,GAAG,WAAW,CAAC;YAC9B,GAAG,CAAC,MAAM,GAAG,GAAG,EAAE;gBAChB,MAAM,MAAM,GAAG,QAAQ,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;gBAChD,MAAM,CAAC,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC;gBACzB,MAAM,CAAC,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC;gBAC3B,MAAM,GAAG,GAAG,MAAM,CAAC,UAAU,CAAC,IAAI,CAAE,CAAC;gBACrC,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;gBACzB,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC,EAAE,CAAC,EAAE,GAAG,CAAC,KAAK,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;YACzD,CAAC,CAAC;YACF,GAAG,CAAC,OAAO,GAAG,MAAM,CAAC;YACrB,GAAG,CAAC,GAAG,GAAG,GAAG,CAAC;QAChB,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,uBAAuB,CAAC,GAAqB;QACnD,MAAM,MAAM,GAAG,QAAQ,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QAChD,MAAM,CAAC,KAAK,GAAG,GAAG,CAAC,YAAY,IAAI,GAAG,CAAC,KAAK,CAAC;QAC7C,MAAM,CAAC,MAAM,GAAG,GAAG,CAAC,aAAa,IAAI,GAAG,CAAC,MAAM,CAAC;QAChD,MAAM,GAAG,GAAG,MAAM,CAAC,UAAU,CAAC,IAAI,CAAE,CAAC;QACrC,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QACzB,OAAO,GAAG,CAAC,YAAY,CAAC,CAAC,EAAE,CAAC,EAAE,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;IAC7D,CAAC;IAED;;OAEG;IACK,iBAAiB,CAAC,MAAyB;QACjD,MAAM,GAAG,GAAG,MAAM,CAAC,UAAU,CAAC,IAAI,CAAE,CAAC;QACrC,OAAO,GAAG,CAAC,YAAY,CAAC,CAAC,EAAE,CAAC,EAAE,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;IAC7D,CAAC;IAED;;OAEG;IACK,sBAAsB,CAAC,MAAmB;QAChD,MAAM,MAAM,GAAG,QAAQ,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QAChD,MAAM,CAAC,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;QAC5B,MAAM,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC9B,MAAM,GAAG,GAAG,MAAM,CAAC,UAAU,CAAC,IAAI,CAAE,CAAC;QACrC,GAAG,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QAC5B,OAAO,GAAG,CAAC,YAAY,CAAC,CAAC,EAAE,CAAC,EAAE,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;IAC7D,CAAC;IAED;;OAEG;IACK,eAAe,CAAC,SAAoB;QAI1C,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,SAAS,CAAC;QAEpC,yDAAyD;QACzD,MAAM,KAAK,GAAG,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACvD,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC,CAAC;QAC3C,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,KAAK,CAAC,CAAC;QAE7C,qCAAqC;QACrC,MAAM,MAAM,GAAG,QAAQ,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QAChD,MAAM,CAAC,KAAK,GAAG,IAAI,CAAC,SAAS,CAAC;QAC9B,MAAM,CAAC,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC;QAC/B,MAAM,GAAG,GAAG,MAAM,CAAC,UAAU,CAAC,IAAI,CAAE,CAAC;QAErC,gDAAgD;QAChD,GAAG,CAAC,SAAS,GAAG,8BAA8B,CAAC;QAC/C,GAAG,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;QAEnD,wCAAwC;QACxC,MAAM,UAAU,GAAG,QAAQ,CAAC,aAAa,CAAC,QAAQ,CAAC,CAAC;QACpD,UAAU,CAAC,KAAK,GAAG,KAAK,CAAC;QACzB,UAAU,CAAC,MAAM,GAAG,MAAM,CAAC;QAC3B,MAAM,OAAO,GAAG,UAAU,CAAC,UAAU,CAAC,IAAI,CAAE,CAAC;QAC7C,OAAO,CAAC,YAAY,CAAC,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;QACtC,GAAG,CAAC,SAAS,CAAC,UAAU,EAAE,CAAC,EAAE,CAAC,EAAE,QAAQ,EAAE,SAAS,CAAC,CAAC;QAErD,iBAAiB;QACjB,MAAM,WAAW,GAAG,GAAG,CAAC,YAAY,CAAC,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;QAE3E,qEAAqE;QACrE,MAAM,UAAU,GAAG,IAAI,YAAY,CAAC,CAAC,GAAG,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC;QACzE,MAAM,IAAI,GAAG,CAAC,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;QACvC,MAAM,GAAG,GAAG,CAAC,MAAM,EAAE,KAAK,EAAE,MAAM,CAAC,CAAC;QAEpC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,EAAE,CAAC,EAAE,EAAE,CAAC;YACzD,MAAM,QAAQ,GAAG,CAAC,GAAG,CAAC,CAAC;YACvB,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAE,GAAG,IAAI,CAAC,CAAC,CAAE,CAAC,GAAG,GAAG,CAAC,CAAC,CAAE,CAAC,CAAC,IAAI;YACxE,UAAU,CAAC,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,GAAG,CAAC,CAAC;gBAC7C,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,GAAG,CAAC,CAAE,GAAG,IAAI,CAAC,CAAC,CAAE,CAAC,GAAG,GAAG,CAAC,CAAC,CAAE,CAAC,CAAC,IAAI;YAC9D,UAAU,CAAC,CAAC,GAAG,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,SAAS,GAAG,CAAC,CAAC;gBACjD,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,GAAG,CAAC,CAAE,GAAG,IAAI,CAAC,CAAC,CAAE,CAAC,GAAG,GAAG,CAAC,CAAC,CAAE,CAAC,CAAC,IAAI;QAChE,CAAC;QAED,OAAO;YACL,MAAM,EAAE,IAAI,cAAc,CAAC,UAAU,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,SAAS,CAAC,EAAE,SAAS,CAAC;YACzF,WAAW,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,EAAE,SAAS,EAAE;SACpD,CAAC;IACJ,CAAC;IAED;;;;;;;;;;;OAWG;IACK,oBAAoB,CAC1B,MAAqB,EACrB,KAAkB;QAElB,MAAM,EAAE,KAAK,EAAE,QAAQ,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC,gBAAiB,CAAC;QAErE,yEAAyE;QACzE,MAAM,MAAM,GAAG,QAAQ,CAAC;QACxB,MAAM,MAAM,GAAG,QAAQ,CAAC;QAExB,MAAM,SAAS,GAAa,EAAE,CAAC;QAC/B,MAAM,SAAS,GAAa,EAAE,CAAC;QAE/B,oBAAoB;QACpB,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,SAAS,CAAC,IAAI,CACZ,KAAK,CAAC,CAAC,GAAG,MAAM,EAChB,KAAK,CAAC,CAAC,GAAG,MAAM,CACjB,CAAC;YACF,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAC9B,CAAC;QAED,yCAAyC;QACzC,KAAK,MAAM,GAAG,IAAI,KAAK,EAAE,CAAC;YACxB,4BAA4B;YAC5B,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,GAAG,MAAM,EAAE,GAAG,CAAC,EAAE,GAAG,MAAM,CAAC,CAAC;YACjD,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAClB,gCAAgC;YAChC,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,GAAG,MAAM,EAAE,GAAG,CAAC,EAAE,GAAG,MAAM,CAAC,CAAC;YACjD,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACpB,CAAC;QAED,gDAAgD;QAChD,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC3B,SAAS,CAAC,IAAI,CAAC,QAAQ,GAAG,CAAC,EAAE,QAAQ,GAAG,CAAC,CAAC,CAAC;YAC3C,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACpB,CAAC;QAED,MAAM,SAAS,GAAG,SAAS,CAAC,MAAM,CAAC;QAEnC,MAAM,MAAM,GAAG,IAAI,GAAG,EAA0B,CAAC;QAEjD,iEAAiE;QACjE,MAAM,CAAC,GAAG,CAAC,cAAc,EAAE,IAAI,cAAc,CAC3C,IAAI,YAAY,CAAC,SAAS,CAAC,EAC3B,CAAC,CAAC,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC,EACpB,SAAS,CACV,CAAC,CAAC;QAEH,4DAA4D;QAC5D,MAAM,CAAC,GAAG,CAAC,cAAc,EAAE,IAAI,cAAc,CAC3C,aAAa,CAAC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,EACjD,CAAC,CAAC,EAAE,CAAC,EAAE,SAAS,CAAC,EACjB,OAAO,CACR,CAAC,CAAC;QAEH,gFAAgF;QAChF,wGAAwG;QAExG,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACK,gBAAgB,CACtB,KAAqB,EACrB,MAAsB,EACtB,SAAiB,EACjB,cAAuB;QAEvB,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC,gBAAiB,CAAC;QACjD,MAAM,UAAU,GAAG,MAAM,CAAC,cAAc,EAAE,CAAC;QAC3C,MAAM,SAAS,GAAG,KAAK,CAAC,cAAc,EAAE,CAAC;QAEzC,yCAAyC;QACzC,MAAM,QAAQ,GAAG,UAAU,CAAC,MAAM,CAAC;QACnC,MAAM,SAAS,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,uBAAuB;QACtD,MAAM,KAAK,GAAG,SAAS,CAAC,CAAC,CAAC,IAAI,MAAM,CAAC;QACrC,MAAM,KAAK,GAAG,SAAS,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC;QAEpC,0BAA0B;QAC1B,IAAI,OAAO,GAAG,CAAC,CAAC;QAChB,IAAI,SAAS,GAAG,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;QAEnC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,EAAE,CAAC,EAAE,EAAE,CAAC;YAClC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,SAAS,EAAE,CAAC;gBACrC,SAAS,GAAG,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;gBAC/B,OAAO,GAAG,CAAC,CAAC;YACd,CAAC;QACH,CAAC;QAED,0DAA0D;QAC1D,MAAM,UAAU,GAAG,IAAI,CAAC,UAAU,CAChC,SAAS,EACT,OAAO,EACP,KAAK,EACL,KAAK,EACL,KAAK,EACL,MAAM,EACN,SAAS,CACV,CAAC;QAEF,MAAM,MAAM,GAA4B;YACtC,IAAI,EAAE,UAAU;YAChB,KAAK;YACL,MAAM;YACN,KAAK,EAAE,SAAS;SACjB,CAAC;QAEF,IAAI,cAAc,IAAI,QAAQ,GAAG,CAAC,EAAE,CAAC;YACnC,MAAM,CAAC,QAAQ,GAAG,EAAE,CAAC;YACrB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,EAAE,CAAC,EAAE,EAAE,CAAC;gBAClC,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAC1B,SAAS,EACT,CAAC,EACD,KAAK,EACL,KAAK,EACL,KAAK,EACL,MAAM,EACN,SAAS,CACV,CAAC;gBACF,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC;oBACnB,IAAI;oBACJ,KAAK,EAAE,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC;iBAC1B,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACK,UAAU,CAChB,SAAuB,EACvB,OAAe,EACf,IAAY,EACZ,IAAY,EACZ,IAAY,EACZ,IAAY,EACZ,SAAiB;QAEjB,MAAM,UAAU,GAAG,IAAI,UAAU,CAAC,IAAI,GAAG,IAAI,CAAC,CAAC;QAC/C,MAAM,UAAU,GAAG,OAAO,GAAG,IAAI,GAAG,IAAI,CAAC;QAEzC,sCAAsC;QACtC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;YAC9B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC9B,4BAA4B;gBAC5B,MAAM,IAAI,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,IAAI,CAAC;gBAC/B,MAAM,IAAI,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,IAAI,CAAC;gBAE/B,yBAAyB;gBACzB,MAAM,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAC5B,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,EAAE,GAAG,CAAC,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC;gBACtC,MAAM,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBAC5B,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,EAAE,GAAG,CAAC,EAAE,IAAI,GAAG,CAAC,CAAC,CAAC;gBAEtC,MAAM,KAAK,GAAG,IAAI,GAAG,EAAE,CAAC;gBACxB,MAAM,KAAK,GAAG,IAAI,GAAG,EAAE,CAAC;gBAExB,MAAM,GAAG,GAAG,SAAS,CAAC,UAAU,GAAG,EAAE,GAAG,IAAI,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;gBACxD,MAAM,GAAG,GAAG,SAAS,CAAC,UAAU,GAAG,EAAE,GAAG,IAAI,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;gBACxD,MAAM,GAAG,GAAG,SAAS,CAAC,UAAU,GAAG,EAAE,GAAG,IAAI,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;gBACxD,MAAM,GAAG,GAAG,SAAS,CAAC,UAAU,GAAG,EAAE,GAAG,IAAI,GAAG,EAAE,CAAC,IAAI,CAAC,CAAC;gBAExD,MAAM,KAAK,GACT,GAAG,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC;oBAC/B,GAAG,GAAG,KAAK,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC;oBACzB,GAAG,GAAG,CAAC,CAAC,GAAG,KAAK,CAAC,GAAG,KAAK;oBACzB,GAAG,GAAG,KAAK,GAAG,KAAK,CAAC;gBAEtB,8BAA8B;gBAC9B,MAAM,OAAO,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;gBAC3C,UAAU,CAAC,CAAC,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,OAAO,GAAG,SAAS,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YAC3D,CAAC;QACH,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAED;;OAEG;IACH,UAAU;QACR,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC3B,IAAI,CAAC,wBAAwB,GAAG,IAAI,CAAC;QACrC,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;QAC7B,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;IAC/B,CAAC;IAED;;OAEG;IACgB,KAAK,CAAC,UAAU,CAAC,KAAiB;QACnD,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QAC9C,MAAM,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;QACnD,OAAO,CAAC,MAAM,CAAC,CAAC;IAClB,CAAC;IAED;;OAEG;IACgB,KAAK,CAAC,WAAW,CAClC,QAA0B,EAC1B,QAA0B;QAE1B,sCAAsC;QACtC,OAAO;YACL,IAAI,EAAE,IAAI,UAAU,CAAC,CAAC,CAAC;YACvB,KAAK,EAAE,CAAC;YACR,MAAM,EAAE,CAAC;YACT,KAAK,EAAE,CAAC;SACT,CAAC;IACJ,CAAC;IAED;;OAEG;IACM,OAAO;QACd,KAAK,CAAC,OAAO,EAAE,CAAC;QAChB,IAAI,CAAC,YAAY,EAAE,OAAO,EAAE,CAAC;QAC7B,IAAI,CAAC,YAAY,EAAE,OAAO,EAAE,CAAC;QAC7B,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC3B,IAAI,CAAC,wBAAwB,GAAG,IAAI,CAAC;QACrC,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;QAC7B,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;QAC7B,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;IAC5B,CAAC;CACF;AAED,+EAA+E;AAC/E,mBAAmB;AACnB,+EAA+E;AAE/E;;GAEG;AACH,MAAM,UAAU,+BAA+B,CAC7C,SAAkC,EAAE;IAEpC,OAAO,IAAI,yBAAyB,CAAC;QACnC,IAAI,EAAE,oBAAoB;QAC1B,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,SAAS;QAChC,OAAO,EAAE,MAAM,CAAC,OAAO;QACvB,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,IAAI;QAC3B,YAAY,EAAE,MAAM,CAAC,YAAY;KAClC,CAAC,CAAC;AACL,CAAC;AAED,oBAAoB;AACpB,gBAAgB,CAAC,oBAAoB,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,IAAI,yBAAyB,CAAC,MAAM,CAAC,CAAC,CAAC"}

package/dist/pipelines/index.d.ts

@@ -6,6 +6,12 @@ export { BasePipeline, registerPipeline, getPipelineFactory, SENTIMENT_LABELS, E
 export { TextClassificationPipeline, SentimentAnalysisPipeline, createTextClassificationPipeline, createSentimentAnalysisPipeline, type TextClassificationOptions, } from './text-classification.js';
 export { FeatureExtractionPipeline, createFeatureExtractionPipeline, type FeatureExtractionOptions, } from './feature-extraction.js';
 export { ImageClassificationPipeline, createImageClassificationPipeline, type ImageClassificationOptions, type ImageInput, } from './image-classification.js';
+export { TextGenerationPipeline, createTextGenerationPipeline, type TextGenerationOptions, type TextGenerationResult, type GenerationStreamEvent, type ChatMessage, type ChatOptions, type ChatTemplateType, type LLMLoadProgress, } from './text-generation.js';
+export { ObjectDetectionPipeline, createObjectDetectionPipeline, COCO_LABELS, type ObjectDetectionOptions, type Detection, type BoundingBox, } from './object-detection.js';
+export { AutomaticSpeechRecognitionPipeline, createASRPipeline, type ASROptions, type ASRResult, type WordTimestamp, type ChunkTimestamp, } from './automatic-speech-recognition.js';
+export { ZeroShotClassificationPipeline, createZeroShotClassificationPipeline, type ZeroShotClassificationOptions, type ZeroShotClassificationResult, } from './zero-shot-classification.js';
+export { QuestionAnsweringPipeline, createQuestionAnsweringPipeline, type QuestionAnsweringOptions, type QuestionAnsweringResult, type QAInput, } from './question-answering.js';
+export { ImageSegmentationPipeline, createImageSegmentationPipeline, type ImageSegmentationOptions, type ImageSegmentationResult, type PointPrompt, type BoxPrompt, type ModelLoadProgress, } from './image-segmentation.js';
 /**
  * Pipeline options for the factory function
  */
@@ -29,10 +35,22 @@ type PipelineTaskMap = {
     'sentiment-analysis': SentimentAnalysisPipeline;
     'feature-extraction': FeatureExtractionPipeline;
     'image-classification': ImageClassificationPipeline;
+    'text-generation': TextGenerationPipeline;
+    'object-detection': ObjectDetectionPipeline;
+    'automatic-speech-recognition': AutomaticSpeechRecognitionPipeline;
+    'zero-shot-classification': ZeroShotClassificationPipeline;
+    'question-answering': QuestionAnsweringPipeline;
+    'image-segmentation': ImageSegmentationPipeline;
 };
 import { TextClassificationPipeline, SentimentAnalysisPipeline } from './text-classification.js';
 import { FeatureExtractionPipeline } from './feature-extraction.js';
 import { ImageClassificationPipeline } from './image-classification.js';
+import { TextGenerationPipeline } from './text-generation.js';
+import { ObjectDetectionPipeline } from './object-detection.js';
+import { AutomaticSpeechRecognitionPipeline } from './automatic-speech-recognition.js';
+import { ZeroShotClassificationPipeline } from './zero-shot-classification.js';
+import { QuestionAnsweringPipeline } from './question-answering.js';
+import { ImageSegmentationPipeline } from './image-segmentation.js';
 /**
  * Create a pipeline for a specific task
  *

package/dist/pipelines/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/pipelines/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAGL,WAAW,EACX,gBAAgB,EACjB,MAAM,kBAAkB,CAAC;AAG1B,OAAO,EACL,YAAY,EACZ,gBAAgB,EAChB,kBAAkB,EAClB,gBAAgB,EAChB,cAAc,EACd,eAAe,EACf,KAAK,cAAc,EACnB,KAAK,wBAAwB,EAC7B,KAAK,uBAAuB,EAC5B,KAAK,yBAAyB,EAC9B,KAAK,qBAAqB,GAC3B,MAAM,WAAW,CAAC;AAGnB,OAAO,EACL,0BAA0B,EAC1B,yBAAyB,EACzB,gCAAgC,EAChC,+BAA+B,EAC/B,KAAK,yBAAyB,GAC/B,MAAM,0BAA0B,CAAC;AAGlC,OAAO,EACL,yBAAyB,EACzB,+BAA+B,EAC/B,KAAK,wBAAwB,GAC9B,MAAM,yBAAyB,CAAC;AAGjC,OAAO,EACL,2BAA2B,EAC3B,iCAAiC,EACjC,KAAK,0BAA0B,EAC/B,KAAK,UAAU,GAChB,MAAM,2BAA2B,CAAC;AAMnC;;GAEG;AACH,MAAM,WAAW,sBAAsB;IACrC,sBAAsB;IACtB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,qBAAqB;IACrB,OAAO,CAAC,EAAE,WAAW,CAAC;IACtB,qBAAqB;IACrB,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB,wBAAwB;IACxB,YAAY,CAAC,EAAE,gBAAgB,CAAC;IAChC,uCAAuC;IACvC,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;CACnB;AAED;;GAEG;AACH,KAAK,eAAe,GAAG;IACrB,qBAAqB,EAAE,0BAA0B,CAAC;IAClD,oBAAoB,EAAE,yBAAyB,CAAC;IAChD,oBAAoB,EAAE,yBAAyB,CAAC;IAChD,sBAAsB,EAAE,2BAA2B,CAAC;CACrD,CAAC;AAGF,OAAO,EAAE,0BAA0B,EAAE,yBAAyB,EAAE,MAAM,0BAA0B,CAAC;AACjG,OAAO,EAAE,yBAAyB,EAAE,MAAM,yBAAyB,CAAC;AACpE,OAAO,EAAE,2BAA2B,EAAE,MAAM,2BAA2B,CAAC;AAExE;;;;;;;;;;;;;;GAcG;AACH,wBAAsB,QAAQ,CAAC,CAAC,SAAS,MAAM,eAAe,EAC5D,IAAI,EAAE,CAAC,EACP,OAAO,CAAC,EAAE,sBAAsB,GAC/B,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAgC7B;AAED;;GAEG;AACH,wBAAsB,eAAe,CAAC,CAAC,SAAS,CAAC,MAAM,eAAe,CAAC,EAAE,EACvE,KAAK,EAAE,CAAC,EACR,OAAO,CAAC,EAAE,sBAAsB,GAC/B,OAAO,CAAC;KAAG,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,GAAG,eAAe,CAAC,CAAC,CAAC;CAAE,CAAC,CAanD"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/pipelines/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAGL,WAAW,EACX,gBAAgB,EACjB,MAAM,kBAAkB,CAAC;AAM1B,OAAO,EACL,YAAY,EACZ,gBAAgB,EAChB,kBAAkB,EAClB,gBAAgB,EAChB,cAAc,EACd,eAAe,EACf,KAAK,cAAc,EACnB,KAAK,wBAAwB,EAC7B,KAAK,uBAAuB,EAC5B,KAAK,yBAAyB,EAC9B,KAAK,qBAAqB,GAC3B,MAAM,WAAW,CAAC;AAGnB,OAAO,EACL,0BAA0B,EAC1B,yBAAyB,EACzB,gCAAgC,EAChC,+BAA+B,EAC/B,KAAK,yBAAyB,GAC/B,MAAM,0BAA0B,CAAC;AAGlC,OAAO,EACL,yBAAyB,EACzB,+BAA+B,EAC/B,KAAK,wBAAwB,GAC9B,MAAM,yBAAyB,CAAC;AAGjC,OAAO,EACL,2BAA2B,EAC3B,iCAAiC,EACjC,KAAK,0BAA0B,EAC/B,KAAK,UAAU,GAChB,MAAM,2BAA2B,CAAC;AAGnC,OAAO,EACL,sBAAsB,EACtB,4BAA4B,EAC5B,KAAK,qBAAqB,EAC1B,KAAK,oBAAoB,EACzB,KAAK,qBAAqB,EAC1B,KAAK,WAAW,EAChB,KAAK,WAAW,EAChB,KAAK,gBAAgB,EACrB,KAAK,eAAe,GACrB,MAAM,sBAAsB,CAAC;AAG9B,OAAO,EACL,uBAAuB,EACvB,6BAA6B,EAC7B,WAAW,EACX,KAAK,sBAAsB,EAC3B,KAAK,SAAS,EACd,KAAK,WAAW,GACjB,MAAM,uBAAuB,CAAC;AAG/B,OAAO,EACL,kCAAkC,EAClC,iBAAiB,EACjB,KAAK,UAAU,EACf,KAAK,SAAS,EACd,KAAK,aAAa,EAClB,KAAK,cAAc,GACpB,MAAM,mCAAmC,CAAC;AAG3C,OAAO,EACL,8BAA8B,EAC9B,oCAAoC,EACpC,KAAK,6BAA6B,EAClC,KAAK,4BAA4B,GAClC,MAAM,+BAA+B,CAAC;AAGvC,OAAO,EACL,yBAAyB,EACzB,+BAA+B,EAC/B,KAAK,wBAAwB,EAC7B,KAAK,uBAAuB,EAC5B,KAAK,OAAO,GACb,MAAM,yBAAyB,CAAC;AAGjC,OAAO,EACL,yBAAyB,EACzB,+BAA+B,EAC/B,KAAK,wBAAwB,EAC7B,KAAK,uBAAuB,EAC5B,KAAK,WAAW,EAChB,KAAK,SAAS,EACd,KAAK,iBAAiB,GACvB,MAAM,yBAAyB,CAAC;AAMjC;;GAEG;AACH,MAAM,WAAW,sBAAsB;IACrC,sBAAsB;IACtB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,qBAAqB;IACrB,OAAO,CAAC,EAAE,WAAW,CAAC;IACtB,qBAAqB;IACrB,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB,wBAAwB;IACxB,YAAY,CAAC,EAAE,gBAAgB,CAAC;IAChC,uCAAuC;IACvC,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC;CACnB;AAED;;GAEG;AACH,KAAK,eAAe,GAAG;IACrB,qBAAqB,EAAE,0BAA0B,CAAC;IAClD,oBAAoB,EAAE,yBAAyB,CAAC;IAChD,oBAAoB,EAAE,yBAAyB,CAAC;IAChD,sBAAsB,EAAE,2BAA2B,CAAC;IACpD,iBAAiB,EAAE,sBAAsB,CAAC;IAC1C,kBAAkB,EAAE,uBAAuB,CAAC;IAC5C,8BAA8B,EAAE,kCAAkC,CAAC;IACnE,0BAA0B,EAAE,8BAA8B,CAAC;IAC3D,oBAAoB,EAAE,yBAAyB,CAAC;IAChD,oBAAoB,EAAE,yBAAyB,CAAC;CACjD,CAAC;AAGF,OAAO,EAAE,0BAA0B,EAAE,yBAAyB,EAAE,MAAM,0BAA0B,CAAC;AACjG,OAAO,EAAE,yBAAyB,EAAE,MAAM,yBAAyB,CAAC;AACpE,OAAO,EAAE,2BAA2B,EAAE,MAAM,2BAA2B,CAAC;AACxE,OAAO,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EAAE,uBAAuB,EAAE,MAAM,uBAAuB,CAAC;AAChE,OAAO,EAAE,kCAAkC,EAAE,MAAM,mCAAmC,CAAC;AACvF,OAAO,EAAE,8BAA8B,EAAE,MAAM,+BAA+B,CAAC;AAC/E,OAAO,EAAE,yBAAyB,EAAE,MAAM,yBAAyB,CAAC;AACpE,OAAO,EAAE,yBAAyB,EAAE,MAAM,yBAAyB,CAAC;AAEpE;;;;;;;;;;;;;;GAcG;AACH,wBAAsB,QAAQ,CAAC,CAAC,SAAS,MAAM,eAAe,EAC5D,IAAI,EAAE,CAAC,EACP,OAAO,CAAC,EAAE,sBAAsB,GAC/B,OAAO,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,CAkE7B;AAED;;GAEG;AACH,wBAAsB,eAAe,CAAC,CAAC,SAAS,CAAC,MAAM,eAAe,CAAC,EAAE,EACvE,KAAK,EAAE,CAAC,EACR,OAAO,CAAC,EAAE,sBAAsB,GAC/B,OAAO,CAAC;KAAG,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,GAAG,eAAe,CAAC,CAAC,CAAC;CAAE,CAAC,CAanD"}