@taskmagic/apps-hugging-face 0.0.1

package/src/lib/actions/create-image.js

@@ -0,0 +1,423 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createImage = void 0;
+const tslib_1 = require("tslib");
+const pieces_framework_1 = require("@taskmagic/pieces-framework");
+const pieces_common_1 = require("@taskmagic/pieces-common");
+const auth_1 = require("../auth");
+const common_1 = require("../common");
+exports.createImage = (0, pieces_framework_1.createAction)({
+    name: 'create_image',
+    auth: auth_1.huggingFaceAuth,
+    displayName: 'Create Image',
+    description: 'Generate stunning images from text prompts using state-of-the-art diffusion models - perfect for marketing, product design, and creative content',
+    props: {
+        useCase: pieces_framework_1.Property.StaticDropdown({
+            displayName: 'Use Case',
+            description: 'What type of image generation do you need?',
+            required: true,
+            options: {
+                disabled: false,
+                options: [
+                    {
+                        label: 'Fast Generation (Quick Prototypes)',
+                        value: 'speed',
+                    },
+                    {
+                        label: 'High Quality (Marketing & Print)',
+                        value: 'quality',
+                    },
+                    {
+                        label: 'Business Content (Products & Brands)',
+                        value: 'business',
+                    },
+                    {
+                        label: 'Search All Models',
+                        value: 'search',
+                    },
+                ],
+            },
+            defaultValue: 'quality',
+        }),
+        model: pieces_framework_1.Property.Dropdown({
+            displayName: 'Image Generation Model',
+            description: 'Select the best model for your use case',
+            required: true,
+            refreshers: ['useCase'],
+            options: (_a) => tslib_1.__awaiter(void 0, [_a], void 0, function* ({ useCase }) {
+                const getModelsByUseCase = (type) => {
+                    switch (type) {
+                        case 'speed':
+                            return [
+                                {
+                                    label: 'FLUX.1 Schnell (⚡ Ultra Fast - 1-4 steps)',
+                                    value: 'black-forest-labs/FLUX.1-schnell',
+                                    description: '634K downloads | Best for rapid prototyping',
+                                },
+                                {
+                                    label: 'SD Turbo (⚡ Real-time - 1 step)',
+                                    value: 'stabilityai/sd-turbo',
+                                    description: '1.2M downloads | Fastest generation',
+                                },
+                                {
+                                    label: 'SDXL Turbo (⚡ Fast - 1-4 steps)',
+                                    value: 'stabilityai/sdxl-turbo',
+                                    description: '223K downloads | Fast with good quality',
+                                },
+                            ];
+                        case 'quality':
+                            return [
+                                {
+                                    label: 'FLUX.1 Dev (Premium Quality)',
+                                    value: 'black-forest-labs/FLUX.1-dev',
+                                    description: '1.4M downloads | State-of-the-art results',
+                                },
+                                {
+                                    label: 'Stable Diffusion XL (🎯 Reliable Quality)',
+                                    value: 'stabilityai/stable-diffusion-xl-base-1.0',
+                                    description: '2.2M downloads | Industry standard',
+                                },
+                                {
+                                    label: 'SD 3 Medium (🔬 Advanced Features)',
+                                    value: 'stabilityai/stable-diffusion-3-medium',
+                                    description: '13K downloads | Latest technology',
+                                },
+                            ];
+                        case 'business':
+                            return [
+                                {
+                                    label: 'Stable Diffusion XL (💼 Professional)',
+                                    value: 'stabilityai/stable-diffusion-xl-base-1.0',
+                                    description: '2.2M downloads | Business-ready quality',
+                                },
+                                {
+                                    label: 'DreamShaper v7 (🎨 Versatile Style)',
+                                    value: 'Lykon/dreamshaper-7',
+                                    description: '833K downloads | Great for varied content',
+                                },
+                                {
+                                    label: 'OpenJourney (🌟 Midjourney Style)',
+                                    value: 'prompthero/openjourney',
+                                    description: '8K downloads | Artistic business content',
+                                },
+                            ];
+                        default:
+                            return [];
+                    }
+                };
+                if (useCase === 'search') {
+                    try {
+                        const response = yield pieces_common_1.httpClient.sendRequest({
+                            method: pieces_common_1.HttpMethod.GET,
+                            url: 'https://huggingface.co/api/models?pipeline_tag=text-to-image&sort=downloads&limit=50',
+                        });
+                        const models = response.body;
+                        return {
+                            disabled: false,
+                            placeholder: 'Select from 50+ popular models...',
+                            options: models
+                                .filter((model) => model.downloads > 50000)
+                                .slice(0, 20)
+                                .map((model) => ({
+                                label: `${model.id} (${(model.downloads / 1000).toFixed(0)}K downloads)`,
+                                value: model.id,
+                            })),
+                        };
+                    }
+                    catch (error) {
+                        return {
+                            disabled: false,
+                            options: getModelsByUseCase('quality'),
+                        };
+                    }
+                }
+                return {
+                    disabled: false,
+                    options: getModelsByUseCase(useCase),
+                };
+            }),
+        }),
+        prompt: pieces_framework_1.Property.LongText({
+            displayName: 'Text Prompt',
+            description: 'Describe the image you want to generate. Be specific about style, colors, composition, and details.',
+            required: true,
+            defaultValue: '',
+        }),
+        aspectRatio: pieces_framework_1.Property.StaticDropdown({
+            displayName: 'Aspect Ratio',
+            description: 'Choose the dimensions for your image',
+            required: false,
+            options: {
+                disabled: false,
+                options: [
+                    { label: '📱 Portrait (512×768) - Social Media', value: 'portrait' },
+                    { label: '🖥️ Landscape (768×512) - Banners', value: 'landscape' },
+                    { label: '⬜ Square (512×512) - Profile Pictures', value: 'square' },
+                    { label: '📺 Wide (1024×576) - Headers', value: 'wide' },
+                    { label: '⚙️ Custom Dimensions', value: 'custom' },
+                ],
+            },
+            defaultValue: 'square',
+        }),
+        customWidth: pieces_framework_1.Property.Number({
+            displayName: 'Custom Width',
+            description: 'Width in pixels (64-1024)',
+            required: false,
+        }),
+        customHeight: pieces_framework_1.Property.Number({
+            displayName: 'Custom Height',
+            description: 'Height in pixels (64-1024)',
+            required: false,
+        }),
+        negativePrompt: pieces_framework_1.Property.LongText({
+            displayName: 'Negative Prompt',
+            description: "Describe what you DON'T want in the image (blur, low quality, distorted, etc.)",
+            required: false,
+        }),
+        qualitySettings: pieces_framework_1.Property.StaticDropdown({
+            displayName: 'Quality vs Speed',
+            description: 'Balance between image quality and generation time',
+            required: false,
+            options: {
+                disabled: false,
+                options: [
+                    { label: '⚡ Fast (10-20 steps)', value: 'fast' },
+                    { label: '⚖️ Balanced (20-30 steps)', value: 'balanced' },
+                    { label: '🎯 High Quality (30-50 steps)', value: 'quality' },
+                    { label: '🏆 Maximum Quality (50+ steps)', value: 'maximum' },
+                    { label: '⚙️ Custom Steps', value: 'custom' },
+                ],
+            },
+            defaultValue: 'balanced',
+        }),
+        customSteps: pieces_framework_1.Property.Number({
+            displayName: 'Custom Inference Steps',
+            description: 'Number of denoising steps (1-100)',
+            required: false,
+        }),
+        guidanceScale: pieces_framework_1.Property.Number({
+            displayName: 'Guidance Scale',
+            description: 'How closely to follow the prompt (1-20). Higher values = more prompt adherence but may reduce creativity.',
+            required: false,
+            defaultValue: 7.5,
+        }),
+        seed: pieces_framework_1.Property.Number({
+            displayName: 'Seed (Optional)',
+            description: 'Set a seed for reproducible results. Leave empty for random generation.',
+            required: false,
+        }),
+        scheduler: pieces_framework_1.Property.StaticDropdown({
+            displayName: 'Scheduler',
+            description: 'Advanced: Choose the noise scheduler algorithm',
+            required: false,
+            options: {
+                disabled: false,
+                options: [
+                    { label: 'DPM++ 2M Karras (Recommended)', value: 'DPM++2MKarras' },
+                    { label: 'Euler A (Fast)', value: 'EulerA' },
+                    { label: 'DDIM (Stable)', value: 'DDIM' },
+                    { label: 'LMS (Classic)', value: 'LMS' },
+                ],
+            },
+        }),
+    },
+    run(context) {
+        return tslib_1.__awaiter(this, void 0, void 0, function* () {
+            const { useCase, model, prompt, aspectRatio, customWidth, customHeight, negativePrompt, qualitySettings, customSteps, guidanceScale, seed, scheduler, } = context.propsValue;
+            if (!(prompt === null || prompt === void 0 ? void 0 : prompt.trim())) {
+                throw new Error('Please provide a text prompt for image generation');
+            }
+            let width, height;
+            switch (aspectRatio) {
+                case 'portrait':
+                    width = 512;
+                    height = 768;
+                    break;
+                case 'landscape':
+                    width = 768;
+                    height = 512;
+                    break;
+                case 'square':
+                    width = 512;
+                    height = 512;
+                    break;
+                case 'wide':
+                    width = 1024;
+                    height = 576;
+                    break;
+                case 'custom':
+                    width = customWidth || 512;
+                    height = customHeight || 512;
+                    break;
+                default:
+                    width = 512;
+                    height = 512;
+            }
+            let numInferenceSteps;
+            switch (qualitySettings) {
+                case 'fast':
+                    numInferenceSteps = 15;
+                    break;
+                case 'balanced':
+                    numInferenceSteps = 25;
+                    break;
+                case 'quality':
+                    numInferenceSteps = 40;
+                    break;
+                case 'maximum':
+                    numInferenceSteps = 60;
+                    break;
+                case 'custom':
+                    numInferenceSteps = customSteps || 25;
+                    break;
+                default:
+                    numInferenceSteps = 25;
+            }
+            // Build parameters object
+            const parameters = {
+                guidance_scale: guidanceScale || 7.5,
+                num_inference_steps: numInferenceSteps,
+                width: width,
+                height: height,
+            };
+            if (negativePrompt === null || negativePrompt === void 0 ? void 0 : negativePrompt.trim()) {
+                parameters['negative_prompt'] = negativePrompt.trim();
+            }
+            if (seed !== undefined && seed !== null) {
+                parameters['seed'] = Math.floor(seed);
+            }
+            if (scheduler) {
+                parameters['scheduler'] = scheduler;
+            }
+            const startTime = Date.now();
+            try {
+                const response = yield pieces_common_1.httpClient.sendRequest({
+                    method: pieces_common_1.HttpMethod.POST,
+                    url: `${common_1.HF_ROUTER_BASE}/${model}`,
+                    headers: {
+                        Authorization: `Bearer ${context.auth}`,
+                        'Content-Type': 'application/json',
+                    },
+                    responseType: 'arraybuffer',
+                    body: {
+                        inputs: prompt.trim(),
+                        parameters: parameters,
+                    },
+                });
+                const imageBuffer = Buffer.from(response.body);
+                const base64Image = imageBuffer.toString('base64');
+                const generationTime = (Date.now() - startTime) / 1000;
+                const imageSizeKB = Math.round(imageBuffer.length / 1024);
+                return {
+                    image: base64Image,
+                    imageData: {
+                        format: 'PNG',
+                        width: width,
+                        height: height,
+                        sizeKB: imageSizeKB,
+                        base64: `data:image/png;base64,${base64Image}`,
+                    },
+                    generation: {
+                        prompt: prompt.trim(),
+                        negativePrompt: (negativePrompt === null || negativePrompt === void 0 ? void 0 : negativePrompt.trim()) || '',
+                        model: model,
+                        useCase: useCase,
+                    },
+                    parameters: {
+                        width: width,
+                        height: height,
+                        aspectRatio: aspectRatio,
+                        guidanceScale: guidanceScale || 7.5,
+                        inferenceSteps: numInferenceSteps,
+                        scheduler: scheduler || 'auto',
+                        seed: seed || 'random',
+                    },
+                    metrics: {
+                        generationTimeSeconds: generationTime,
+                        imageSizeKB: imageSizeKB,
+                        resolution: `${width}×${height}`,
+                        qualitySetting: qualitySettings || 'balanced',
+                        estimatedCost: calculateEstimatedCost(model, width, height, numInferenceSteps),
+                    },
+                    businessInsights: {
+                        useCase: getUseCaseDescription(useCase),
+                        qualityTips: getQualityTips(prompt, negativePrompt, numInferenceSteps),
+                        nextSteps: getNextSteps(useCase, imageSizeKB),
+                    },
+                };
+            }
+            catch (error) {
+                const errorMessage = error instanceof Error ? error.message : String(error);
+                throw new Error(`Image generation failed: ${errorMessage}`);
+            }
+        });
+    },
+});
+function getUseCaseDescription(useCase) {
+    const descriptions = {
+        speed: 'Optimized for rapid prototyping and quick content creation',
+        quality: 'High-resolution generation perfect for marketing and professional use',
+        business: 'Business-focused models ideal for product imagery and brand content',
+        search: 'Custom model selection for specialized requirements',
+    };
+    return (descriptions[useCase] ||
+        'AI-powered image generation');
+}
+function getQualityTips(prompt, negativePrompt, steps) {
+    const tips = [];
+    if (prompt.length < 20) {
+        tips.push('💡 Add more descriptive details to your prompt for better results');
+    }
+    if (!negativePrompt) {
+        tips.push('🚫 Consider adding negative prompts to avoid unwanted elements (blur, low quality, etc.)');
+    }
+    if (steps < 20) {
+        tips.push('⚡ Using fast generation - increase steps for higher quality if needed');
+    }
+    if (steps > 50) {
+        tips.push('⏱️ High step count may take longer - consider balanced setting for faster results');
+    }
+    if (!prompt.toLowerCase().includes('high quality') &&
+        !prompt.toLowerCase().includes('detailed')) {
+        tips.push('🎯 Add quality keywords like "high resolution", "detailed", or "professional" to your prompt');
+    }
+    if (tips.length === 0) {
+        tips.push('✅ Good generation parameters - expect quality results');
+    }
+    return tips;
+}
+function getNextSteps(useCase, imageSizeKB) {
+    const steps = [];
+    if (useCase === 'business') {
+        steps.push('🎨 Try variations with different styles or angles');
+        steps.push('📐 Consider generating multiple aspect ratios for different platforms');
+    }
+    if (useCase === 'speed') {
+        steps.push('🔄 Generate variations quickly with different seeds');
+        steps.push('⬆️ Upscale to higher quality when you find the perfect concept');
+    }
+    if (imageSizeKB > 1000) {
+        steps.push('📉 Consider optimizing image size for web use');
+    }
+    steps.push('💾 Save successful prompts for consistent brand imagery');
+    steps.push('🔄 Use the same seed to create variations of this concept');
+    return steps;
+}
+function calculateEstimatedCost(model, width, height, steps) {
+    const basePixelCost = 0.000001;
+    const stepMultiplier = steps / 25;
+    const modelMultiplier = model.includes('FLUX.1-dev')
+        ? 2.5
+        : model.includes('stable-diffusion-3')
+            ? 2.0
+            : model.includes('turbo')
+                ? 0.5
+                : 1.0;
+    const pixels = width * height;
+    const estimatedCost = pixels * basePixelCost * stepMultiplier * modelMultiplier;
+    if (estimatedCost < 0.001) {
+        return '< $0.001';
+    }
+    return `~$${estimatedCost.toFixed(4)}`;
+}
+//# sourceMappingURL=create-image.js.map

