import { GoogleGenerativeAI } from '@google/generative-ai';

const genAI = new GoogleGenerativeAI(process.env.GEMINI_API_KEY!);
const model = genAI.getGenerativeModel({ model: 'gemini-pro' });

// Generate color palette
const generatePalette = async (description: string) => {
  const prompt = `
    Generate a harmonious color palette based on: "${description}"
    
    Return 5 colors in this JSON format:
    {
      "colors": [
        {"hex": "#RRGGBB", "name": "descriptive name"},
        ...
      ],
      "reasoning": "why this palette works"
    }
  `;
  
  const result = await model.generateContent(prompt);
  const text = result.response.text();
  return JSON.parse(extractJSON(text));
};

// Analyze color harmony
const analyzeHarmony = async (colors: string[]) => {
  const prompt = `
    Analyze this color palette: ${colors.join(', ')}
    
    Provide:
    1. Harmony type (complementary, analogous, etc.)
    2. Mood/emotion conveyed
    3. Suggested use cases
    4. Potential improvements
  `;
  
  const result = await model.generateContent(prompt);
  return result.response.text();
};

const applyChromaticAberration = (
  ctx: CanvasRenderingContext2D,
  imageData: ImageData,
  offset: number
) => {
  const { data, width, height } = imageData;
  const output = ctx.createImageData(width, height);
  
  for (let y = 0; y < height; y++) {
    for (let x = 0; x < width; x++) {
      const i = (y * width + x) * 4;
      
      // Red channel: shift right
      const redX = Math.min(x + offset, width - 1);
      const redI = (y * width + redX) * 4;
      output.data[i] = data[redI];
      
      // Green channel: no shift
      output.data[i + 1] = data[i + 1];
      
      // Blue channel: shift left
      const blueX = Math.max(x - offset, 0);
      const blueI = (y * width + blueX) * 4;
      output.data[i + 2] = data[blueI + 2];
      
      // Alpha: unchanged
      output.data[i + 3] = data[i + 3];
    }
  }
  
  ctx.putImageData(output, 0, 0);
};

{
  "colors": [
    {"hex": "#FF006E", "name": "Neon Pink"},
    {"hex": "#8338EC", "name": "Electric Purple"},
    {"hex": "#00F5FF", "name": "Cyan Glow"},
    {"hex": "#0A0E27", "name": "Deep Night"},
    {"hex": "#FFBE0B", "name": "Golden Accent"}
  ],
  "reasoning": "This palette captures the essence of cyberpunk Tokyo with high-contrast neon colors (pink, purple, cyan) against a dark base. The golden accent represents street lights and signage, creating depth and visual interest."
}