import { ProviderInfo } from "../types/heatmap";

// Function to generate evenly distributed colors
function generateUniqueColors(count: number): string[] {
  const colors: string[] = [];
  
  for (let i = 0; i < count; i++) {
    // Use golden ratio for better distribution
    const goldenRatio = 0.618033988749895;
    const hue = (i * goldenRatio * 360) % 360;
    
    // Vary saturation and lightness for more variety
    const saturation = 65 + (i % 4) * 8; // 65-89%
    const lightness = 45 + ((i % 3) * 7); // 45-59%
    
    // Convert HSL to RGB
    const h = hue / 360;
    const s = saturation / 100;
    const l = lightness / 100;
    
    const hue2rgb = (p: number, q: number, t: number) => {
      if (t < 0) t += 1;
      if (t > 1) t -= 1;
      if (t < 1/6) return p + (q - p) * 6 * t;
      if (t < 1/2) return q;
      if (t < 2/3) return p + (q - p) * (2/3 - t) * 6;
      return p;
    };
    
    const q = l < 0.5 ? l * (1 + s) : l + s - l * s;
    const p = 2 * l - q;
    
    const r = Math.round(hue2rgb(p, q, h + 1/3) * 255);
    const g = Math.round(hue2rgb(p, q, h) * 255);
    const b = Math.round(hue2rgb(p, q, h - 1/3) * 255);
    
    colors.push(`#${r.toString(16).padStart(2, '0')}${g.toString(16).padStart(2, '0')}${b.toString(16).padStart(2, '0')}`);
  }
  
  return colors;
}

// Generate 94 unique colors
const uniqueColors = generateUniqueColors(94);

