Update app.py

app.py

@@ -1,139 +1,227 @@
-import random
-import datetime
-import json
 import os
 import base64
 import requests
 import time
+import concurrent.futures
+import random
+import datetime
+import json
 from io import BytesIO
 from PIL import Image
 import gradio as gr
-from openai import OpenAI
+
+# Configuración para Hugging Face
+HF_SPACE = os.getenv('SPACE_ID') is not None
 
 # --------------------------
-# Configuración Sambanova Maverick AI
+# CONFIGURACIÓN SAMBANOVA - MODELO ÚNICO
 # --------------------------
-api_key_sambanova = os.getenv("SAMBANOVA_API_KEY")
-client_sambanova = OpenAI(api_key=api_key_sambanova, base_url="https://api.sambanova.ai/v1")
+try:
+    from openai import OpenAI
+    SAMBANOVA_AVAILABLE = True
+except ImportError:
+    SAMBANOVA_AVAILABLE = False
 
-MODELS = {
-    "general_fast": "Meta-Llama-3.1-8B-Instruct",
-    "general_smart": "Meta-Llama-3.3-70B-Instruct",
-    "coding_expert": "DeepSeek-V3.1",
-    "coding_alt": "DeepSeek-V3-0324",
-    "massive_brain": "gpt-oss-120b",
-    "specialized_1": "DeepSeek-V3.1-Terminus",
-    "specialized_2": "Llama-3.3-Swallow-70B-Instruct-v0.4",
-    "multilingual": "Qwen3-32B",
-    "arabic_special": "ALLaM-7B-Instruct-preview",
-    "vision_expert": "Llama-4-Maverick-17B-128E-Instruct"
-}
+SAMBANOVA_API_KEY = os.getenv("SAMBANOVA_API_KEY")
+if SAMBANOVA_AVAILABLE and SAMBANOVA_API_KEY:
+    client_sambanova = OpenAI(api_key=SAMBANOVA_API_KEY, base_url="https://api.sambanova.ai/v1")
+    SAMBANOVA_ENABLED = True
+else:
+    client_sambanova = None
+    SAMBANOVA_ENABLED = False
 
-def select_best_model(user_prompt):
-    router_system_prompt = (
-        "You are an AI router. Analyze the user's request and select the best ID from this list:\n"
-        f"{list(MODELS.values())}\n"
-        "Rules:\n"
-        "- For coding/math: Choose 'DeepSeek-V3.1' or 'DeepSeek-V3-0324'.\n"
-        "- For complex logic: Choose 'Meta-Llama-3.3-70B-Instruct' or 'gpt-oss-120b'.\n"
-        "- For Japanese/Cultural: Choose 'Llama-3.3-Swallow-70B-Instruct-v0.4'.\n"
-        "- For Arabic/Middle East: Choose 'ALLaM-7B-Instruct-preview'.\n"
-        "- For General/Fast chat: Choose 'Meta-Llama-3.1-8B-Instruct'.\n"
-        "- For Multilingual/Chinese: Choose 'Qwen3-32B'.\n"
-        "Reply ONLY with the exact model name strings from the list. Nothing else."
-    )
+# MODELO ÚNICO PRINCIPAL - Llama 4 Maverick para todo
+PRIMARY_MODEL = "Llama-4-Maverick-17B-128E-Instruct"
+
+def enhance_prompt_with_ai(base_prompt, enhancement_type="creative"):
+    """
+    Mejora el prompt usando Llama 4 Maverick según el tipo de mejora
+    """
+    if not SAMBANOVA_ENABLED:
+        return base_prompt
+    
+    enhancement_instructions = {
+        "creative": """
+        Como experto en creación de prompts para generación de imágenes, mejora este prompt 
+        para hacerlo más visualmente impactante, detallado y efectivo.
+        
+        INSTRUCCIONES:
+        1. Añade detalles específicos de iluminación, texturas, colores y composición
+        2. Incluye terminología fotográfica profesional (depth of field, bokeh, lighting ratios, etc.)
+        3. Mantén el aspecto ratio 9:16 y formato vertical
+        4. Enriquece la descripción de personajes, ambiente y emociones
+        5. Asegura realismo fotográfico con detalles técnicos de cámara
+        6. Optimiza para modelos de generación de imágenes como REVE CREATE
+        
+        Responde SOLO con el prompt mejorado, sin explicaciones.
+        """,
+        
+        "technical": """
+        Como técnico especializado en prompts para AI, optimiza este prompt 
+        enfocándote en aspectos técnicos y de realismo.
+        
+        INSTRUCCIONES:
+        1. Especifica detalles técnicos de cámara, lentes y configuración
+        2. Añade parámetros de iluminación profesional
+        3. Describe texturas, materiales y superficies con precisión
+        4. Incluye efectos visuales y post-procesado realista
+        5. Mantén coherencia técnica y física
+        
+        Responde SOLO con el prompt mejorado.
+        """,
+        
+        "artistic": """
+        Como artista conceptual, transforma este prompt en una descripción 
+        artística y cinematográfica.
+        
+        INSTRUCCIONES:
+        1. Enfócate en atmósfera, mood y emociones
+        2. Usa lenguaje cinematográfico y composición artística
+        3. Añade elementos de narrativa visual
+        4. Describe colores, luces y sombras de forma poética pero precisa
+        5. Crea una escena con impacto visual fuerte
+        
+        Responde SOLO con el prompt mejorado.
+        """
+    }
+    
+    instruction = enhancement_instructions.get(enhancement_type, enhancement_instructions["creative"])
+    
     try:
         response = client_sambanova.chat.completions.create(
-            model="Meta-Llama-3.1-8B-Instruct",
+            model=PRIMARY_MODEL,
             messages=[
-                {"role": "system", "content": router_system_prompt},
-                {"role": "user", "content": f"User request: {user_prompt}"}
+                {"role": "system", "content": "Eres un especialista en creación y optimización de prompts para generación de imágenes AI."},
+                {"role": "user", "content": f"{instruction}\n\nPROMPT ORIGINAL:\n{base_prompt}"}
             ],
-            temperature=0.0,
-            max_tokens=50
+            temperature=0.4,
+            max_tokens=800,
+            top_p=0.9
         )
