import { 
  MessageSquare, 
  ImageIcon, 
  Video, 
  Mic, 
  Sparkles,
  LucideIcon
} from 'lucide-react';

// 模型分类图标映射
export const categoryIcons: Record<string, LucideIcon> = {
  '文本': MessageSquare,
  '图象': ImageIcon,
  '视频': Video,
  '语音': Mic,
};

// 模型分类到分类的映射
export const getModelCategory = (category: number): '文本' | '图象' | '视频' | '语音' => {
  switch (category) {
    case 0: return '文本';      // LLM
    case 1: return '文本';      // 多模态
    case 2: return '语音';      // TTS
    case 3: return '语音';      // STT
    case 4: return '图象';      // 生图
    case 5: return '视频';      // 生视频
    default: return '文本';
  }
};

// 根据模型分类获取图标
export const getCategoryIcon = (category: number): LucideIcon => {
  const categoryName = getModelCategory(category);
  return categoryIcons[categoryName] || Sparkles;
};

// 导出所有图标
export {
  MessageSquare,
  ImageIcon,
  Video,
  Mic,
  Sparkles
};