playground/frontend/lib/google-ai.ts

// Cliente: Solo tipos y constantes (sin módulos de Node.js)

// Modelos disponibles para generación de imágenes con Gemini
export const geminiImageModels = [
  {
    id: 'gemini-2.5-flash-image',
    name: 'Gemini 2.5 Flash Image',
    description: 'Modelo optimizado para generación de imágenes',
    supportsImages: true,
    capabilities: ['text-to-image', 'image-generation'],
  },
  {
    id: 'gemini-2.0-flash-exp',
    name: 'Gemini 2.0 Flash Experimental',
    description: 'Modelo experimental con generación de imágenes rápida',
    supportsImages: true,
    capabilities: ['text-to-image', 'image-editing'],
  },
  {
    id: 'gemini-exp-1206',
    name: 'Gemini Experimental 1206',
    description: 'Versión experimental con capacidades mejoradas',
    supportsImages: true,
    capabilities: ['text-to-image', 'multimodal'],
  },
  {
    id: 'gemini-3-pro-image-preview',
    name: 'Gemini 3 Pro Image Preview',
    description: 'Vista previa de Gemini 3 Pro para generación de imágenes',
    supportsImages: true,
    capabilities: ['text-to-image', 'image-generation', 'high-quality'],
  },
  {
    id: 'gemini-1.5-pro',
    name: 'Gemini 1.5 Pro',
    description: 'Modelo más capaz para tareas complejas',
    supportsImages: true,
    capabilities: ['multimodal', 'long-context'],
  },
  {
    id: 'gemini-1.5-flash',
    name: 'Gemini 1.5 Flash',
    description: 'Modelo rápido y eficiente',
    supportsImages: true,
    capabilities: ['multimodal', 'fast'],
  },
] as const;

export type GeminiImageModel = typeof geminiImageModels[number];

// Modelos disponibles para generación de videos con Veo
export const veoVideoModels = [
  {
    id: 'veo-2.0-generate-001',
    name: 'Veo 2.0 Generate',
    description: 'Modelo base para generación de videos',
    supportsVideo: true,
    capabilities: ['text-to-video', 'video-generation'],
  },
  {
    id: 'veo-2.0-generate-exp',
    name: 'Veo 2.0 Generate Experimental',
    description: 'Versión experimental de Veo 2.0',
    supportsVideo: true,
    capabilities: ['text-to-video', 'video-generation'],
  },
  {
    id: 'veo-2.0-generate-preview',
    name: 'Veo 2.0 Generate Preview',
    description: 'Vista previa de Veo 2.0',
    supportsVideo: true,
    capabilities: ['text-to-video', 'video-generation'],
  },
  {
    id: 'veo-3.0-generate-001',
    name: 'Veo 3.0 Generate',
    description: 'Modelo de última generación para videos',
    supportsVideo: true,
    capabilities: ['text-to-video', 'video-generation', 'high-quality'],
  },
  {
    id: 'veo-3.0-fast-generate-001',
    name: 'Veo 3.0 Fast Generate',
    description: 'Generación rápida de videos con Veo 3.0',
    supportsVideo: true,
    capabilities: ['text-to-video', 'video-generation', 'fast'],
  },
  {
    id: 'veo-3.1-generate-preview',
    name: 'Veo 3.1 Generate Preview',
    description: 'Vista previa de Veo 3.1',
    supportsVideo: true,
    capabilities: ['text-to-video', 'video-generation', 'latest'],
  },
  {
    id: 'veo-3.1-fast-generate-preview',
    name: 'Veo 3.1 Fast Generate Preview',
    description: 'Generación rápida con Veo 3.1 preview',
    supportsVideo: true,
    capabilities: ['text-to-video', 'video-generation', 'fast', 'latest'],
  },
  {
    id: 'test',
    name: 'Vtest',
    description: 'Test para probar la api de gemini',
    supportsVideo: true,
    capabilities: ['text-to-video', 'video-generation', 'fast', 'latest'],
  },
] as const;

export type VeoVideoModel = typeof veoVideoModels[number];

// Configuración de dimensiones por aspect ratio
export const aspectRatioSizes: Record<string, { width: number; height: number }> = {
  '1:1': { width: 1024, height: 1024 },
  '3:4': { width: 768, height: 1024 },
  '4:3': { width: 1024, height: 768 },
  '9:16': { width: 576, height: 1024 },
  '16:9': { width: 1024, height: 576 },
};

// Configuración de duración de videos
export const videoDurations = [
  { value: 5, label: '5 segundos' },
  { value: 8, label: '8 segundos' },
  { value: 10, label: '10 segundos' },
] as const;

// Tipos para la configuración
export interface ImageGenerationConfig {
  model: string;
  prompt: string;
  aspectRatio?: string;
  numberOfImages?: number;
  negativePrompt?: string;
  seed?: number;
  guidanceScale?: number;
  temperature?: number;
  safetyFilterLevel?: 'block_none' | 'block_some' | 'block_most';
  referenceImages?: Array<{
    data: string; // Base64
    mimeType: string;
  }>;
}

export interface VideoGenerationConfig {
  model: string;
  prompt: string;
  aspectRatio?: string;
  duration?: number; // en segundos
  negativePrompt?: string;
  seed?: number;
  temperature?: number;
  safetyFilterLevel?: 'block_none' | 'block_some' | 'block_most';
}

// Helper para obtener información del modelo
export function getModelInfo(modelId: string) {
  return geminiImageModels.find(m => m.id === modelId);
}

// Helper para validar aspect ratio
export function isValidAspectRatio(modelId: string, aspectRatio: string): boolean {
  return true; // Gemini acepta cualquier aspect ratio
}

// Safety settings helpers
export function getSafetySettings(level: 'block_none' | 'block_some' | 'block_most') {
  const threshold =
    level === 'block_none' ? 'BLOCK_NONE' :
    level === 'block_some' ? 'BLOCK_ONLY_HIGH' :
    'BLOCK_MEDIUM_AND_ABOVE';

  return [
    { category: 'HARM_CATEGORY_HATE_SPEECH', threshold },
    { category: 'HARM_CATEGORY_DANGEROUS_CONTENT', threshold },
    { category: 'HARM_CATEGORY_SEXUALLY_EXPLICIT', threshold },
    { category: 'HARM_CATEGORY_HARASSMENT', threshold },
  ];
}