Update generation.py

generation.py

@@ -1,7 +1,7 @@
 import os
 import requests
 import random
-import base64
+import re
 from datetime import datetime
 from typing import Optional
 
@@ -11,37 +11,70 @@ os.makedirs(OUTPUT_DIR, exist_ok=True)
 def generate_image_from_prompt(
     prompt: str,
     negative_prompt: str = "",
-    model_name: str = "black-forest-labs/flux-schnell", # Modelo ultra rápido y barato en OpenRouter
+    model_name: str = "ignored", # Ignoramos lo que venga de fuera, usaremos el bueno aquí
     seed: Optional[int] = None,
 ) -> tuple[Optional[str], str]:
     try:
-        api_key = os.getenv("OPENROUTER_API_KEY").strip()
+        # 1. Limpieza de Clave (Seguridad contra espacios invisibles)
+        api_key = os.getenv("OPENROUTER_API_KEY")
         if not api_key:
             return None, "❌ Falta OPENROUTER_API_KEY en los Secrets."
+        api_key = api_key.strip()
+
+        # 2. Configuración del modelo real de OpenRouter
+        # Flux Schnell es muy barato y rápido
+        real_model = "black-forest-labs/flux-schnell"
+
+        print(f"Enviando prompt a OpenRouter: {prompt}")
 
-        # Petición a OpenRouter
         response = requests.post(
             url="https://openrouter.ai/api/v1/chat/completions",
             headers={
                 "Authorization": f"Bearer {api_key}",
                 "Content-Type": "application/json",
+                "HTTP-Referer": "https://huggingface.co", # Requerido por OpenRouter
+                "X-Title": "Sofia AI Studio",
             },
             json={
-                "model": model_name,
-                "messages": [{"role": "user", "content": prompt}]
+                "model": real_model,
+                "messages": [
+                    {"role": "user", "content": prompt}
+                ]
             }
         )
         
+        # 3. Procesar respuesta
+        if response.status_code != 200:
+            return None, f"❌ Error OpenRouter ({response.status_code}): {response.text}"
+
         data = response.json()
         
-        # OpenRouter suele devolver la imagen en el contenido si es un modelo de imagen
-        if "choices" in data:
-            content = data["choices"][0]["message"]["content"]
-            # Aquí la lógica varía según si devuelve URL o B64, 
-            # pero este paso confirmará si tu saldo y conexión funcionan.
-            return None, f"✅ ¡Conectado! OpenRouter respondió: {content[:100]}"
+        # OpenRouter con Flux suele devolver la URL dentro del contenido del mensaje
+        # A veces viene como markdown: ![image](url)
+        content = data['choices'][0]['message']['content']
+        
+        # Buscar la URL de la imagen usando una expresión regular
+        url_match = re.search(r'\((https://.*?)\)', content) # Busca (https://...)
+        if not url_match:
+             # Si no está entre paréntesis, buscamos directamente http
+             url_match = re.search(r'(https://[^\s]+)', content)
+             
+        if url_match:
+            image_url = url_match.group(1)
+            # Descargar la imagen al disco
+            img_data = requests.get(image_url).content
+            
+            timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+            if seed is None: seed = random.randint(0, 9999)
+            filename = f"sofia_{timestamp}_{seed}.png"
+            file_path = os.path.join(OUTPUT_DIR, filename)
+            
+            with open(file_path, 'wb') as f:
+                f.write(img_data)
+                
+            return file_path, f"✅ ¡ÉXITO! Imagen generada con {real_model}"
         else:
-            return None, f"❌ Error OpenRouter: {data.get('error', 'Error desconocido')}"
+            return None, f"⚠️ OpenRouter respondió texto pero no encontré imagen: {content}"
 
     except Exception as e:
-        return None, f"❌ Error técnico: {str(e)}"
+        return None, f"❌ Error técnico crítico: {str(e)}"