Update generation.py

generation.py

@@ -2,50 +2,44 @@ import os
 import random
 from datetime import datetime
 from typing import Optional
-
 from huggingface_hub import InferenceClient
 
 # Directorio donde se guardan las imágenes generadas
 OUTPUT_DIR = "generated_images"
 os.makedirs(OUTPUT_DIR, exist_ok=True)
 
-# Cliente de inferencia (igual que en Sofia Rivera)
+# Cliente de inferencia
+# Si tienes un Token de HF en tus Secrets, lo detectará automáticamente
 client = InferenceClient()
 
 def generate_image_from_prompt(
     prompt: str,
     negative_prompt: str = "",
-    model_name: str = "black-forest-labs/FLUX.1-dev",
+    model_name: str = "black-forest-labs/FLUX.1-schnell", # CAMBIADO A VERSIÓN GRATUITA
     seed: Optional[int] = None,
 ) -> tuple[Optional[str], str]:
-    """
-    Genera una imagen usando Hugging Face InferenceClient.text_to_image
-    y la guarda en OUTPUT_DIR.
-
-    Devuelve (image_path, status_message).
-    Si hay error, image_path = None y status_message contiene el error.
-    """
     try:
         if seed is None:
             seed = random.randint(0, 2_147_483_647)
-
+        
+        # Parámetros optimizados para el modelo gratuito FLUX Schnell
         image = client.text_to_image(
             prompt=prompt,
-            negative_prompt=negative_prompt,
             model=model_name,
-            guidance_scale=7.5,
-            num_inference_steps=50,
+            seed=seed,
+            num_inference_steps=4, # Schnell solo necesita 4 pasos (es muy rápido)
+            # guidance_scale se omite en Schnell para evitar errores de API gratuita
         )
 
         timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
         filename = f"sofia_{timestamp}_{seed}.png"
         file_path = os.path.join(OUTPUT_DIR, filename)
-
+        
         image.save(file_path)
-
-        status = f"✅ Imagen generada y guardada: {filename}\nModelo: {model_name}\nSeed: {seed}"
+        
+        status = f"✅ ¡Sofía está lista!\nModelo: {model_name}\nSeed: {seed}"
         return file_path, status
 
     except Exception as e:
         error_msg = f"❌ Error al generar imagen: {str(e)}"
-        return None, error_msg
+        return None, error_msg