package/src/lib/actions/create-image.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"create-image.js","sourceRoot":"","sources":["../../../../../../../../packages/pieces/community/hugging-face/src/lib/actions/create-image.ts"],"names":[],"mappings":";;;;AAAA,kEAAqE;AACrE,4DAAkE;AAClE,kCAA0C;AAC1C,sCAA2C;AAE9B,QAAA,WAAW,GAAG,IAAA,+BAAY,EAAC;IACtC,IAAI,EAAE,cAAc;IACpB,IAAI,EAAE,sBAAe;IACrB,WAAW,EAAE,cAAc;IAC3B,WAAW,EACT,kJAAkJ;IACpJ,KAAK,EAAE;QACL,OAAO,EAAE,2BAAQ,CAAC,cAAc,CAAC;YAC/B,WAAW,EAAE,UAAU;YACvB,WAAW,EAAE,4CAA4C;YACzD,QAAQ,EAAE,IAAI;YACd,OAAO,EAAE;gBACP,QAAQ,EAAE,KAAK;gBACf,OAAO,EAAE;oBACP;wBACE,KAAK,EAAE,oCAAoC;wBAC3C,KAAK,EAAE,OAAO;qBACf;oBACD;wBACE,KAAK,EAAE,kCAAkC;wBACzC,KAAK,EAAE,SAAS;qBACjB;oBACD;wBACE,KAAK,EAAE,sCAAsC;wBAC7C,KAAK,EAAE,UAAU;qBAClB;oBACD;wBACE,KAAK,EAAE,mBAAmB;wBAC1B,KAAK,EAAE,QAAQ;qBAChB;iBACF;aACF;YACD,YAAY,EAAE,SAAS;SACxB,CAAC;QACF,KAAK,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YACvB,WAAW,EAAE,wBAAwB;YACrC,WAAW,EAAE,yCAAyC;YACtD,QAAQ,EAAE,IAAI;YACd,UAAU,EAAE,CAAC,SAAS,CAAC;YACvB,OAAO,EAAE,KAAoB,EAAE,oDAAf,EAAE,OAAO,EAAE;gBACzB,MAAM,kBAAkB,GAAG,CAAC,IAAY,EAAE,EAAE;oBAC1C,QAAQ,IAAI,EAAE,CAAC;wBACb,KAAK,OAAO;4BACV,OAAO;gCACL;oCACE,KAAK,EAAE,2CAA2C;oCAClD,KAAK,EAAE,kCAAkC;oCACzC,WAAW,EAAE,6CAA6C;iCAC3D;gCACD;oCACE,KAAK,EAAE,iCAAiC;oCACxC,KAAK,EAAE,sBAAsB;oCAC7B,WAAW,EAAE,qCAAqC;iCACnD;gCACD;oCACE,KAAK,EAAE,iCAAiC;oCACxC,KAAK,EAAE,wBAAwB;oCAC/B,WAAW,EAAE,yCAAyC;iCACvD;6BACF,CAAC;wBACJ,KAAK,SAAS;4BACZ,OAAO;gCACL;oCACE,KAAK,EAAE,8BAA8B;oCACrC,KAAK,EAAE,8BAA8B;oCACrC,WAAW,EAAE,2CAA2C;iCACzD;gCACD;oCACE,KAAK,EAAE,2CAA2C;oCAClD,KAAK,EAAE,0CAA0C;oCACjD,WAAW,EAAE,oCAAoC;iCAClD;gCACD;oCACE,KAAK,EAAE,oCAAoC;oCAC3C,KAAK,EAAE,uCAAuC;oCAC9C,WAAW,EAAE,mCAAmC;iCACjD;6BACF,CAAC;wBACJ,KAAK,UAAU;4BACb,OAAO;gCACL;oCACE,KAAK,EAAE,uCAAuC;oCAC9C,KAAK,EAAE,0CAA0C;oCACjD,WAAW,EAAE,yCAAyC;iCACvD;gCACD;oCACE,KAAK,EAAE,qCAAqC;oCAC5C,KAAK,EAAE,qBAAqB;oCAC5B,WAAW,EAAE,2CAA2C;iCACzD;gCACD;oCACE,KAAK,EAAE,mCAAmC;oCAC1C,KAAK,EAAE,wBAAwB;oCAC/B,WAAW,EAAE,0CAA0C;iCACxD;6BACF,CAAC;wBACJ;4BACE,OAAO,EAAE,CAAC;oBACd,CAAC;gBACH,CAAC,CAAC;gBAEF,IAAI,OAAO,KAAK,QAAQ,EAAE,CAAC;oBACzB,IAAI,CAAC;wBACH,MAAM,QAAQ,GAAG,MAAM,0BAAU,CAAC,WAAW,CAAC;4BAC5C,MAAM,EAAE,0BAAU,CAAC,GAAG;4BACtB,GAAG,EAAE,sFAAsF;yBAC5F,CAAC,CAAC;wBAEH,MAAM,MAAM,GAAG,QAAQ,CAAC,IAItB,CAAC;wBAEH,OAAO;4BACL,QAAQ,EAAE,KAAK;4BACf,WAAW,EAAE,mCAAmC;4BAChD,OAAO,EAAE,MAAM;iCACZ,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,SAAS,GAAG,KAAK,CAAC;iCAC1C,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC;iCACZ,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;gCACf,KAAK,EAAE,GAAG,KAAK,CAAC,EAAE,KAAK,CAAC,KAAK,CAAC,SAAS,GAAG,IAAI,CAAC,CAAC,OAAO,CACrD,CAAC,CACF,cAAc;gCACf,KAAK,EAAE,KAAK,CAAC,EAAE;6BAChB,CAAC,CAAC;yBACN,CAAC;oBACJ,CAAC;oBAAC,OAAO,KAAK,EAAE,CAAC;wBACf,OAAO;4BACL,QAAQ,EAAE,KAAK;4BACf,OAAO,EAAE,kBAAkB,CAAC,SAAS,CAAC;yBACvC,CAAC;oBACJ,CAAC;gBACH,CAAC;gBAED,OAAO;oBACL,QAAQ,EAAE,KAAK;oBACf,OAAO,EAAE,kBAAkB,CAAC,OAAiB,CAAC;iBAC/C,CAAC;YACJ,CAAC,CAAA;SACF,CAAC;QACF,MAAM,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YACxB,WAAW,EAAE,aAAa;YAC1B,WAAW,EACT,qGAAqG;YACvG,QAAQ,EAAE,IAAI;YACd,YAAY,EAAE,EAAE;SACjB,CAAC;QACF,WAAW,EAAE,2BAAQ,CAAC,cAAc,CAAC;YACnC,WAAW,EAAE,cAAc;YAC3B,WAAW,EAAE,sCAAsC;YACnD,QAAQ,EAAE,KAAK;YACf,OAAO,EAAE;gBACP,QAAQ,EAAE,KAAK;gBACf,OAAO,EAAE;oBACP,EAAE,KAAK,EAAE,sCAAsC,EAAE,KAAK,EAAE,UAAU,EAAE;oBACpE,EAAE,KAAK,EAAE,mCAAmC,EAAE,KAAK,EAAE,WAAW,EAAE;oBAClE,EAAE,KAAK,EAAE,uCAAuC,EAAE,KAAK,EAAE,QAAQ,EAAE;oBACnE,EAAE,KAAK,EAAE,8BAA8B,EAAE,KAAK,EAAE,MAAM,EAAE;oBACxD,EAAE,KAAK,EAAE,sBAAsB,EAAE,KAAK,EAAE,QAAQ,EAAE;iBACnD;aACF;YACD,YAAY,EAAE,QAAQ;SACvB,CAAC;QACF,WAAW,EAAE,2BAAQ,CAAC,MAAM,CAAC;YAC3B,WAAW,EAAE,cAAc;YAC3B,WAAW,EAAE,2BAA2B;YACxC,QAAQ,EAAE,KAAK;SAChB,CAAC;QACF,YAAY,EAAE,2BAAQ,CAAC,MAAM,CAAC;YAC5B,WAAW,EAAE,eAAe;YAC5B,WAAW,EAAE,4BAA4B;YACzC,QAAQ,EAAE,KAAK;SAChB,CAAC;QACF,cAAc,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YAChC,WAAW,EAAE,iBAAiB;YAC9B,WAAW,EACT,gFAAgF;YAClF,QAAQ,EAAE,KAAK;SAChB,CAAC;QACF,eAAe,EAAE,2BAAQ,CAAC,cAAc,CAAC;YACvC,WAAW,EAAE,kBAAkB;YAC/B,WAAW,EAAE,mDAAmD;YAChE,QAAQ,EAAE,KAAK;YACf,OAAO,EAAE;gBACP,QAAQ,EAAE,KAAK;gBACf,OAAO,EAAE;oBACP,EAAE,KAAK,EAAE,sBAAsB,EAAE,KAAK,EAAE,MAAM,EAAE;oBAChD,EAAE,KAAK,EAAE,2BAA2B,EAAE,KAAK,EAAE,UAAU,EAAE;oBACzD,EAAE,KAAK,EAAE,+BAA+B,EAAE,KAAK,EAAE,SAAS,EAAE;oBAC5D,EAAE,KAAK,EAAE,gCAAgC,EAAE,KAAK,EAAE,SAAS,EAAE;oBAC7D,EAAE,KAAK,EAAE,iBAAiB,EAAE,KAAK,EAAE,QAAQ,EAAE;iBAC9C;aACF;YACD,YAAY,EAAE,UAAU;SACzB,CAAC;QACF,WAAW,EAAE,2BAAQ,CAAC,MAAM,CAAC;YAC3B,WAAW,EAAE,wBAAwB;YACrC,WAAW,EAAE,mCAAmC;YAChD,QAAQ,EAAE,KAAK;SAChB,CAAC;QACF,aAAa,EAAE,2BAAQ,CAAC,MAAM,CAAC;YAC7B,WAAW,EAAE,gBAAgB;YAC7B,WAAW,EACT,2GAA2G;YAC7G,QAAQ,EAAE,KAAK;YACf,YAAY,EAAE,GAAG;SAClB,CAAC;QACF,IAAI,EAAE,2BAAQ,CAAC,MAAM,CAAC;YACpB,WAAW,EAAE,iBAAiB;YAC9B,WAAW,EACT,yEAAyE;YAC3E,QAAQ,EAAE,KAAK;SAChB,CAAC;QACF,SAAS,EAAE,2BAAQ,CAAC,cAAc,CAAC;YACjC,WAAW,EAAE,WAAW;YACxB,WAAW,EAAE,gDAAgD;YAC7D,QAAQ,EAAE,KAAK;YACf,OAAO,EAAE;gBACP,QAAQ,EAAE,KAAK;gBACf,OAAO,EAAE;oBACP,EAAE,KAAK,EAAE,+BAA+B,EAAE,KAAK,EAAE,eAAe,EAAE;oBAClE,EAAE,KAAK,EAAE,gBAAgB,EAAE,KAAK,EAAE,QAAQ,EAAE;oBAC5C,EAAE,KAAK,EAAE,eAAe,EAAE,KAAK,EAAE,MAAM,EAAE;oBACzC,EAAE,KAAK,EAAE,eAAe,EAAE,KAAK,EAAE,KAAK,EAAE;iBACzC;aACF;SACF,CAAC;KACH;IACK,GAAG,CAAC,OAAO;;YACf,MAAM,EACJ,OAAO,EACP,KAAK,EACL,MAAM,EACN,WAAW,EACX,WAAW,EACX,YAAY,EACZ,cAAc,EACd,eAAe,EACf,WAAW,EACX,aAAa,EACb,IAAI,EACJ,SAAS,GACV,GAAG,OAAO,CAAC,UAAU,CAAC;YAEvB,IAAI,CAAC,CAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,IAAI,EAAE,CAAA,EAAE,CAAC;gBACpB,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAC;YACvE,CAAC;YAED,IAAI,KAAa,EAAE,MAAc,CAAC;YAClC,QAAQ,WAAW,EAAE,CAAC;gBACpB,KAAK,UAAU;oBACb,KAAK,GAAG,GAAG,CAAC;oBACZ,MAAM,GAAG,GAAG,CAAC;oBACb,MAAM;gBACR,KAAK,WAAW;oBACd,KAAK,GAAG,GAAG,CAAC;oBACZ,MAAM,GAAG,GAAG,CAAC;oBACb,MAAM;gBACR,KAAK,QAAQ;oBACX,KAAK,GAAG,GAAG,CAAC;oBACZ,MAAM,GAAG,GAAG,CAAC;oBACb,MAAM;gBACR,KAAK,MAAM;oBACT,KAAK,GAAG,IAAI,CAAC;oBACb,MAAM,GAAG,GAAG,CAAC;oBACb,MAAM;gBACR,KAAK,QAAQ;oBACX,KAAK,GAAG,WAAW,IAAI,GAAG,CAAC;oBAC3B,MAAM,GAAG,YAAY,IAAI,GAAG,CAAC;oBAC7B,MAAM;gBACR;oBACE,KAAK,GAAG,GAAG,CAAC;oBACZ,MAAM,GAAG,GAAG,CAAC;YACjB,CAAC;YAED,IAAI,iBAAyB,CAAC;YAC9B,QAAQ,eAAe,EAAE,CAAC;gBACxB,KAAK,MAAM;oBACT,iBAAiB,GAAG,EAAE,CAAC;oBACvB,MAAM;gBACR,KAAK,UAAU;oBACb,iBAAiB,GAAG,EAAE,CAAC;oBACvB,MAAM;gBACR,KAAK,SAAS;oBACZ,iBAAiB,GAAG,EAAE,CAAC;oBACvB,MAAM;gBACR,KAAK,SAAS;oBACZ,iBAAiB,GAAG,EAAE,CAAC;oBACvB,MAAM;gBACR,KAAK,QAAQ;oBACX,iBAAiB,GAAG,WAAW,IAAI,EAAE,CAAC;oBACtC,MAAM;gBACR;oBACE,iBAAiB,GAAG,EAAE,CAAC;YAC3B,CAAC;YAED,0BAA0B;YAC1B,MAAM,UAAU,GAA4B;gBAC1C,cAAc,EAAE,aAAa,IAAI,GAAG;gBACpC,mBAAmB,EAAE,iBAAiB;gBACtC,KAAK,EAAE,KAAK;gBACZ,MAAM,EAAE,MAAM;aACf,CAAC;YAEF,IAAI,cAAc,aAAd,cAAc,uBAAd,cAAc,CAAE,IAAI,EAAE,EAAE,CAAC;gBAC3B,UAAU,CAAC,iBAAiB,CAAC,GAAG,cAAc,CAAC,IAAI,EAAE,CAAC;YACxD,CAAC;YAED,IAAI,IAAI,KAAK,SAAS,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;gBACxC,UAAU,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACxC,CAAC;YAED,IAAI,SAAS,EAAE,CAAC;gBACd,UAAU,CAAC,WAAW,CAAC,GAAG,SAAS,CAAC;YACtC,CAAC;YAED,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAE7B,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,MAAM,0BAAU,CAAC,WAAW,CAAS;oBACpD,MAAM,EAAE,0BAAU,CAAC,IAAI;oBACvB,GAAG,EAAE,GAAG,uBAAc,IAAI,KAAK,EAAE;oBACjC,OAAO,EAAE;wBACP,aAAa,EAAE,UAAU,OAAO,CAAC,IAAc,EAAE;wBACjD,cAAc,EAAE,kBAAkB;qBACnC;oBACD,YAAY,EAAE,aAAa;oBAC3B,IAAI,EAAE;wBACJ,MAAM,EAAE,MAAM,CAAC,IAAI,EAAE;wBACrB,UAAU,EAAE,UAAU;qBACvB;iBACF,CAAC,CAAC;gBAEH,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;gBAC/C,MAAM,WAAW,GAAG,WAAW,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;gBAEnD,MAAM,cAAc,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC;gBACvD,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC;gBAE1D,OAAO;oBACL,KAAK,EAAE,WAAW;oBAClB,SAAS,EAAE;wBACT,MAAM,EAAE,KAAK;wBACb,KAAK,EAAE,KAAK;wBACZ,MAAM,EAAE,MAAM;wBACd,MAAM,EAAE,WAAW;wBACnB,MAAM,EAAE,yBAAyB,WAAW,EAAE;qBAC/C;oBACD,UAAU,EAAE;wBACV,MAAM,EAAE,MAAM,CAAC,IAAI,EAAE;wBACrB,cAAc,EAAE,CAAA,cAAc,aAAd,cAAc,uBAAd,cAAc,CAAE,IAAI,EAAE,KAAI,EAAE;wBAC5C,KAAK,EAAE,KAAK;wBACZ,OAAO,EAAE,OAAO;qBACjB;oBACD,UAAU,EAAE;wBACV,KAAK,EAAE,KAAK;wBACZ,MAAM,EAAE,MAAM;wBACd,WAAW,EAAE,WAAW;wBACxB,aAAa,EAAE,aAAa,IAAI,GAAG;wBACnC,cAAc,EAAE,iBAAiB;wBACjC,SAAS,EAAE,SAAS,IAAI,MAAM;wBAC9B,IAAI,EAAE,IAAI,IAAI,QAAQ;qBACvB;oBACD,OAAO,EAAE;wBACP,qBAAqB,EAAE,cAAc;wBACrC,WAAW,EAAE,WAAW;wBACxB,UAAU,EAAE,GAAG,KAAK,IAAI,MAAM,EAAE;wBAChC,cAAc,EAAE,eAAe,IAAI,UAAU;wBAC7C,aAAa,EAAE,sBAAsB,CACnC,KAAK,EACL,KAAK,EACL,MAAM,EACN,iBAAiB,CAClB;qBACF;oBACD,gBAAgB,EAAE;wBAChB,OAAO,EAAE,qBAAqB,CAAC,OAAiB,CAAC;wBACjD,WAAW,EAAE,cAAc,CAAC,MAAM,EAAE,cAAc,EAAE,iBAAiB,CAAC;wBACtE,SAAS,EAAE,YAAY,CAAC,OAAiB,EAAE,WAAW,CAAC;qBACxD;iBACF,CAAC;YACJ,CAAC;YAAC,OAAO,KAAc,EAAE,CAAC;gBACxB,MAAM,YAAY,GAChB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBACzD,MAAM,IAAI,KAAK,CAAC,4BAA4B,YAAY,EAAE,CAAC,CAAC;YAC9D,CAAC;QACH,CAAC;KAAA;CACF,CAAC,CAAC;AAEH,SAAS,qBAAqB,CAAC,OAAe;IAC5C,MAAM,YAAY,GAAG;QACnB,KAAK,EAAE,4DAA4D;QACnE,OAAO,EACL,uEAAuE;QACzE,QAAQ,EACN,qEAAqE;QACvE,MAAM,EAAE,qDAAqD;KAC9D,CAAC;IAEF,OAAO,CACL,YAAY,CAAC,OAAoC,CAAC;QAClD,6BAA6B,CAC9B,CAAC;AACJ,CAAC;AAED,SAAS,cAAc,CACrB,MAAc,EACd,cAAkC,EAClC,KAAa;IAEb,MAAM,IAAI,GAAa,EAAE,CAAC;IAE1B,IAAI,MAAM,CAAC,MAAM,GAAG,EAAE,EAAE,CAAC;QACvB,IAAI,CAAC,IAAI,CACP,mEAAmE,CACpE,CAAC;IACJ,CAAC;IAED,IAAI,CAAC,cAAc,EAAE,CAAC;QACpB,IAAI,CAAC,IAAI,CACP,0FAA0F,CAC3F,CAAC;IACJ,CAAC;IAED,IAAI,KAAK,GAAG,EAAE,EAAE,CAAC;QACf,IAAI,CAAC,IAAI,CACP,uEAAuE,CACxE,CAAC;IACJ,CAAC;IAED,IAAI,KAAK,GAAG,EAAE,EAAE,CAAC;QACf,IAAI,CAAC,IAAI,CACP,mFAAmF,CACpF,CAAC;IACJ,CAAC;IAED,IACE,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,cAAc,CAAC;QAC9C,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,UAAU,CAAC,EAC1C,CAAC;QACD,IAAI,CAAC,IAAI,CACP,8FAA8F,CAC/F,CAAC;IACJ,CAAC;IAED,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACtB,IAAI,CAAC,IAAI,CAAC,uDAAuD,CAAC,CAAC;IACrE,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,YAAY,CAAC,OAAe,EAAE,WAAmB;IACxD,MAAM,KAAK,GAAa,EAAE,CAAC;IAE3B,IAAI,OAAO,KAAK,UAAU,EAAE,CAAC;QAC3B,KAAK,CAAC,IAAI,CAAC,mDAAmD,CAAC,CAAC;QAChE,KAAK,CAAC,IAAI,CACR,uEAAuE,CACxE,CAAC;IACJ,CAAC;IAED,IAAI,OAAO,KAAK,OAAO,EAAE,CAAC;QACxB,KAAK,CAAC,IAAI,CAAC,qDAAqD,CAAC,CAAC;QAClE,KAAK,CAAC,IAAI,CACR,gEAAgE,CACjE,CAAC;IACJ,CAAC;IAED,IAAI,WAAW,GAAG,IAAI,EAAE,CAAC;QACvB,KAAK,CAAC,IAAI,CAAC,+CAA+C,CAAC,CAAC;IAC9D,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,yDAAyD,CAAC,CAAC;IACtE,KAAK,CAAC,IAAI,CAAC,2DAA2D,CAAC,CAAC;IAExE,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,sBAAsB,CAC7B,KAAa,EACb,KAAa,EACb,MAAc,EACd,KAAa;IAEb,MAAM,aAAa,GAAG,QAAQ,CAAC;IAC/B,MAAM,cAAc,GAAG,KAAK,GAAG,EAAE,CAAC;IAClC,MAAM,eAAe,GAAG,KAAK,CAAC,QAAQ,CAAC,YAAY,CAAC;QAClD,CAAC,CAAC,GAAG;QACL,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,oBAAoB,CAAC;YACtC,CAAC,CAAC,GAAG;YACL,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC;gBACzB,CAAC,CAAC,GAAG;gBACL,CAAC,CAAC,GAAG,CAAC;IAER,MAAM,MAAM,GAAG,KAAK,GAAG,MAAM,CAAC;IAC9B,MAAM,aAAa,GACjB,MAAM,GAAG,aAAa,GAAG,cAAc,GAAG,eAAe,CAAC;IAE5D,IAAI,aAAa,GAAG,KAAK,EAAE,CAAC;QAC1B,OAAO,UAAU,CAAC;IACpB,CAAC;IAED,OAAO,KAAK,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;AACzC,CAAC"}

