Add FLUX.1-Kontext-Dev external API integration - Add external API connection to black-forest-labs/FLUX.1-Kontext-Dev - Add fallback error handling for API calls - Update README with external model documentation

README.md

@@ -24,6 +24,7 @@ Un generador de imágenes con IA con múltiples modelos de alta calidad optimiza
 ### Modelos FLUX (Alta Calidad)
 - **FLUX.1-dev** - Modelo de desarrollo de alta calidad
 - **FLUX.1-schnell** - Modelo rápido de alta calidad
+- **FLUX.1-Kontext-Dev** - Modelo externo via API (requiere conexión)
 
 ### Modelos Estándar
 - **SDXL Base** - Modelo base de Stable Diffusion XL
@@ -43,13 +44,19 @@ Un generador de imágenes con IA con múltiples modelos de alta calidad optimiza
 - **Configuraciones automáticas**: Cada modelo usa sus parámetros óptimos
 - **Carga bajo demanda**: Los modelos se cargan solo cuando se necesitan
 - **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
 
 ## 🔧 Configuraciones por Modelo
 
 - **Turbo/Lightning**: guidance_scale=0.0, steps=1-4
 - **FLUX**: guidance_scale=7.5, steps=20
+- **FLUX.1-Kontext-Dev**: guidance_scale=2.5, steps=28 (via API)
 - **Realistic Vision**: guidance_scale=7.5, steps=25
 
+## 🌐 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.
+
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -1,6 +1,11 @@
 import gradio as gr
 import numpy as np
 import random
+import requests
+import base64
+from PIL import Image
+import io
+import json
 
 # import spaces #[uncomment to use ZeroGPU]
 from diffusers import DiffusionPipeline
@@ -13,6 +18,7 @@ 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",
@@ -40,6 +46,53 @@ def load_model(model_id):
         pipe = pipe.to(device)
         current_model_id = 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:
+        # URL de la API del Space externo
+        api_url = "https://black-forest-labs-flux-1-kontext-dev.hf.space/api/predict/"
+        
+        # Crear una imagen base simple para la API (requiere input_image)
+        base_image = Image.new('RGB', (width, height), color='white')
+        img_byte_arr = io.BytesIO()
+        base_image.save(img_byte_arr, format='PNG')
+        img_byte_arr = img_byte_arr.getvalue()
+        
+        # Preparar los datos para la API
+        files = {
+            'data': (None, json.dumps([
+                base64.b64encode(img_byte_arr).decode('utf-8'),  # input_image
+                prompt,  # prompt
+                seed,  # seed
+                randomize_seed,  # randomize_seed
+                guidance_scale,  # guidance_scale
+                num_inference_steps  # steps
+            ]))
+        }
+        
+        # Hacer la petición a la API
+        response = requests.post(api_url, files=files)
+        
+        if response.status_code == 200:
+            result = response.json()
+            # La API devuelve [image_data, seed]
+            image_data = result['data'][0]
+            new_seed = result['data'][1]
+            
+            # Decodificar la imagen
+            image_bytes = base64.b64decode(image_data.split(',')[1])
+            image = Image.open(io.BytesIO(image_bytes))
+            
+            return image, new_seed
+        else:
+            raise Exception(f"API Error: {response.status_code}")
+            
+    except Exception as 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
+
 # @spaces.GPU #[uncomment to use ZeroGPU]
 def infer(
     prompt,
@@ -53,6 +106,10 @@ def infer(
     model_name,
     progress=gr.Progress(track_tqdm=True),
 ):
+    # Verificar si es el modelo externo
+    if model_name == "FLUX.1-Kontext-Dev (API External)":
+        return use_external_api(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps)
+    
     # Cargar el modelo seleccionado
     model_id = MODELS[model_name]
     load_model(model_id)
@@ -126,7 +183,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"
+                info="Select a high-quality model (FLUX.1-Kontext-Dev uses external API)"
             )
             
             negative_prompt = gr.Text(