-        selected_model = response.choices[0].message.content.strip().replace('"', '').replace("'", "")
-        if selected_model not in MODELS.values():
-            return "Meta-Llama-3.3-70B-Instruct"
-        return selected_model
+        enhanced = response.choices[0].message.content.strip()
+        return enhanced if enhanced and len(enhanced) > 50 else base_prompt
     except Exception as e:
-        print(f"Error en router Maverick: {e}")
-        return "Meta-Llama-3.3-70B-Instruct"
+        print(f"Error mejorando prompt: {e}")
+        return base_prompt
 
-def smart_prompt_suggestions(input_text):
-    target_model = select_best_model(input_text)
-    messages = [
-        {"role": "system", "content": "You are a helpful assistant."},
-        {"role": "user", "content": input_text}
-    ]
+def describe_image_with_ai(image_path, description_style="detailed"):
+    """
+    Genera una descripción de imagen usando Llama 4 Maverick
+    """
+    if not SAMBANOVA_ENABLED:
+        return "⚠️ Servicio de descripción no disponible. Configura SAMBANOVA_API_KEY."
+    
+    if not image_path or not os.path.exists(image_path):
+        return "❌ Imagen no encontrada para describir."
+    
+    style_instructions = {
+        "detailed": "Describe la imagen en detalle incluyendo composición, colores, iluminación, personajes y ambiente.",
+        "technical": "Analiza la imagen desde perspectiva técnica: composición, iluminación, estilo fotográfico y elementos visuales.",
+        "prompt_style": "Crea un prompt de generación de imágenes basado en esta imagen, en estilo detallado y técnico."
+    }
+    
+    instruction = style_instructions.get(description_style, style_instructions["detailed"])
+    
+    try:
+        # Simulamos una descripción basada en el nombre del archivo
+        description_prompt = f"""
+        {instruction}
+        
+        Información de la imagen: {os.path.basename(image_path)}
+        
+        Genera una descripción detallada y útil.
+        """
+        
+        response = client_sambanova.chat.completions.create(
+            model=PRIMARY_MODEL,
+            messages=[
+                {"role": "system", "content": "Eres un experto en análisis y descripción de imágenes."},
+                {"role": "user", "content": description_prompt}
+            ],
+            temperature=0.3,
+            max_tokens=500
+        )
+        
+        description = response.choices[0].message.content.strip()
+        return description
+        
+    except Exception as e:
+        return f"❌ Error en descripción de imagen: {str(e)}"
+
+def generate_prompt_from_description(user_description, style="realistic"):
+    """
+    Genera un prompt completo a partir de una descripción del usuario
+    """
+    if not SAMBANOVA_ENABLED:
+        return "⚠️ Servicio de generación no disponible. Configura SAMBANOVA_API_KEY."
+    
+    style_prompts = {
+        "realistic": "Genera un prompt detallado para creación de imágenes realistas y fotográficas",
+        "cinematic": "Crea un prompt con estilo cinematográfico y dramático",
+        "artistic": "Diseña un prompt artístico y creativo para generación de imágenes",
+        "professional": "Elabora un prompt profesional para fotografía de alta calidad"
+    }
+    
+    base_instruction = style_prompts.get(style, style_prompts["realistic"])
+    
     try:
