Add 6 new high-quality models and improve FLUX authentication - Add OpenJourney v4, Anything v3, Waifu Diffusion, Deliberate v2, Dreamlike Diffusion - Improve HF_TOKEN detection (supports both HF_TOKEN and HUGGING_FACE_HUB_TOKEN) - Add conditional FLUX models loading (only if token is available) - Add specific configurations for each new model - Improve logging and error handling - Update README with new models and better documentation

README.md

@@ -30,6 +30,13 @@ Un generador de imágenes con IA con múltiples modelos de alta calidad optimiza
 - **SDXL Base** - Modelo base de Stable Diffusion XL
 - **Realistic Vision** - Especializado en retratos realistas
 
+### Modelos Especializados 🎯
+- **OpenJourney v4** - Especializado en arte Midjourney-style
+- **Anything v3** - Modelo versátil para cualquier estilo
+- **Waifu Diffusion** - Especializado en anime y manga
+- **Deliberate v2** - Modelo de alta calidad para arte digital
+- **Dreamlike Diffusion** - Especializado en arte surrealista y onírico
+
 ## 🔐 Configuración para Modelos Gated (FLUX)
 
 Para usar los modelos FLUX, necesitas configurar tu token de Hugging Face:
@@ -72,18 +79,38 @@ Para usar los modelos FLUX, necesitas configurar tu token de Hugging Face:
 - **Configuraciones específicas**: Cada modelo tiene sus propios parámetros de guidance y steps
 - **API Externa**: FLUX.1-Kontext-Dev usa la API del Space oficial
 - **Autenticación automática**: Manejo automático de tokens para modelos gated
+- **Detección inteligente**: Los modelos FLUX solo aparecen si hay token configurado
 
 ## 🔧 Configuraciones por Modelo
 
+### Modelos Rápidos
 - **Turbo/Lightning**: guidance_scale=0.0, steps=1-4
-- **FLUX**: guidance_scale=7.5, steps=20
+
+### Modelos FLUX
+- **FLUX.1-dev**: guidance_scale=7.5, steps=20
+- **FLUX.1-schnell**: guidance_scale=7.5, steps=20
 - **FLUX.1-Kontext-Dev**: guidance_scale=2.5, steps=28 (via API)
+
+### Modelos Especializados
 - **Realistic Vision**: guidance_scale=7.5, steps=25
+- **OpenJourney v4**: guidance_scale=7.5, steps=20
+- **Anything v3**: guidance_scale=7.5, steps=20
+- **Waifu Diffusion**: guidance_scale=7.5, steps=20
+- **Deliberate v2**: guidance_scale=7.5, steps=20
+- **Dreamlike Diffusion**: guidance_scale=7.5, steps=20
 
 ## 🌐 Modelo Externo
 
 **FLUX.1-Kontext-Dev** se conecta a la API oficial del Space `black-forest-labs/FLUX.1-Kontext-Dev` para generar imágenes de alta calidad. Este modelo requiere conexión a internet y puede tener tiempos de respuesta variables.
 
+## 🎨 Ejemplos de Prompts
+
+- **Anime**: "Anime girl with blue hair, detailed, high quality"
+- **Cyberpunk**: "Cyberpunk city at night, neon lights, detailed, 8k"
+- **Fantasy**: "Fantasy landscape with mountains and dragons, epic, detailed"
+- **Realistic**: "Portrait of a beautiful woman, realistic, high quality, detailed"
+- **Sci-fi**: "Futuristic AI assistant in a glowing galaxy, neon lights, sci-fi style, cinematic"
+
 ## 🐛 Solución de Problemas
 
 ### Error de Autenticación
@@ -91,12 +118,18 @@ Si ves errores como "Cannot access gated repo":
 1. Verifica que el `HF_TOKEN` esté configurado correctamente
 2. Asegúrate de tener acceso a los modelos FLUX
 3. Reinicia el Space después de configurar el token
