Proeycto de images-worker creado

mvp/image-worker/src/pipeline/image-generator.service.ts

@@ -0,0 +1,84 @@
+import { Injectable, Logger } from '@nestjs/common';
+import { ConfigService } from '@nestjs/config';
+import axios from 'axios';
+
+const OPENROUTER_URL = 'https://openrouter.ai/api/v1/chat/completions';
+
+@Injectable()
+export class ImageGeneratorService {
+  private readonly logger = new Logger(ImageGeneratorService.name);
+
+  constructor(private readonly config: ConfigService) {}
+
+  async generarRender(prompt: string, fotoAntesDataUri: string): Promise<string> {
+    const apiKey = this.config.get<string>('OPENROUTER_API_KEY');
+    const model = this.config.get<string>('OPENROUTER_MODEL_IMAGEN', 'google/gemini-2.0-flash-exp-image-generation');
+
+    const intentosRateLimit = 1;
+    for (let attempt = 0; attempt <= intentosRateLimit; attempt++) {
+      try {
+        const response = await axios.post(
+          OPENROUTER_URL,
+          {
+            model,
+            messages: [
+              {
+                role: 'user',
+                content: [
+                  { type: 'text', text: prompt },
+                  { type: 'image_url', image_url: { url: fotoAntesDataUri } },
+                ],
+              },
+            ],
+          },
+          {
+            headers: {
+              Authorization: `Bearer ${apiKey}`,
+              'Content-Type': 'application/json',
+              'HTTP-Referer': 'https://reformix.es',
+              'X-Title': 'Reformix Image Worker',
+            },
+            timeout: 60000,
+          },
+        );
+
+        const content = response.data.choices?.[0]?.message?.content;
+        if (!content) throw new Error('OpenRouter no devolvio contenido');
+
+        const imagen = this.extraerImagenDeRespuesta(content, response.data);
+        if (!imagen) throw new Error('No se pudo extraer imagen de la respuesta');
+
+        return imagen;
+      } catch (err: any) {
+        if (err.response?.status === 429 && attempt < intentosRateLimit) {
+          this.logger.warn('Rate limit (429), esperando 5s y reintentando...');
+          await new Promise((r) => setTimeout(r, 5000));
+          continue;
+        }
+        throw err;
+      }
+    }
+
+    throw new Error('Fallaron todos los intentos de generacion de imagen');
+  }
+
+  private extraerImagenDeRespuesta(content: string, rawResponse?: any): string | null {
+    if (content.startsWith('data:image')) return content;
+
+    const dataUriMatch = content.match(/data:image\/[a-zA-Z]+;base64,[^\s"']+/);
+    if (dataUriMatch) return dataUriMatch[0];
+
+    const urlMatch = content.match(/https?:\/\/[^\s"'()]+\.(png|jpg|jpeg|webp)/i);
+    if (urlMatch) return urlMatch[0];
+
+    const parts = rawResponse?.choices?.[0]?.message?.content;
+    if (Array.isArray(parts)) {
+      for (const part of parts) {
+        if (part.type === 'image_url' && part.image_url?.url) return part.image_url.url;
+        if (part.image_url?.url?.startsWith('data:image')) return part.image_url.url;
+      }
+    }
+
+    return null;
+  }
+}