forked from innovacion/playground
Initial commit
This commit is contained in:
Sebastian
180
frontend/lib/google-ai.ts
Normal file
180
frontend/lib/google-ai.ts
Normal file
@@ -0,0 +1,180 @@
|
||||
// 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 },
|
||||
];
|
||||
}
|
||||
Reference in New Issue
Block a user