package/src/lib/actions/document-question-answering.d.ts

@@ -0,0 +1,11 @@
+export declare const documentQuestionAnswering: import("@taskmagic/pieces-framework").IAction<import("@taskmagic/pieces-framework").SecretTextProperty<true>, {
+    model: import("@taskmagic/pieces-framework").StaticDropdownProperty<string, true>;
+    image: import("@taskmagic/pieces-framework").FileProperty<true>;
+    question: import("@taskmagic/pieces-framework").ShortTextProperty<true>;
+    top_k: import("@taskmagic/pieces-framework").NumberProperty<false>;
+    max_answer_len: import("@taskmagic/pieces-framework").NumberProperty<false>;
+    handle_impossible_answer: import("@taskmagic/pieces-framework").CheckboxProperty<false>;
+    lang: import("@taskmagic/pieces-framework").StaticDropdownProperty<string, false>;
+    use_cache: import("@taskmagic/pieces-framework").CheckboxProperty<false>;
+    wait_for_model: import("@taskmagic/pieces-framework").CheckboxProperty<false>;
+}>;

package/src/lib/actions/document-question-answering.js

@@ -0,0 +1,140 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.documentQuestionAnswering = void 0;
+const tslib_1 = require("tslib");
+const pieces_framework_1 = require("@taskmagic/pieces-framework");
+const auth_1 = require("../auth");
+const common_1 = require("../common");
+exports.documentQuestionAnswering = (0, pieces_framework_1.createAction)({
+    name: 'document_question_answering',
+    auth: auth_1.huggingFaceAuth,
+    displayName: 'Document Question Answering',
+    description: 'Answer questions from document images using Hugging Face models',
+    props: {
+        model: pieces_framework_1.Property.StaticDropdown({
+            displayName: 'Model',
+            description: 'Hugging Face document question answering model',
+            required: true,
+            options: {
+                disabled: false,
+                options: [
+                    {
+                        label: 'impira/layoutlm-document-qa (Recommended)',
+                        value: 'impira/layoutlm-document-qa',
+                    },
+                    {
+                        label: 'microsoft/layoutlmv3-base',
+                        value: 'microsoft/layoutlmv3-base',
+                    },
+                    {
+                        label: 'nielsr/layoutlmv2-finetuned-docvqa',
+                        value: 'nielsr/layoutlmv2-finetuned-docvqa',
+                    },
+                ],
+            },
+            defaultValue: 'impira/layoutlm-document-qa',
+        }),
+        image: pieces_framework_1.Property.File({
+            displayName: 'Document Image',
+            description: 'Image of the document to analyze (invoice, contract, etc.)',
+            required: true,
+        }),
+        question: pieces_framework_1.Property.ShortText({
+            displayName: 'Question',
+            description: "Question to ask about the document (e.g., 'What is the invoice total?')",
+            required: true,
+        }),
+        top_k: pieces_framework_1.Property.Number({
+            displayName: 'Number of Answers',
+            description: 'Number of top answers to return',
+            required: false,
+            defaultValue: 1,
+        }),
+        max_answer_len: pieces_framework_1.Property.Number({
+            displayName: 'Max Answer Length',
+            description: 'Maximum length of predicted answers',
+            required: false,
+        }),
+        handle_impossible_answer: pieces_framework_1.Property.Checkbox({
+            displayName: 'Handle Impossible Answers',
+            description: "Whether to accept 'impossible' as an answer when no answer is found",
+            required: false,
+            defaultValue: true,
+        }),
+        lang: pieces_framework_1.Property.StaticDropdown({
+            displayName: 'OCR Language',
+            description: 'Language to use for OCR text extraction',
+            required: false,
+            options: {
+                disabled: false,
+                options: [
+                    { label: 'English', value: 'en' },
+                    { label: 'Spanish', value: 'es' },
+                    { label: 'French', value: 'fr' },
+                    { label: 'German', value: 'de' },
+                    { label: 'Italian', value: 'it' },
+                    { label: 'Portuguese', value: 'pt' },
+                ],
+            },
+            defaultValue: 'en',
+        }),
+        use_cache: pieces_framework_1.Property.Checkbox({
+            displayName: 'Use Cache',
+            description: 'Use cached results if available',
+            required: false,
+            defaultValue: true,
+        }),
+        wait_for_model: pieces_framework_1.Property.Checkbox({
+            displayName: 'Wait for Model',
+            description: 'Wait for model to load if not ready',
+            required: false,
+            defaultValue: false,
+        }),
+    },
+    run(context) {
+        return tslib_1.__awaiter(this, void 0, void 0, function* () {
+            const { model, image, question, top_k, max_answer_len, handle_impossible_answer, lang, use_cache, wait_for_model, } = context.propsValue;
+            const base64Image = Buffer.from(image.data).toString('base64');
+            const parameters = {};
+            if (top_k !== undefined) {
+                parameters.top_k = top_k;
+            }
+            if (max_answer_len !== undefined) {
+                parameters.max_answer_len = max_answer_len;
+            }
+            if (handle_impossible_answer !== undefined) {
+                parameters.handle_impossible_answer = handle_impossible_answer;
+            }
+            if (lang) {
+                parameters.lang = lang;
+            }
+            const body = {
+                inputs: {
+                    image: base64Image,
+                    question: question,
+                },
+                options: {
+                    use_cache: use_cache !== null && use_cache !== void 0 ? use_cache : true,
+                    wait_for_model: wait_for_model !== null && wait_for_model !== void 0 ? wait_for_model : false,
+                },
+            };
+            if (Object.keys(parameters).length > 0) {
+                body['parameters'] = parameters;
+            }
+            const rawResult = yield (0, common_1.callPipelineTask)({
+                auth: context.auth,
+                model: model,
+                body,
+            });
+            const result = Array.isArray(rawResult) ? rawResult[0] : rawResult;
+            return {
+                answer: result.answer,
+                score: result.score,
+                start: result.start,
+                end: result.end,
+                raw_result: rawResult,
+            };
+        });
+    },
+});
+//# sourceMappingURL=document-question-answering.js.map

