.

app6.py

@@ -3,29 +3,29 @@ import torch
 from diffusers import DiffusionPipeline
 from PIL import Image
 
-# --- Configuración del Modelo (Sencilla) ---
+# --- Configuración del Modelo Qwen (Forzado a CPU) ---
 
-# Forzamos el dispositivo a CPU (para evitar errores de CUDA/VRAM)
+# Establecemos el dispositivo a CPU (CUDA no está disponible en tu entorno).
 device = "cpu"
-dtype_config = torch.float32 
+dtype_config = torch.float32 # Usamos float32 para la CPU
 
-# Modelo Image-to-Image LucyEdit
-model_id = "kpsss34/LucyEdit_hybrid" 
+# Modelo Qwen para edición de imagen
+model_id = "Qwen/Qwen-Image-Edit-2509" 
 
 pipe = None
 try:
-    # Carga sencilla, sin device_map, sin aceleración avanzada
+    # Carga sencilla, forzada a CPU. Qwen no requiere dtype especial ni device_map si no hay GPU.
     pipe = DiffusionPipeline.from_pretrained(
         model_id, 
         torch_dtype=dtype_config, 
         use_safetensors=True
     ).to(device) 
     
-    print("✅ Modelo LucyEdit cargado con éxito en la CPU.")
+    print("✅ Modelo Qwen-Image-Edit cargado con éxito en la CPU.")
 
 except Exception as e:
-    print(f"❌ Error CRÍTICO al cargar el modelo: {e}")
-    print("El modelo NO ha podido cargarse. Comprueba tu conexión o memoria RAM.")
+    print(f"❌ Error CRÍTICO al cargar el modelo Qwen: {e}")
+    print("El modelo NO ha podido cargarse. Podría ser un problema de memoria RAM, incluso con CPU.")
 
 
 # --- Función de Procesamiento con Difusión (i2i) ---
@@ -35,33 +35,33 @@ def procesar_con_difusion(imagen_entrada, estilo_radial):
     """
     # Manejo de error de carga
     if pipe is None:
-        # Devuelve el error si el modelo no se cargó correctamente
         return Image.new('RGB', (512, 512), color = 'red') 
 
     if imagen_entrada is None:
         return None 
     
-    # 1. Prompt Base Fijo (Lo que debe hacer el modelo)
-    prompt_base = "fotografía de alta calidad, retrato detallado, con estilo" 
+    # 1. Prompt Base Fijo
+    prompt_base = "fotografía de alta calidad, retrato detallado" 
     
     estilo_prompts = {
         # El prompt fijo + el modificador del estilo
-        "Blanco y Negro (Monocromático)": ", monocromático, alto contraste, dramático, blanco y negro",
-        "Alto Contraste y Saturación": ", colores vívidos, alto contraste, HDR, saturación extrema",
-        "Original (Poco Ruido)": ", fotografía de alta calidad, realista, colores naturales, sutil",
+        "Blanco y Negro (Monocromático)": ", monocromático, alto contraste, dramático, blanco y negro, película antigua",
+        "Alto Contraste y Saturación": ", colores vívidos, alto contraste, saturación extrema, cinematográfico, iluminación fuerte",
+        "Original (Poco Ruido)": ", fotografía de alta calidad, realista, colores naturales, sutil, cinematic lighting",
     }
     
     full_prompt = prompt_base + estilo_prompts.get(estilo_radial, "")
 
-    # 2. Fuerza de Difusión FIJA
-    # Un valor alto (0.85) para asegurar que el efecto (B/N o contraste) se aplique bien.
-    STRENGTH_FIJA = 0.85 
+    # 2. Fuerza de Difusión FIJA (Valor alto para asegurar el efecto)
+    STRENGTH_FIJA = 0.9 
     
-    # 3. Preprocesar la imagen (LucyEdit trabaja en 512x512)
+    # 3. Preprocesar la imagen (Ajustamos el tamaño de entrada)
+    # Qwen-Image-Edit usa 512x512
     init_image = imagen_entrada.convert("RGB").resize((512, 512))
     
     try:
         # 4. Ejecutar el pipeline de difusión i2i
+        # El modelo Qwen también acepta los parámetros strength y guidance_scale
         image = pipe(
             prompt=full_prompt, 
             image=init_image, 
@@ -77,11 +77,12 @@ def procesar_con_difusion(imagen_entrada, estilo_radial):
 
 
 # --- Interfaz Gradio con gr.Blocks() ---
-with gr.Blocks(title="LucyEdit Estilos Fijos") as demo:
+with gr.Blocks(title="Qwen Image Edit Estilos Fijos") as demo:
     gr.Markdown(
         """
-        # 🖼️ Tarea de Difusión (Image-to-Image) con LucyEdit
-        Carga una imagen y selecciona un **Estilo Radial** (Blanco y Negro o Contraste) para que el modelo de difusión la transforme. 
+        # 🖼️ Tarea de Difusión (Image-to-Image) con Qwen Edit
+        Carga una imagen y selecciona un **Estilo Radial** para que el modelo Qwen la transforme. 
+        **Nota:** La generación puede ser lenta al ejecutarse en CPU.
         """
     )
     with gr.Row():
@@ -97,7 +98,7 @@ with gr.Blocks(title="LucyEdit Estilos Fijos") as demo:
                 value="Blanco y Negro (Monocromático)"
             )
             
-            process_button = gr.Button("✨ Aplicar Difusión", variant="primary")
+            process_button = gr.Button("✨ Aplicar Difusión Qwen", variant="primary")
             
         with gr.Column(scale=1):
             image_output = gr.Image(