export const ORGANIZATIONS: ProviderInfo[] = [
  { color: uniqueColors[0], authors: ["openfree","VIDraft","seawolf2357","ginigen","ginipick","fantaxy","fantos","aiqtech","aiqcamp","Heartsync"] },  
  { color: uniqueColors[1], authors: ["kakaobrain"] },
  { color: uniqueColors[2], authors: ["NCSOFT"] },
  { color: uniqueColors[3], authors: ["upstage"] },
  { color: uniqueColors[4], authors: ["moreh"] },
  { color: uniqueColors[5], authors: ["LGAI-EXAONE"] },
  { color: uniqueColors[6], authors: ["naver-hyperclovax"] },
  { color: uniqueColors[7], authors: ["OnomaAIResearch"] },
  { color: uniqueColors[8], authors: ["kakaocorp"] },
  { color: uniqueColors[9], authors: ["Baidu","PaddlePaddle"] },
  { color: uniqueColors[10], authors: ["tencent"] },
  { color: uniqueColors[11], authors: ["BAAI"] },
  { color: uniqueColors[12], authors: ["OpenGVLab","InternLM"] },
  { color: uniqueColors[13], authors: ["Skywork"] },
  { color: uniqueColors[14], authors: ["MiniMaxAI"] },
  { color: uniqueColors[15], authors: ["stepfun-ai"] },
  { color: uniqueColors[16], authors: ["ByteDance","Bytedance Seed","bytedance-research"]},
  { color: uniqueColors[17], authors: ["openbmb"] },
  { color: uniqueColors[18], authors: ["THUDM"] },
  { color: uniqueColors[19], authors: ["rednote-hilab"] },
  { color: uniqueColors[20], authors: ["deepseek-ai"] },
  { color: uniqueColors[21], authors: ["Qwen", "wan-ai"]},
  { color: uniqueColors[22], authors: ["XiaomiMiMo"] },
  { color: uniqueColors[23], authors: ["IndexTeam"] },
  { color: uniqueColors[24], authors: ["John6666"] },
  { color: uniqueColors[25], authors: ["Nymbo"] },  
  { color: uniqueColors[26], authors: ["MaziyarPanahi"] },
  { color: uniqueColors[27], authors: ["prithivMLmods"] },
  { color: uniqueColors[28], authors: ["fffiloni"] },
  { color: uniqueColors[29], authors: ["gokaygokay"] },
  { color: uniqueColors[30], authors: ["enzostvs"] },  
  { color: uniqueColors[31], authors: ["black-forest-labs"] },
  { color: uniqueColors[32], authors: ["lllyasviel"] },
  { color: uniqueColors[33], authors: ["briaai"] },
  { color: uniqueColors[34], authors: ["multimodalart"] },
  { color: uniqueColors[35], authors: ["unsloth"] },
  { color: uniqueColors[36], authors: ["Xenova"] },
  { color: uniqueColors[37], authors: ["mistralai"] },
  { color: uniqueColors[38], authors: ["meta-llama","facebook"] },
  { color: uniqueColors[39], authors: ["openai"] },
  { color: uniqueColors[40], authors: ["Anthropic"] },
  { color: uniqueColors[41], authors: ["google"] },
  { color: uniqueColors[42], authors: ["allenai"] },
  { color: uniqueColors[43], authors: ["apple"] },
  { color: uniqueColors[44], authors: ["microsoft"] },
  { color: uniqueColors[45], authors: ["nvidia"] },
  { color: uniqueColors[46], authors: ["CohereLabs"] },
  { color: uniqueColors[47], authors: ["ibm-granite"] },
  { color: uniqueColors[48], authors: ["stabilityai"] },
  { color: uniqueColors[49], authors: ["huggingface", "OpenEvals", "HuggingFaceTB","HuggingFaceH4", "HuggingFaceM4", "HuggingFaceFW", "HuggingFaceFV","open-r1","parler-tts","nanotron","lerobot","distilbert"] },
  { color: uniqueColors[50], authors: ["agents-course"] },
  { color: uniqueColors[51], authors: ["Agents-MCP-Hackathon"] },
  { color: uniqueColors[52], authors: ["akhaliq"] },
  { color: uniqueColors[53], authors: ["alexnasa"] },
  { color: uniqueColors[54], authors: ["Alibaba-NLP"] },
  { color: uniqueColors[55], authors: ["ArtificialAnalysis"] },
  { color: uniqueColors[56], authors: ["bartowski"] },
  { color: uniqueColors[57], authors: ["bibibi12345"] },
  { color: uniqueColors[58], authors: ["calcuis"] },
  { color: uniqueColors[59], authors: ["ChenDY"] },
  { color: uniqueColors[60], authors: ["city96"] },
  { color: uniqueColors[61], authors: ["Comfy-Org"] },
  { color: uniqueColors[62], authors: ["fancyfeast"] },
  { color: uniqueColors[63], authors: ["fal"] },
  { color: uniqueColors[64], authors: ["hf-audio"] },
  { color: uniqueColors[65], authors: ["huggingface-projects"] },
  { color: uniqueColors[66], authors: ["hysts"] },
  { color: uniqueColors[67], authors: ["intfloat"] },
  { color: uniqueColors[68], authors: ["jbilcke-hf"] },
  { color: uniqueColors[69], authors: ["jinaai"] },
  { color: uniqueColors[70], authors: ["Kwai-Kolors"] },
  { color: uniqueColors[71], authors: ["kyutai"] },
  { color: uniqueColors[72], authors: ["leonelhs"] },
  { color: uniqueColors[73], authors: ["Menlo"] },
  { color: uniqueColors[74], authors: ["merve"] },
  { color: uniqueColors[75], authors: ["mlabonne"] },
  { color: uniqueColors[76], authors: ["MohamedRashad"] },
  { color: uniqueColors[77], authors: ["mrfakename"] },
  { color: uniqueColors[78], authors: ["ostris"] },
  { color: uniqueColors[79], authors: ["QuantStack"] },
  { color: uniqueColors[80], authors: ["r3gm"] },
  { color: uniqueColors[81], authors: ["Salesforce"] },
  { color: uniqueColors[82], authors: ["sentence-transformers"] },
  { color: uniqueColors[83], authors: ["Shakker-Labs"] },
  { color: uniqueColors[84], authors: ["skytnt"] },
  { color: uniqueColors[85], authors: ["strangerzonehf"] },
  { color: uniqueColors[86], authors: ["TheStinger"] },
  { color: uniqueColors[87], authors: ["tonyassi"] },
  { color: uniqueColors[88], authors: ["VAST-AI"] },
  { color: uniqueColors[89], authors: ["victor"] },
  { color: uniqueColors[90], authors: ["webml-community"] },
  { color: uniqueColors[91], authors: ["WeShopAI"] },
  { color: uniqueColors[92], authors: ["XLabs-AI"] },
  { color: uniqueColors[93], authors: ["zama-fhe"] },
];