package/src/lib/actions/document-question-answering.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"document-question-answering.js","sourceRoot":"","sources":["../../../../../../../../packages/pieces/community/hugging-face/src/lib/actions/document-question-answering.ts"],"names":[],"mappings":";;;;AAAA,kEAAqE;AACrE,kCAA0C;AAC1C,sCAA6C;AAShC,QAAA,yBAAyB,GAAG,IAAA,+BAAY,EAAC;IACpD,IAAI,EAAE,6BAA6B;IACnC,IAAI,EAAE,sBAAe;IACrB,WAAW,EAAE,6BAA6B;IAC1C,WAAW,EACT,iEAAiE;IACnE,KAAK,EAAE;QACL,KAAK,EAAE,2BAAQ,CAAC,cAAc,CAAC;YAC7B,WAAW,EAAE,OAAO;YACpB,WAAW,EAAE,gDAAgD;YAC7D,QAAQ,EAAE,IAAI;YACd,OAAO,EAAE;gBACP,QAAQ,EAAE,KAAK;gBACf,OAAO,EAAE;oBACP;wBACE,KAAK,EAAE,2CAA2C;wBAClD,KAAK,EAAE,6BAA6B;qBACrC;oBACD;wBACE,KAAK,EAAE,2BAA2B;wBAClC,KAAK,EAAE,2BAA2B;qBACnC;oBACD;wBACE,KAAK,EAAE,oCAAoC;wBAC3C,KAAK,EAAE,oCAAoC;qBAC5C;iBACF;aACF;YACD,YAAY,EAAE,6BAA6B;SAC5C,CAAC;QACF,KAAK,EAAE,2BAAQ,CAAC,IAAI,CAAC;YACnB,WAAW,EAAE,gBAAgB;YAC7B,WAAW,EAAE,4DAA4D;YACzE,QAAQ,EAAE,IAAI;SACf,CAAC;QACF,QAAQ,EAAE,2BAAQ,CAAC,SAAS,CAAC;YAC3B,WAAW,EAAE,UAAU;YACvB,WAAW,EACT,yEAAyE;YAC3E,QAAQ,EAAE,IAAI;SACf,CAAC;QACF,KAAK,EAAE,2BAAQ,CAAC,MAAM,CAAC;YACrB,WAAW,EAAE,mBAAmB;YAChC,WAAW,EAAE,iCAAiC;YAC9C,QAAQ,EAAE,KAAK;YACf,YAAY,EAAE,CAAC;SAChB,CAAC;QACF,cAAc,EAAE,2BAAQ,CAAC,MAAM,CAAC;YAC9B,WAAW,EAAE,mBAAmB;YAChC,WAAW,EAAE,qCAAqC;YAClD,QAAQ,EAAE,KAAK;SAChB,CAAC;QACF,wBAAwB,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YAC1C,WAAW,EAAE,2BAA2B;YACxC,WAAW,EACT,qEAAqE;YACvE,QAAQ,EAAE,KAAK;YACf,YAAY,EAAE,IAAI;SACnB,CAAC;QACF,IAAI,EAAE,2BAAQ,CAAC,cAAc,CAAC;YAC5B,WAAW,EAAE,cAAc;YAC3B,WAAW,EAAE,yCAAyC;YACtD,QAAQ,EAAE,KAAK;YACf,OAAO,EAAE;gBACP,QAAQ,EAAE,KAAK;gBACf,OAAO,EAAE;oBACP,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,IAAI,EAAE;oBACjC,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,IAAI,EAAE;oBACjC,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE;oBAChC,EAAE,KAAK,EAAE,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE;oBAChC,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,IAAI,EAAE;oBACjC,EAAE,KAAK,EAAE,YAAY,EAAE,KAAK,EAAE,IAAI,EAAE;iBACrC;aACF;YACD,YAAY,EAAE,IAAI;SACnB,CAAC;QACF,SAAS,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YAC3B,WAAW,EAAE,WAAW;YACxB,WAAW,EAAE,iCAAiC;YAC9C,QAAQ,EAAE,KAAK;YACf,YAAY,EAAE,IAAI;SACnB,CAAC;QACF,cAAc,EAAE,2BAAQ,CAAC,QAAQ,CAAC;YAChC,WAAW,EAAE,gBAAgB;YAC7B,WAAW,EAAE,qCAAqC;YAClD,QAAQ,EAAE,KAAK;YACf,YAAY,EAAE,KAAK;SACpB,CAAC;KACH;IACK,GAAG,CAAC,OAAO;;YACf,MAAM,EACJ,KAAK,EACL,KAAK,EACL,QAAQ,EACR,KAAK,EACL,cAAc,EACd,wBAAwB,EACxB,IAAI,EACJ,SAAS,EACT,cAAc,GACf,GAAG,OAAO,CAAC,UAAU,CAAC;YAEvB,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;YAE/D,MAAM,UAAU,GAKZ,EAAE,CAAC;YAEP,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;gBACxB,UAAU,CAAC,KAAK,GAAG,KAAK,CAAC;YAC3B,CAAC;YAED,IAAI,cAAc,KAAK,SAAS,EAAE,CAAC;gBACjC,UAAU,CAAC,cAAc,GAAG,cAAc,CAAC;YAC7C,CAAC;YAED,IAAI,wBAAwB,KAAK,SAAS,EAAE,CAAC;gBAC3C,UAAU,CAAC,wBAAwB,GAAG,wBAAwB,CAAC;YACjE,CAAC;YAED,IAAI,IAAI,EAAE,CAAC;gBACT,UAAU,CAAC,IAAI,GAAG,IAAI,CAAC;YACzB,CAAC;YAED,MAAM,IAAI,GAA4B;gBACpC,MAAM,EAAE;oBACN,KAAK,EAAE,WAAW;oBAClB,QAAQ,EAAE,QAAQ;iBACnB;gBACD,OAAO,EAAE;oBACP,SAAS,EAAE,SAAS,aAAT,SAAS,cAAT,SAAS,GAAI,IAAI;oBAC5B,cAAc,EAAE,cAAc,aAAd,cAAc,cAAd,cAAc,GAAI,KAAK;iBACxC;aACF,CAAC;YAEF,IAAI,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACvC,IAAI,CAAC,YAAY,CAAC,GAAG,UAAU,CAAC;YAClC,CAAC;YAED,MAAM,SAAS,GAAG,MAAM,IAAA,yBAAgB,EAEtC;gBACA,IAAI,EAAE,OAAO,CAAC,IAAc;gBAC5B,KAAK,EAAE,KAAK;gBACZ,IAAI;aACL,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YAEnE,OAAO;gBACL,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,KAAK,EAAE,MAAM,CAAC,KAAK;gBACnB,GAAG,EAAE,MAAM,CAAC,GAAG;gBACf,UAAU,EAAE,SAAS;aACtB,CAAC;QACJ,CAAC;KAAA;CACF,CAAC,CAAC"}

