Auto-sync from website-creator

skills/image-generation/SKILL.md

@@ -0,0 +1,61 @@
+---
+name: image-generation
+description: Generate images from text prompts using Chutes AI image generation. Use this skill when the user wants to create AI-generated images from descriptions.
+---
+
+# Image Generation
+
+Generate AI images from text prompts via `python3 scripts/image_gen.py generate <prompt> [options]`.
+
+## Commands
+
+| Command | Args | Description |
+|---------|------|-------------|
+| `generate` | `<prompt> [--width W] [--height H] [--steps N] [--seed N]` | Generate image from prompt |
+
+## Options
+
+| Option | Default | Range | Description |
+|--------|---------|-------|-------------|
+| `--width` | 1024 | 576-2048 | Image width in pixels |
+| `--height` | 1024 | 576-2048 | Image height in pixels |
+| `--steps` | 9 | 1-100 | Number of inference steps |
+| `--seed` | null | 0-4294967295 | Random seed (null = random) |
+| `--guidance-scale` | 0 | 0-5 | Guidance scale for generation |
+| `--shift` | 3 | 1-10 | Shift parameter |
+| `--max-seq-len` | 512 | 256-2048 | Max sequence length |
+
+## Examples
+
+```bash
+# Basic generation
+python3 scripts/image_gen.py generate "a high quality photo of a sunrise over the mountains"
+
+# Custom dimensions
+python3 scripts/image_gen.py generate "a futuristic city at night" --width 1280 --height 720
+
+# With seed for reproducibility
+python3 scripts/image_gen.py generate "a cute cat sitting on a windowsill" --seed 42
+
+# High quality with more steps
+python3 scripts/image_gen.py generate "a detailed portrait of a woman in renaissance style" --steps 20
+```
+
+## Workflow
+
+1. Run `generate` with your prompt
+2. Script saves image as `generated_[timestamp].png`
+3. Returns image path: `generated_1234567890.png [12345]`
+
+## Output Format
+
+- Success: `Image saved: filename.png [id]`
+- Error: `Error: message` (to stderr)
+- Images saved to current working directory as PNG files
+
+## Notes
+
+- Requires `CHUTES_API_TOKEN` in environment
+- Prompt length: 3-1200 characters
+- Large images (2048x2048) take longer to generate
+- Images are saved locally, not returned as base64 to save memory

skills/image-generation/scripts/.env.example

@@ -0,0 +1,7 @@
+# Chutes AI API Token
+# Get your token from your Chutes AI account
+#
+# WARNING: Never commit this file with actual credentials!
+# Keep your .env file private and add it to .gitignore
+
+CHUTES_API_TOKEN=your_chutes_api_token_here

skills/image-generation/scripts/image_gen.py

