Update generation.py

generation.py

@@ -1,43 +1,47 @@
 import os
+import requests
 import random
+import base64
 from datetime import datetime
 from typing import Optional
-from huggingface_hub import InferenceClient
 
 OUTPUT_DIR = "generated_images"
 os.makedirs(OUTPUT_DIR, exist_ok=True)
 
-# Usaremos un modelo de la comunidad que suele ser gratuito
-client = InferenceClient(token=os.getenv("HF_TOKEN"))
-
 def generate_image_from_prompt(
     prompt: str,
-    negative_prompt: str = "blurry, bad quality, distorted, ugly",
-    model_name: str = "SG161222/RealVisXL_V4.0_Lightning", # Modelo experto en realismo GRATIS
+    negative_prompt: str = "",
+    model_name: str = "black-forest-labs/flux-schnell", # Modelo ultra rápido y barato en OpenRouter
     seed: Optional[int] = None,
 ) -> tuple[Optional[str], str]:
     try:
-        if seed is None:
-            seed = random.randint(0, 2_147_483_647)
-        
-        # Este modelo funciona de maravilla con InferenceClient
-        image = client.text_to_image(
-            prompt=prompt,
-            model=model_name,
-            negative_prompt=negative_prompt,
-            num_inference_steps=8, # SDXL Lightning es muy rápido
-            guidance_scale=1.5,
-        )
+        api_key = os.getenv("OPENROUTER_API_KEY")
+        if not api_key:
+            return None, "❌ Falta OPENROUTER_API_KEY en los Secrets."
 
-        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-        filename = f"sofia_{timestamp}_{seed}.png"
-        file_path = os.path.join(OUTPUT_DIR, filename)
+        # 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",
+            },
+            json={
+                "model": model_name,
+                "messages": [{"role": "user", "content": prompt}]
+            }
+        )
         
-        image.save(file_path)
+        data = response.json()
         
-        status = f"✅ ¡Sofía generada con éxito!\nModelo: RealVisXL (Gratis)\nSeed: {seed}"
-        return file_path, status
+        # 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]}"
+        else:
+            return None, f"❌ Error OpenRouter: {data.get('error', 'Error desconocido')}"
 
     except Exception as e:
-        error_msg = f"❌ Error: {str(e)}"
-        return None, error_msg
+        return None, f"❌ Error técnico: {str(e)}"