+4. Los modelos FLUX solo aparecen en el dropdown si hay token configurado
 
 ### Error de API Externa
 Si la API externa falla:
 - El modelo FLUX.1-Kontext-Dev mostrará una imagen roja de error
 - Usa los modelos locales como alternativa
 
+### Modelos No Disponibles
+Si no ves los modelos FLUX en el dropdown:
+- Verifica que el `HF_TOKEN` esté configurado
+- Los modelos gated solo aparecen con autenticación válida
+
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -14,11 +14,12 @@ import torch
 from huggingface_hub import login
 
 # Configurar autenticación con Hugging Face
-HF_TOKEN = os.getenv("HF_TOKEN")
+HF_TOKEN = os.getenv("HF_TOKEN") or os.getenv("HUGGING_FACE_HUB_TOKEN")
 if HF_TOKEN:
     try:
         login(token=HF_TOKEN)
         print("✅ Autenticado con Hugging Face")
+        print(f"🔑 Token configurado: {HF_TOKEN[:10]}...")
     except Exception as e:
         print(f"⚠️ Error de autenticación: {e}")
 else:
@@ -30,16 +31,30 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 # Modelos disponibles de alta calidad
 MODELS = {
     "SDXL Turbo (stabilityai/sdxl-turbo)": "stabilityai/sdxl-turbo",
-    "FLUX.1-dev (black-forest-labs/FLUX.1-dev)": "black-forest-labs/FLUX.1-dev",
-    "FLUX.1-schnell (black-forest-labs/FLUX.1-schnell)": "black-forest-labs/FLUX.1-schnell",
-    "FLUX.1-Kontext-Dev (API External)": "api_external",
     "SDXL Lightning (ByteDance/SDXL-Lightning)": "ByteDance/SDXL-Lightning",
     "SDXL Lightning 4Step (ByteDance/SDXL-Lightning-4Step)": "ByteDance/SDXL-Lightning-4Step",
     "SD Turbo (stabilityai/sd-turbo)": "stabilityai/sd-turbo",
     "SDXL Base (stabilityai/stable-diffusion-xl-base-1.0)": "stabilityai/stable-diffusion-xl-base-1.0",
-    "Realistic Vision (SG161222/Realistic_Vision_V5.1_noVAE)": "SG161222/Realistic_Vision_V5.1_noVAE"
+    "Realistic Vision (SG161222/Realistic_Vision_V5.1_noVAE)": "SG161222/Realistic_Vision_V5.1_noVAE",
+    "OpenJourney v4 (prompthero/openjourney-v4)": "prompthero/openjourney-v4",
+    "Anything v3 (Linaqruf/anything-v3.0)": "Linaqruf/anything-v3.0",
+    "Waifu Diffusion (hakurei/waifu-diffusion)": "hakurei/waifu-diffusion",
+    "Deliberate v2 (XpucT/deliberate-v2)": "XpucT/deliberate-v2",
+    "Dreamlike Diffusion (dreamlike-art/dreamlike-diffusion-1.0)": "dreamlike-art/dreamlike-diffusion-1.0",
+    "FLUX.1-Kontext-Dev (API External)": "api_external",
 }
 
+# Modelos FLUX (solo si hay token)
+if HF_TOKEN:
+    FLUX_MODELS = {
+        "FLUX.1-dev (black-forest-labs/FLUX.1-dev)": "black-forest-labs/FLUX.1-dev",
+        "FLUX.1-schnell (black-forest-labs/FLUX.1-schnell)": "black-forest-labs/FLUX.1-schnell",
+    }
+    MODELS.update(FLUX_MODELS)
+    print("🔓 Modelos FLUX habilitados con autenticación")
+else:
+    print("🔒 Modelos FLUX deshabilitados - requiere HF_TOKEN")
+
 # Estado del pipeline
 pipe = None
 current_model_id = None