@@ -0,0 +1,160 @@
+#!/usr/bin/env python3
+
+import os
+import sys
+import argparse
+import time
+from pathlib import Path
+import requests
+import base64
+
+
+def load_env():
+    env_path = Path(__file__).parent / ".env"
+    if env_path.exists():
+        for line in env_path.read_text().splitlines():
+            line = line.strip()
+            if line and not line.startswith("#") and "=" in line:
+                k, v = line.split("=", 1)
+                os.environ.setdefault(k.strip(), v.strip().strip("\"'"))
+
+
+load_env()
+
+API_TOKEN = os.environ.get("CHUTES_API_TOKEN")
+API_URL = "https://chutes-z-image-turbo.chutes.ai/generate"
+
+
+def generate_image(
+    prompt,
+    width=1024,
+    height=1024,
+    steps=9,
+    seed=None,
+    guidance_scale=0,
+    shift=3,
+    max_seq_len=512,
+):
+    if not API_TOKEN:
+        print("Error: CHUTES_API_TOKEN not set in environment", file=sys.stderr)
+        sys.exit(1)
+
+    if not prompt or len(prompt) < 3:
+        print("Error: Prompt must be at least 3 characters", file=sys.stderr)
+        sys.exit(1)
+    if len(prompt) > 1200:
+        print(
+            "Error: Prompt exceeds maximum length of 1200 characters", file=sys.stderr
+        )
+        sys.exit(1)
+
+    payload = {
+        "prompt": prompt,
+        "width": width,
+        "height": height,
+        "num_inference_steps": steps,
+        "guidance_scale": guidance_scale,
+        "shift": shift,
+        "max_sequence_length": max_seq_len,
+        "seed": seed,
+    }
+
+    try:
+        headers = {
+            "Authorization": f"Bearer {API_TOKEN}",
+            "Content-Type": "application/json",
+        }
+
+        response = requests.post(API_URL, headers=headers, json=payload, timeout=300)
+        response.raise_for_status()
+
+        content_type = response.headers.get("Content-Type", "")
+
+        if "image/" in content_type:
+            image_bytes = response.content
+        else:
+            result = response.json()
+            if isinstance(result, list) and len(result) > 0:
+                item = result[0]
+                image_data = item.get("data", "")
+                if image_data.startswith("data:image"):
+                    image_bytes = base64.b64decode(image_data.split(",", 1)[1])
+                else:
+                    image_bytes = base64.b64decode(image_data)
+            else:
+                print("Error: Invalid response format", file=sys.stderr)
+                sys.exit(1)
+
+        timestamp = int(time.time())
+        filename = f"generated_{timestamp}.png"
+
+        with open(filename, "wb") as f:
+            f.write(image_bytes)
+
+        print(f"Image saved: {filename} [{timestamp}]")
+
+    except requests.exceptions.RequestException as e:
+        print(f"Error: API request failed - {e}", file=sys.stderr)
+        sys.exit(1)
+    except Exception as e:
+        print(f"Error: {e}", file=sys.stderr)
+        sys.exit(1)
+
+
+def main():
+    parser = argparse.ArgumentParser(description="Generate images from text prompts")
+    parser.add_argument("prompt", help="Text prompt for image generation")
+    parser.add_argument(
+        "--width", type=int, default=1024, help="Image width (576-2048)"
+    )
+    parser.add_argument(
+        "--height", type=int, default=1024, help="Image height (576-2048)"
+    )
+    parser.add_argument("--steps", type=int, default=9, help="Inference steps (1-100)")
+    parser.add_argument("--seed", type=int, default=None, help="Random seed")
+    parser.add_argument(
+        "--guidance-scale", type=float, default=0, help="Guidance scale (0-5)"
+    )
+    parser.add_argument("--shift", type=float, default=3, help="Shift parameter (1-10)")
+    parser.add_argument(
+        "--max-seq-len", type=int, default=512, help="Max sequence length (256-2048)"
+    )
+
+    args = parser.parse_args()
+
+    if not (576 <= args.width <= 2048):
+        print("Error: width must be between 576 and 2048", file=sys.stderr)
+        sys.exit(1)
+    if not (576 <= args.height <= 2048):
+        print("Error: height must be between 576 and 2048", file=sys.stderr)
+        sys.exit(1)
+    if not (1 <= args.steps <= 100):
+        print("Error: steps must be between 1 and 100", file=sys.stderr)
+        sys.exit(1)
+    if args.seed is not None and not (0 <= args.seed <= 4294967295):
+        print("Error: seed must be between 0 and 4294967295", file=sys.stderr)
+        sys.exit(1)
+    if not (0 <= args.guidance_scale <= 5):
+        print("Error: guidance-scale must be between 0 and 5", file=sys.stderr)
+        sys.exit(1)
+    if not (1 <= args.shift <= 10):
+        print("Error: shift must be between 1 and 10", file=sys.stderr)
+        sys.exit(1)
+    if not (256 <= args.max_seq_len <= 2048):
+        print("Error: max-seq-len must be between 256 and 2048", file=sys.stderr)
+        sys.exit(1)
+
+    generate_image(
+        prompt=args.prompt,
+        width=args.width,
+        height=args.height,
+        steps=args.steps,
+        seed=args.seed,
+        guidance_scale=args.guidance_scale,
+        shift=args.shift,
+        max_seq_len=args.max_seq_len,
+    )
+
+
+if __name__ == "__main__":
+    main()

skills/image-generation/scripts/requirements.txt

@@ -0,0 +1 @@
+requests>=2.28.0