package/src/lib/actions/image-classification.d.ts

@@ -0,0 +1,13 @@
+export declare const imageClassification: import("@taskmagic/pieces-framework").IAction<import("@taskmagic/pieces-framework").SecretTextProperty<true>, {
+    classificationMode: import("@taskmagic/pieces-framework").StaticDropdownProperty<string, true>;
+    useCase: import("@taskmagic/pieces-framework").StaticDropdownProperty<string, true>;
+    model: import("@taskmagic/pieces-framework").DropdownProperty<string, true>;
+    imageSource: import("@taskmagic/pieces-framework").StaticDropdownProperty<string, true>;
+    imageFile: import("@taskmagic/pieces-framework").FileProperty<true>;
+    imageUrl: import("@taskmagic/pieces-framework").ShortTextProperty<true>;
+    customCategories: import("@taskmagic/pieces-framework").ArrayProperty<true>;
+    hypothesisTemplate: import("@taskmagic/pieces-framework").ShortTextProperty<false>;
+    topK: import("@taskmagic/pieces-framework").NumberProperty<false>;
+    confidenceThreshold: import("@taskmagic/pieces-framework").NumberProperty<false>;
+    outputFormat: import("@taskmagic/pieces-framework").StaticDropdownProperty<string, false>;
+}>;