@@ -58,13 +73,15 @@ def load_model(model_id):
     if pipe is None or model_id != current_model_id:
         try:
             # Usar token de autenticación si está disponible
-            if HF_TOKEN:
+            if HF_TOKEN and ("flux" in model_id.lower() or "black-forest" in model_id.lower()):
+                print(f"🔐 Cargando modelo gated: {model_id}")
                 pipe = DiffusionPipeline.from_pretrained(
                     model_id, 
                     torch_dtype=torch_dtype,
                     use_auth_token=HF_TOKEN
                 )
             else:
+                print(f"📦 Cargando modelo público: {model_id}")
                 pipe = DiffusionPipeline.from_pretrained(
                     model_id, 
                     torch_dtype=torch_dtype
@@ -79,6 +96,7 @@ def load_model(model_id):
 # Función para usar la API externa de FLUX.1-Kontext-Dev
 def use_external_api(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
     try:
+        print("🌐 Conectando a API externa FLUX.1-Kontext-Dev...")
         # URL de la API del Space externo
         api_url = "https://black-forest-labs-flux-1-kontext-dev.hf.space/api/predict/"
         
@@ -113,12 +131,13 @@ def use_external_api(prompt, negative_prompt, seed, randomize_seed, width, heigh
             image_bytes = base64.b64decode(image_data.split(',')[1])
             image = Image.open(io.BytesIO(image_bytes))
             
+            print("✅ API externa exitosa")
             return image, new_seed
         else:
             raise Exception(f"API Error: {response.status_code} - {response.text}")
             
     except Exception as e:
-        print(f"Error usando API externa: {e}")
+        print(f"❌ Error usando API externa: {e}")
         # Fallback: crear una imagen de error
         error_image = Image.new('RGB', (width, height), color='red')
         return error_image, seed
@@ -163,6 +182,26 @@ def infer(
             # Realistic Vision
             guidance_scale = 7.5
             num_inference_steps = 25
+        elif "openjourney" in model_id.lower():
+            # OpenJourney
+            guidance_scale = 7.5
+            num_inference_steps = 20
+        elif "anything" in model_id.lower():
+            # Anything v3
+            guidance_scale = 7.5
+            num_inference_steps = 20
+        elif "waifu" in model_id.lower():
+            # Waifu Diffusion
+            guidance_scale = 7.5
+            num_inference_steps = 20
+        elif "deliberate" in model_id.lower():
+            # Deliberate v2
+            guidance_scale = 7.5
+            num_inference_steps = 20
+        elif "dreamlike" in model_id.lower():
+            # Dreamlike Diffusion
+            guidance_scale = 7.5
+            num_inference_steps = 20
 
         image = pipe(
             prompt=prompt,
@@ -177,7 +216,7 @@ def infer(
         return image, seed
         
     except Exception as e:
-        print(f"Error en inferencia: {e}")
+        print(f"❌ Error en inferencia: {e}")
         # Crear imagen de error
         error_image = Image.new('RGB', (width, height), color='red')
         return error_image, seed
@@ -189,6 +228,9 @@ examples = [
     "A delicious ceviche cheesecake slice",
     "Futuristic AI assistant in a glowing galaxy, neon lights, sci-fi style, cinematic",
     "Portrait of a beautiful woman, realistic, high quality, detailed",
+    "Anime girl with blue hair, detailed, high quality",
+    "Cyberpunk city at night, neon lights, detailed, 8k",
+    "Fantasy landscape with mountains and dragons, epic, detailed",
 ]
 
 css = """
@@ -220,7 +262,7 @@ with gr.Blocks(css=css) as demo:
                 choices=list(MODELS.keys()),
                 value=list(MODELS.keys())[0],
                 label="Model",
-                info="Select a high-quality model (FLUX.1-Kontext-Dev uses external API)"
+                info="Select a high-quality model (FLUX models require HF_TOKEN)"
             )
             
             negative_prompt = gr.Text(