-        stream = client_sambanova.chat.completions.create(
-            model=target_model,
-            messages=messages,
-            temperature=0.2,
-            top_p=0.9,
-            stream=True
+        system_prompt = """
+        Eres un creador de prompts experto para generación de imágenes AI. 
+        Crea prompts detallados, técnicos y visualmente ricos que incluyan:
+        - Composición y encuadre
+        - Iluminación y atmosfera
+        - Detalles técnicos de cámara
+        - Descripción de personajes y ambiente
+        - Estilo y mood
+        - Formato 9:16 vertical
+        """
+        
+        response = client_sambanova.chat.completions.create(
+            model=PRIMARY_MODEL,
+            messages=[
+                {"role": "system", "content": system_prompt},
+                {"role": "user", "content": f"{base_instruction} basado en esta descripción: {user_description}"}
+            ],
+            temperature=0.5,
+            max_tokens=700,
+            top_p=0.85
         )
-        response_text = ""
-        for chunk in stream:
-            content = chunk.choices[0].delta.content
-            if content:
-                response_text += content
-        return response_text
+        
+        generated_prompt = response.choices[0].message.content.strip()
+        return generated_prompt
+        
     except Exception as e:
-        return f"Error en Maverick IA: {str(e)}"
+        return f"❌ Error generando prompt: {str(e)}"
 
 # --------------------------
-# GENERADOR PROMPTS SIN ETNIAS
+# GENERADOR PRINCIPAL DE PROMPTS
 # --------------------------
 class HyperRealisticConfig:
-    BODY_TYPES = [
-        "athletic slender", "voluptuous curves", "petite feminine",
-        "toned hourglass", "soft natural", "elegant tall"
-    ]
-    SKIN_DETAILS = [
-        "with subtle freckles and skin texture", "with dewy natural complexion",
-        "showing realistic skin pores and veins", "with natural skin imperfections",
-        "with subsurface scattering effect", "with delicate skin translucency"
-    ]
-    LACE_BIKINI_STYLES = [
-        "delicate black Chantilly lace bikini with floral embroidery and sheer panels",
-        "sheer nude illusion bikini with scalloped lace edges and ribbon ties"
-    ]
-    HOSIERY_STYLES = [
-        "black sheer thigh-high stockings with lace tops and stay-up silicone",
-        "nude ultra-sheer stockings with reinforced toes and back seam"
-    ]
-    HEEL_STYLES = [
-        "black patent leather stilettos with pointed toe and slim heel",
-        "nude pumps with platform and ankle strap for stability"
-    ]
-    HAIRSTYLES = [
-        "long wavy hair with natural volume and soft highlights",
-        "sleek straight hair with middle part and glossy finish"
-    ]
-    MAKEUP_STYLES = [
-        "natural glam with dewy skin, soft contour, and nude lips",
-        "smoky eye with winged liner and voluminous lashes"
-    ]
-    LIGHTING_DETAILS = [
-        "soft diffused lighting with a warm golden glow",
-        "dramatic chiaroscuro lighting with deep shadows"
-    ]
-    PHOTOGRAPHY_STYLES = [
-        "cinematic lighting with rim light and soft shadows",
-        "natural window light with lens flare and soft focus"
-    ]
-    CAMERAS = [
-        "Canon EOS R5", "Sony A7R IV", "Nikon Z7 II"
-    ]
-    LENSES = [
-        "Canon RF 85mm f/1.2L", "Sony FE 50mm f/1.2 GM"
-    ]
     PROFESSIONAL_ROLES = [
-        {"role": "Executive Secretary", "uniform": "form-fitting tailored black dress with silk blouse slightly unbuttoned", "environment": "luxurious corporate office with marble floors", "accessories": "slim tablet and leather portfolio", "pose": "bending over conference table arranging documents", "accidental_hint": "causing the hem to naturally ride up just enough to reveal a delicate lace edge"},
-        {"role": "Luxury Hotel Manager", "uniform": "elegant navy blue blazer and pencil skirt that shifts slightly when seated", "environment": "five-star hotel lobby with golden lighting", "accessories": "gold name tag and master key cards", "pose": "leaning forward to assist guest at reception", "accidental_hint": "causing the skirt to hug the hips and subtly outline lace beneath"},
-        {"role": "Fashion Boutique Manager", "uniform": "chic black designer dress with elegant draping that accents curves", "environment": "high-end boutique with minimalist decor and spotlights", "accessories": "designer handbag and inventory clipboard", "pose": "kneeling to adjust mannequin display", "accidental_hint": "causing the fabric to stretch gently across the hips, revealing faint lace texture"}
+        {
+            "role": "Executive Secretary", 
+            "uniform": "form-fitting tailored black dress with silk blouse", 
+            "environment": "luxurious corporate office with marble floors", 
+            "accessories": "slim tablet and leather portfolio", 
+            "pose": "bending over conference table arranging documents", 
+            "accidental_hint": "causing the hem to naturally ride up just enough to reveal a delicate lace edge"
+        },
+        {
+            "role": "Luxury Hotel Manager", 
+            "uniform": "elegant navy blue blazer and pencil skirt", 
+            "environment": "five-star hotel lobby with golden lighting", 
+            "accessories": "gold name tag and master key cards", 
+            "pose": "leaning forward to assist guest at reception", 
+            "accidental_hint": "causing the skirt to hug the hips and subtly outline lace beneath"
+        },
+        {
+            "role": "Fashion Boutique Manager", 
+            "uniform": "chic black designer dress with elegant draping", 
+            "environment": "high-end boutique with minimalist decor", 
+            "accessories": "designer handbag and inventory clipboard", 
+            "pose": "kneeling to adjust mannequin display", 
+            "accidental_hint": "causing the fabric to stretch gently across the hips, revealing faint lace texture"
+        }
     ]
 
 class HyperRealisticPromptGenerator:
@@ -141,94 +229,80 @@ class HyperRealisticPromptGenerator:
         self.config = HyperRealisticConfig()
         self.history = []
 
-    def _random_style(self):
-        return (
-            random.choice(self.config.BODY_TYPES),
-            random.choice(self.config.SKIN_DETAILS),
-            random.choice(self.config.LACE_BIKINI_STYLES),
-            random.choice(self.config.HOSIERY_STYLES),
-            random.choice(self.config.HEEL_STYLES),
-            random.choice(self.config.HAIRSTYLES),
-            random.choice(self.config.MAKEUP_STYLES),
-            random.choice(self.config.LIGHTING_DETAILS),
-            random.choice(self.config.PHOTOGRAPHY_STYLES),
-            random.choice(self.config.CAMERAS),
-            random.choice(self.config.LENSES)
-        )
-
-    def generate_prompt(self, role_name: str, nsfw_mode: bool = False) -> str:
+    def generate_base_prompt(self, role_name: str, nsfw_mode: bool = False) -> str:
+        """Genera el prompt base sin mejoras de IA"""
         role = next((r for r in self.config.PROFESSIONAL_ROLES if r["role"] == role_name), None)
         if not role:
-            return "Error: Rol no encontrado."
-        body, skin, lace, hosiery, heels, hairstyle, makeup, lighting, photo_style, camera, lens = self._random_style()
-
-        underwear_desc = f"wearing delicate {lace} under {role['uniform']}" if not nsfw_mode else f"wearing {lace} under {role['uniform']}"
-        sensuality = "professional elegance with natural feminine grace" if not nsfw_mode else "artistic intimate moment with effortless natural beauty"
-
+            return "❌ Error: Rol no encontrado."
+        
         prompt = (
-            f"Hyper-realistic 4K photograph, 9:16 vertical full-body composition, dynamic low-angle view, "
-            f"of a stunning {role['role'].lower()} with {body} figure {skin}, "
-            f"{underwear_desc}, {hosiery}, {heels}, styled with {hairstyle}, and {makeup}. "
-            f"In {role['environment']}, captured while {role['pose']}, {role['accidental_hint']}. "
-            f"{lighting}, {photo_style}, Camera: {camera} with {lens} lens, ultra-high resolution 16K, "
-            f"extreme detail showing skin texture with visible pores, realistic fabric wrinkles, "
-            f"subsurface scattering effect, cinematic depth of field. Professional atmosphere, elegant composition, {sensuality}. "
-            f"Include {role['accessories']} in scene. --no explicit_nudity --style raw --stylize 150"
+            f"Hyper-realistic photograph, 9:16 vertical composition, "
+            f"professional {role['role'].lower()} in {role['environment']}, "
+            f"wearing {role['uniform']}, {role['pose']}, {role['accidental_hint']}. "
+            f"Professional atmosphere, elegant composition, cinematic lighting, "
+            f"ultra-high detail, realistic textures. Include {role['accessories']}. "
+            f"--style raw --stylize 150"
         )
 
         self.history.append({
-            "prompt": prompt, "role": role_name, "nsfw": nsfw_mode,
+            "prompt": prompt, 
+            "role": role_name, 
+            "nsfw": nsfw_mode,
             "timestamp": datetime.datetime.now().isoformat(),
-            "votes": 0, "notes": ""
+            "enhanced": False
         })
+        
+        if len(self.history) > 30:
+            self.history = self.history[-30:]
+            
         return prompt
 
-    def suggest_prompts(self, count=3):
-        if not self.history:
-            roles = [r["role"] for r in self.config.PROFESSIONAL_ROLES]
-            return [self.generate_prompt(random.choice(roles)) for _ in range(count)]
-        top_prompts = sorted(self.history, key=lambda x: x["votes"], reverse=True)[:count]
-        suggestions = []
-        for entry in top_prompts:
-            base_role = entry["role"]
-            suggestion = self.generate_prompt(base_role)
-            suggestions.append(suggestion)
-        return suggestions
-
-    def vote_prompt(self, index, vote_change=1):
-        if 0 <= index < len(self.history):
-            self.history[index]["votes"] += vote_change
-
-    def annotate_prompt(self, index, note_text):
-        if 0 <= index < len(self.history):
-            self.history[index]["notes"] = note_text
+    def generate_enhanced_prompt(self, role_name: str, nsfw_mode: bool = False, enhancement_type: str = "creative") -> str:
+        """Genera prompt base y luego lo mejora con IA"""
+        base_prompt = self.generate_base_prompt(role_name, nsfw_mode)
+        
+        if not SAMBANOVA_ENABLED:
+            return base_prompt
+        
+        enhanced_prompt = enhance_prompt_with_ai(base_prompt, enhancement_type)
+        
+        # Actualizar el historial con el prompt mejorado
+        if self.history:
+            self.history[-1]["prompt"] = enhanced_prompt
+            self.history[-1]["enhanced"] = True
+            self.history[-1]["enhancement_type"] = enhancement_type
+            
+        return enhanced_prompt
 
     def clear_history(self):
         self.history = []
 
     def get_history(self):
-        return self.history.copy()
+        return self.history.copy()# --------------------------
+# CONFIGURACIÓN REVE CREATE
+# --------------------------
+API_URL = "https://api.reve.com/v1/image/create"
+CARPETA_SALIDA = "generated_images"
 
-# -------------------------------
-# INTEGRACION API REVE CREATE (IMÁGENES)
-# -------------------------------
-API_URL_REVE = "https://api.aimlapi.com/v1/images/generations"
-IMAGES_FOLDER = "generated_images"
-if not os.path.exists(IMAGES_FOLDER):
-    os.makedirs(IMAGES_FOLDER)
+if not os.path.exists(CARPETA_SALIDA):
+    os.makedirs(CARPETA_SALIDA)
 
-def save_image(img, index):
-    timestamp = int(time.time())
-    filename = f"image_{timestamp}_{index}.png"
-    full_path = os.path.join(IMAGES_FOLDER, filename)
-    img.save(full_path)
-    return full_path
+def guardar_imagen_local(img, index):
+    try:
+        timestamp = int(time.time())
+        nombre_archivo = f"reve_{timestamp}_{index}.png"
+        ruta_completa = os.path.join(CARPETA_SALIDA, nombre_archivo)
+        img.save(ruta_completa)
+        return ruta_completa
+    except Exception as e:
+        print(f"⚠️ Error guardando imagen: {e}")
+        return None
 
-def call_image_api(prompt, api_key, index):
+def llamar_api(prompt, ratio, version, api_key, index):
     payload = {
-        "model": "reve/create-image",
         "prompt": prompt,
-        "aspect_ratio": "9:16"
+        "aspect_ratio": ratio,
+        "version": version
     }
     headers = {
         "Authorization": f"Bearer {api_key}",
@@ -236,83 +310,288 @@ def call_image_api(prompt, api_key, index):
         "Content-Type": "application/json"
     }
     try:
-        response = requests.post(API_URL_REVE, headers=headers, json=payload, timeout=60)
+        response = requests.post(API_URL, headers=headers, json=payload, timeout=120)
         if response.status_code == 200:
             data = response.json()
             if "image" in data:
                 img_bytes = base64.b64decode(data["image"])
                 img = Image.open(BytesIO(img_bytes))
-                save_image(img, index)
-                return img, None
-        return None, f"Error {response.status_code}: {response.text}"
+                saved_path = guardar_imagen_local(img, index)
+                return img, data.get('credits_used', 0), None, saved_path
+        return None, 0, f"Error {response.status_code}: {response.text}", None
     except Exception as e:
-        return None, str(e)
+        return None, 0, f"Excepción: {str(e)}", None
 
-def generate_images_batch(prompt, api_key, num_images):
-    images = []
-    errors = []
-    for i in range(num_images):
-        img, err = call_image_api(prompt, api_key, i)
-        if img:
-            images.append(img)
-        elif err:
-            errors.append(err)
-    return images, errors
+def generar_imagenes_batch(prompt, api_key_ui, ratio, version, num_imagenes):
+    api_key = api_key_ui.strip() if api_key_ui else os.getenv("REVE_API_KEY")
+    
+    if not api_key:
+        return [], "❌ ¡Falta la API Key de REVE CREATE!", []
+    
+    if not prompt or prompt.strip() == "":
+        return [], "❌ ¡El prompt está vacío!", []
+    
+    if HF_SPACE:
+        num_imagenes = min(num_imagenes, 4)
+    
+    imagenes_nuevas = []
+    errores = []
+    creditos_totales = 0
+    image_paths = []
+    
+    with concurrent.futures.ThreadPoolExecutor(max_workers=min(num_imagenes, 4)) as executor:
+        futuros = [
+            executor.submit(llamar_api, prompt, ratio, version, api_key, i) 
+            for i in range(num_imagenes)
+        ]
+        for i, futuro in enumerate(concurrent.futures.as_completed(futuros)):
+            img, creditos, error, img_path = futuro.result()
+            if img:
+                imagenes_nuevas.append(img)
+                creditos_totales += creditos
+                if img_path:
+                    image_paths.append(img_path)
+            if error:
+                errores.append(f"Imagen {i+1}: {error}")
+    
+    if imagenes_nuevas:
+        mensaje = f"✅ {len(imagenes_nuevas)} imagen(es) generada(s) | Créditos: {creditos_totales}"
+        if errores:
+            mensaje += f" | Errores: {len(errores)}"
+        return imagenes_nuevas, mensaje, image_paths
+    else:
+        error_msg = "❌ Error: " + "; ".join(errores[:3])
+        return [], error_msg, []
 
-# ------------------------------
-# INTERFAZ GRADIO COMPLETA
-# ------------------------------
+# --------------------------
+# INTERFAZ GRADIO OPTIMIZADA
+# --------------------------
 generator = HyperRealisticPromptGenerator()
 
-def generate_prompt_ui(role, nsfw):
-    return generator.generate_prompt(role, nsfw)
+def generar_todo(role, nsfw, api_key, ratio, version, cantidad, use_ai_enhancement, enhancement_type):
+    """Función principal que integra todo el flujo"""
+    if use_ai_enhancement and SAMBANOVA_ENABLED:
+        prompt = generator.generate_enhanced_prompt(role, nsfw, enhancement_type)
+        enhancement_status = f"🚀 Prompt MEJORADO ({enhancement_type})"
+    else:
+        prompt = generator.generate_base_prompt(role, nsfw)
+        enhancement_status = "📝 Prompt BÁSICO"
+    
+    imagenes, mensaje, image_paths = generar_imagenes_batch(prompt, api_key, ratio, version, cantidad)
+    
+    # Añadir información del enhancement al mensaje
+    final_message = f"{enhancement_status} | {mensaje}"
+    if not SAMBANOVA_ENABLED and use_ai_enhancement:
+        final_message += " | ⚠️ SAMBANOVA_API_KEY no configurada"
+    
+    return prompt, imagenes, final_message, image_paths
 
-def suggest_prompt_ui(user_input):
-    return smart_prompt_suggestions(user_input)
+def generate_from_description_ui(user_description, prompt_style):
+    """Genera prompt desde descripción del usuario"""
+    if not user_description or user_description.strip() == "":
+        return "❌ Por favor, describe lo que quieres generar"
+    
+    generated_prompt = generate_prompt_from_description(user_description, prompt_style)
+    
+    # Añadir al historial
+    generator.history.append({
+        "prompt": generated_prompt,
+        "role": "custom_description",
+        "timestamp": datetime.datetime.now().isoformat(),
+        "enhanced": True,
+        "source": "description_generation"
+    })
+    
+    return generated_prompt
 
-def generate_images_ui(prompt, api_key, num_images):
-    if not api_key:
-        return [], "Por favor, ingresa tu API Key."
-    imgs, errors = generate_images_batch(prompt, api_key, num_images)
-    msg = "Imágenes generadas con éxito." if imgs else "Error generando imágenes: " + "; ".join(errors)
-    return imgs, msg
+def describe_image_ui(image_path, description_style):
+    """Describe una imagen generada"""
+    if not image_path:
+        return "❌ Selecciona una imagen primero"
+    
+    description = describe_image_with_ai(image_path, description_style)
+    return description
 
 def clear_history_ui():
     generator.clear_history()
-    return json.dumps(generator.get_history(), indent=2)
+    return gr.update(value=json.dumps(generator.get_history(), indent=2))
 
-with gr.Blocks() as demo:
-    gr.Markdown("# HyperRealistic Roles AI - Generador + REVE CREATE + Maverick IA")
+# Interfaz Gradio optimizada
+with gr.Blocks(
+    theme=gr.themes.Soft(primary_hue="blue"),
+    title="AI Image Generator - Llama 4 Maverick",
+    css="""
+    .success { color: #22c55e; font-weight: bold; }
+    .warning { color: #f59e0b; font-weight: bold; }
+    .error { color: #ef4444; font-weight: bold; }
+    """
+) as demo:
+    
+    gr.Markdown(f"""
+    # 🚀 AI Image Generator + Llama 4 Maverick
+    
+    **Modelo único: {PRIMARY_MODEL}** - Generación, mejora y descripción de imágenes
+    """)
+    
+    with gr.Tab("🎨 Generar Imágenes"):
+        with gr.Row():
+            with gr.Column(scale=1):
+                gr.Markdown("### Configuración Principal")
+                
+                role_dropdown = gr.Dropdown(
+                    choices=[r["role"] for r in generator.config.PROFESSIONAL_ROLES],
+                    label="Rol Profesional",
+                    value=generator.config.PROFESSIONAL_ROLES[0]["role"]
+                )
+                
+                nsfw_checkbox = gr.Checkbox(label="Modo NSFW", value=False)
+                
+                use_ai_enhancement = gr.Checkbox(
+                    label="Usar Mejora con Llama 4 Maverick", 
+                    value=SAMBANOVA_ENABLED,
+                    interactive=SAMBANOVA_ENABLED
+                )
+                
+                enhancement_type = gr.Radio(
+                    choices=["creative", "technical", "artistic"],
+                    label="Tipo de Mejora",
+                    value="creative",
+                    interactive=SAMBANOVA_ENABLED
+                )
+                
+                with gr.Row():
+                    gen_prompt_btn = gr.Button("✨ Generar Prompt", variant="primary")
+                    clear_hist_btn = gr.Button("🗑️ Limpiar", variant="secondary")
+                
+                gr.Markdown("### Configuración REVE CREATE")
+                
+                api_key_input = gr.Textbox(
+                    label="API Key REVE CREATE", 
+                    type="password",
+                    placeholder="Ingresa tu API Key..."
+                )
+                
+                with gr.Row():
+                    ratio = gr.Dropdown(
+                        ["16:9", "9:16", "1:1", "4:3", "3:2"], 
+                        label="Aspect Ratio", 
+                        value="9:16"
+                    )
+                    version = gr.Dropdown(
+                        ["latest", "reve-create@20250915"], 
+                        label="Versión", 
+                        value="latest"
+                    )
+                
+                cantidad = gr.Slider(1, 4, step=1, value=1, label="Cantidad de Imágenes")
+                
+                gen_images_btn = gr.Button("🎨 Generar Imágenes", variant="primary", size="lg")
 
-    with gr.Row():
-        with gr.Column(scale=1):
-            role_dropdown = gr.Dropdown(
-                choices=[r["role"] for r in generator.config.PROFESSIONAL_ROLES],
-                label="Seleccione Rol Profesional",
-                value=generator.config.PROFESSIONAL_ROLES[0]["role"]
-            )
-            nsfw_checkbox = gr.Checkbox(label="Modo NSFW", value=False)
-            gen_prompt_btn = gr.Button("Generar Prompt")
-            user_input_text = gr.Textbox(label="Texto para sugerencia inteligente")
-            suggest_btn = gr.Button("Sugerir Prompt Inteligente")
-
-            api_key_input = gr.Textbox(label="API Key REVE CREATE", placeholder="Tu API Key REVE CREATE", type="password")
-            num_images = gr.Slider(1, 5, value=1, step=1, label="Cantidad de Imágenes")
-            gen_images_btn = gr.Button("Generar Imágenes")
-            clear_hist_btn = gr.Button("Limpiar Historial")
-
-        with gr.Column(scale=2):
-            prompt_out = gr.Textbox(label="Prompt Generado", lines=10)
-            suggestion_out = gr.Textbox(label="Sugerencias Maverick IA", lines=15)
-            images_gallery = gr.Gallery(label="Imágenes Generadas", columns=2)
-            
-            history_out = gr.JSON(label="Historial")
+            with gr.Column(scale=2):
+                prompt_out = gr.Textbox(
+                    label="Prompt Generado", 
+                    lines=6,
+                    show_copy_button=True
+                )
+                
+                gallery_out = gr.Gallery(
+                    label="Imágenes Generadas", 
+                    columns=2, 
+                    height="auto",
+                    show_download_button=True
+                )
+                
+                msg_out = gr.HTML()
+                
+                # Descripción de imágenes
+                with gr.Accordion("🔍 Describir Imágenes (Llama 4 Maverick)", open=False):
+                    with gr.Row():
+                        image_to_describe = gr.Dropdown(
+                            label="Seleccionar Imagen para Describir",
+                            choices=[],
+                            interactive=True
+                        )
+                        desc_style = gr.Radio(
+                            choices=["detailed", "technical", "prompt_style"],
+                            label="Estilo de Descripción",
+                            value="detailed"
+                        )
+                    describe_btn = gr.Button("📝 Describir Imagen", variant="secondary")
+                    image_description = gr.Textbox(
+                        label="Descripción Generada",
+                        lines=4,
+                        show_copy_button=True
+                    )
+    
+    with gr.Tab("📝 Generar desde Descripción"):
+        gr.Markdown("### Generar Prompt desde tu Descripción")
+        
+        user_description = gr.Textbox(
+            label="Describe la imagen que quieres generar",
+            placeholder="Ej: Una mujer ejecutiva en una oficina moderna con iluminación dramática al atardecer...",
+            lines=4
+        )
+        
+        prompt_style = gr.Radio(
+            choices=["realistic", "cinematic", "artistic", "professional"],
+            label="Estilo del Prompt",
+            value="realistic"
+        )
+        
+        generate_from_desc_btn = gr.Button("🚀 Generar Prompt desde Descripción", variant="primary")
+        
+        generated_prompt_out = gr.Textbox(
+            label="Prompt Generado",
+            lines=6,
+            show_copy_button=True
+        )
+    
+    with gr.Tab("📊 Historial"):
+        gr.Markdown("### Historial de Generaciones")
+        history_out = gr.JSON(label="Historial", show_label=False)
 
-    gen_prompt_btn.click(generate_prompt_ui, inputs=[role_dropdown, nsfw_checkbox], outputs=prompt_out)
-    suggest_btn.click(suggest_prompt_ui, inputs=[user_input_text], outputs=suggestion_out)
-    gen_images_btn.click(generate_images_ui, inputs=[prompt_out, api_key_input, num_images], outputs=[images_gallery, prompt_out])
-    clear_hist_btn.click(clear_history_ui, inputs=[], outputs=history_out)
+    # Event handlers
+    gen_prompt_btn.click(
+        fn=lambda role, nsfw, enhancement, enh_type: generator.generate_enhanced_prompt(role, nsfw, enh_type) 
+        if enhancement and SAMBANOVA_ENABLED else generator.generate_base_prompt(role, nsfw),
+        inputs=[role_dropdown, nsfw_checkbox, use_ai_enhancement, enhancement_type],
+        outputs=prompt_out
+    ).then(
+        fn=lambda: json.dumps(generator.get_history(), indent=2),
+        outputs=history_out
+    )
+    
+    gen_images_btn.click(
+        fn=generar_todo,
+        inputs=[role_dropdown, nsfw_checkbox, api_key_input, ratio, version, cantidad, use_ai_enhancement, enhancement_type],
+        outputs=[prompt_out, gallery_out, msg_out, image_to_describe]
+    ).then(
+        fn=lambda: json.dumps(generator.get_history(), indent=2),
+        outputs=history_out
+    ).then(
+        fn=lambda: gr.update(choices=os.listdir(CARPETA_SALIDA) if os.path.exists(CARPETA_SALIDA) else []),
+        outputs=image_to_describe
+    )
+    
+    generate_from_desc_btn.click(
+        fn=generate_from_description_ui,
+        inputs=[user_description, prompt_style],
+        outputs=generated_prompt_out
+    ).then(
+        fn=lambda: json.dumps(generator.get_history(), indent=2),
+        outputs=history_out
+    )
+    
+    describe_btn.click(
+        fn=describe_image_ui,
+        inputs=[image_to_describe, desc_style],
+        outputs=image_description
+    )
+    
+    clear_hist_btn.click(
+        fn=clear_history_ui,
+        outputs=history_out
+    )
 
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860, share=True)
-    
+    demo.launch(server_name="0.0.0.0", server_port=7860, share=False)