Update app.py

app.py

@@ -1,6 +1,6 @@
 """
 App de Mejora de Prompts para Generación de Imágenes
-Optimizada para Hugging Face Spaces - Compatible con Gradio 6.0
+Compatible con Gradio 6.0+
 """
 
 import gradio as gr
@@ -11,16 +11,16 @@ from model_configs import MODEL_CONFIGS, ARTIST_STYLES, ENHANCEMENT_RULES
 enhancer = PromptEnhancer()
 analyzer = PromptAnalyzer()
 
+
 # ===== FUNCIONES PRINCIPALES =====
 
 def enhance_prompt(prompt: str, model: str, style: str, 
                    add_quality: bool, add_negative: bool) -> tuple:
     """Función principal de mejora de prompts"""
     
-    if not prompt.strip():
+    if not prompt or not prompt.strip():
         return ("⚠️ Por favor, ingresa un prompt", "", "", "", "")
     
-    # Mapear nombres de modelo
     model_map = {
         "Stable Diffusion 1.5": "stable-diffusion-1.5",
         "Stable Diffusion XL": "sdxl",
@@ -30,9 +30,8 @@ def enhance_prompt(prompt: str, model: str, style: str,
     }
     
     model_key = model_map.get(model, "stable-diffusion-1.5")
-    style_key = style.lower() if style != "Ninguno" else None
+    style_key = style.lower() if style and style != "Ninguno" else None
     
-    # Mejorar el prompt
     result = enhancer.enhance_for_model(
         prompt=prompt,
         model=model_key,
@@ -41,12 +40,12 @@ def enhance_prompt(prompt: str, model: str, style: str,
         add_negative=add_negative
     )
     
-    # Formatear análisis
     analysis = result["analysis"]
     score_emoji = "🟢" if analysis["score"] >= 70 else "🟡" if analysis["score"] >= 40 else "🔴"
     
-    analysis_text = f"""
-### 📊 Análisis del Prompt
+    suggestions_text = '\n'.join(analysis['suggestions']) if analysis['suggestions'] else '✅ ¡El prompt está bien estructurado!'
+    
+    analysis_text = f"""### 📊 Análisis del Prompt
 
 **Puntuación:** {score_emoji} {analysis['score']}/100
 
@@ -56,10 +55,9 @@ def enhance_prompt(prompt: str, model: str, style: str,
 - Tokens estimados: {result['tokens_estimate']}/{result['max_tokens']}
 
 **Sugerencias:**
-{chr(10).join(analysis['suggestions']) if analysis['suggestions'] else '✅ ¡El prompt está bien estructurado!'}
+{suggestions_text}
 """
     
-    # Generar variaciones
     variations = enhancer.suggest_variations(result["enhanced"], 3)
     variations_text = "\n\n".join([f"**Variación {i+1}:**\n`{v}`" for i, v in enumerate(variations)])
     
@@ -75,7 +73,7 @@ def enhance_prompt(prompt: str, model: str, style: str,
 def fix_prompt(prompt: str) -> tuple:
     """Corrige errores comunes en el prompt"""
     
-    if not prompt.strip():
+    if not prompt or not prompt.strip():
         return ("⚠️ Por favor, ingresa un prompt", "")
     
     fixed, fixes = enhancer.fix_common_errors(prompt)
@@ -92,12 +90,10 @@ def build_prompt(subject: str, style: str, lighting: str, mood: str,
                  camera: str, quality_opts: list, extra_details: str) -> str:
     """Construye un prompt desde componentes"""
     
-    if not subject.strip():
+    if not subject or not subject.strip():
         return "⚠️ Por favor, ingresa al menos un sujeto"
     
     builder = PromptBuilder()
-    
-    # Configurar componentes
     builder.set_subject(subject)
     
     if style and style != "Ninguno":
@@ -126,7 +122,7 @@ def build_prompt(subject: str, style: str, lighting: str, mood: str,
                 quality_terms.extend(quality_map[opt])
         builder.add_quality(quality_terms)
     
-    if extra_details.strip():
+    if extra_details and extra_details.strip():
         builder.add_details([extra_details.strip()])
     
     return builder.build()
@@ -135,13 +131,12 @@ def build_prompt(subject: str, style: str, lighting: str, mood: str,
 def analyze_only(prompt: str) -> str:
     """Solo analiza el prompt sin mejorarlo"""
     
-    if not prompt.strip():
+    if not prompt or not prompt.strip():
         return "⚠️ Por favor, ingresa un prompt para analizar"
     
     analysis = analyzer.analyze(prompt)
-    
-    # Crear reporte detallado
     score = analysis["score"]
+    
     if score >= 80:
         grade = "🏆 Excelente"
     elif score >= 60:
@@ -151,7 +146,6 @@ def analyze_only(prompt: str) -> str:
     else:
         grade = "❌ Necesita trabajo"
     
-    issues_text = ""
     if analysis["issues"]:
         issue_names = {
             "muy_corto": "❌ Prompt muy corto",
@@ -165,8 +159,9 @@ def analyze_only(prompt: str) -> str:
     else:
         issues_text = "✅ No se detectaron problemas"
     
-    report = f"""
-## 📊 Análisis Detallado
+    suggestions_text = '\n'.join(analysis['suggestions']) if analysis['suggestions'] else '¡El prompt está bien estructurado!'
+    
+    report = f"""## 📊 Análisis Detallado
 
 ### Calificación: {grade} ({score}/100)
 
@@ -186,7 +181,7 @@ def analyze_only(prompt: str) -> str:
 ---
 
 ### 💡 Sugerencias de Mejora
-{chr(10).join(analysis['suggestions']) if analysis['suggestions'] else '¡El prompt está bien estructurado!'}
+{suggestions_text}
 
 ---
 
@@ -204,214 +199,289 @@ def analyze_only(prompt: str) -> str:
 def get_style_examples(style: str) -> str:
     """Muestra ejemplos para un estilo específico"""
     
-    if style == "Ninguno" or not style:
+    if not style or style == "Ninguno":
         return "Selecciona un estilo para ver ejemplos"
     
     examples = {
-        "Fotorealista": """
-**Ejemplos de prompts fotorealistas:**
+        "Fotorealista": """**Ejemplos de prompts fotorealistas:**
 
 1. `professional photograph of a woman, natural lighting, 85mm lens, shallow depth of field, photorealistic, 8k uhd`
 
 2. `hyperrealistic photo of a mountain landscape, golden hour, dramatic clouds, raw photo, high dynamic range`
 
-3. `product photography of a luxury watch, studio lighting, macro shot, commercial photography, pristine details`
-""",
-        "Anime": """
-**Ejemplos de prompts anime:**
+3. `product photography of a luxury watch, studio lighting, macro shot, commercial photography, pristine details`""",
+        
+        "Anime": """**Ejemplos de prompts anime:**
 
 1. `anime girl with blue hair, cherry blossom background, studio ghibli style, vibrant colors, detailed eyes, key visual`
 
 2. `epic anime battle scene, dynamic action pose, energy effects, dramatic lighting, anime key visual, trending on pixiv`
 
-3. `cozy anime room interior, warm lighting, detailed background, slice of life aesthetic, makoto shinkai style`
-""",
-        "Arte Digital": """
-**Ejemplos de arte digital:**
+3. `cozy anime room interior, warm lighting, detailed background, slice of life aesthetic, makoto shinkai style`""",
+        
+        "Arte Digital": """**Ejemplos de arte digital:**
 
 1. `digital painting of a fantasy castle, dramatic sky, concept art, artstation trending, highly detailed, epic composition`
 
 2. `character design, full body portrait, detailed armor, digital art, fantasy warrior, professional illustration`
 
-3. `environment concept art, alien planet, sci-fi landscape, matte painting, cinematic, artstation winner`
-""",
-        "Pintura": """
-**Ejemplos de pintura:**
+3. `environment concept art, alien planet, sci-fi landscape, matte painting, cinematic, artstation winner`""",
+        
+        "Pintura": """**Ejemplos de pintura:**
 
 1. `oil painting of a serene lake, impressionist style, visible brush strokes, golden hour lighting, museum quality`
 
 2. `renaissance portrait, classical technique, dramatic chiaroscuro, oil on canvas, by old masters`
 
-3. `watercolor landscape, soft edges, wet on wet technique, delicate colors, paper texture visible`
-""",
-        "3D": """
-**Ejemplos de 3D/CGI:**
+3. `watercolor landscape, soft edges, wet on wet technique, delicate colors, paper texture visible`""",
+        
+        "3D": """**Ejemplos de 3D/CGI:**
 
 1. `3D render of a cute robot character, octane render, subsurface scattering, studio lighting, pixar style`
 
 2. `architectural visualization, modern house, ray tracing, unreal engine 5, photorealistic materials`
 
-3. `3D character model, fantasy creature, zbrush detail, substance painter textures, turntable view`
-""",
-        "Cinematico": """
-**Ejemplos cinematográficos:**
+3. `3D character model, fantasy creature, zbrush detail, substance painter textures, turntable view`""",
+        
+        "Cinematico": """**Ejemplos cinematográficos:**
 
 1. `cinematic still from a sci-fi movie, anamorphic lens, dramatic lighting, movie poster composition, 4k`
 
 2. `film noir scene, high contrast, dramatic shadows, mysterious atmosphere, classic hollywood style`
 
-3. `epic movie landscape, sweeping vista, volumetric fog, golden hour, widescreen composition`
-"""
+3. `epic movie landscape, sweeping vista, volumetric fog, golden hour, widescreen composition`"""
     }
     
     return examples.get(style, "No hay ejemplos disponibles para este estilo")
 
 
-# ===== INTERFAZ GRADIO =====
-
-# CSS personalizado
-custom_css = """
-.gradio-container {
-    max-width: 1200px !important;
-}
-.main-title {
-    text-align: center;
-    color: #2563eb;
-    margin-bottom: 0.5em;
-}
-.subtitle {
-    text-align: center;
-    color: #6b7280;
-    font-size: 1.1em;
-    margin-bottom: 1.5em;
-}
-.prompt-output {
-    font-family: monospace;
-    background-color: #f3f4f6;
-    padding: 10px;
-    border-radius: 8px;
-}
-"""
+# ===== INTERFAZ GRADIO 6.0+ =====
 
-# Crear la interfaz
-with gr.Blocks(title="Prompt Enhancer Pro") as app:
-    
-    gr.HTML("""
-        <h1 class='main-title'>🎨 Prompt Enhancer Pro</h1>
-        <p class='subtitle'>Mejora y optimiza tus prompts para generación de imágenes con IA</p>
-    """)
+def create_app():
+    """Crea la aplicación Gradio"""
     
-    with gr.Tabs():
+    with gr.Blocks(title="Prompt Enhancer Pro") as app:
+        
+        gr.HTML("""
+            <div style="text-align: center; margin-bottom: 1.5em;">
+                <h1 style="color: #2563eb; margin-bottom: 0.3em;">🎨 Prompt Enhancer Pro</h1>
+                <p style="color: #6b7280; font-size: 1.1em;">Mejora y optimiza tus prompts para generación de imágenes con IA</p>
+            </div>
+        """)
         
-        # ===== TAB 1: MEJORAR PROMPT =====
-        with gr.TabItem("✨ Mejorar Prompt"):
-            gr.Markdown("### Transforma tu prompt simple en uno profesional")
+        with gr.Tabs():
             
-            with gr.Row():
-                with gr.Column(scale=1):
-                    input_prompt = gr.Textbox(
-                        label="Tu prompt original",
-                        placeholder="Escribe tu prompt aquí... (puede ser en español o inglés)",
-                        lines=3
-                    )
+            # ===== TAB 1: MEJORAR PROMPT =====
+            with gr.TabItem("✨ Mejorar Prompt"):
+                gr.Markdown("### Transforma tu prompt simple en uno profesional")
+                
+                with gr.Row():
+                    with gr.Column(scale=1):
+                        input_prompt = gr.Textbox(
+                            label="Tu prompt original",
+                            placeholder="Escribe tu prompt aquí... (puede ser en español o inglés)",
+                            lines=3
+                        )
+                        
+                        with gr.Row():
+                            model_select = gr.Dropdown(
+                                choices=["Stable Diffusion 1.5", "Stable Diffusion XL", 
+                                        "FLUX.1", "Midjourney", "DALL-E 3"],
+                                value="Stable Diffusion XL",
+                                label="Modelo destino"
+                            )
+                            
+                            style_select = gr.Dropdown(
+                                choices=["Ninguno", "Fotorealista", "Anime", "Arte Digital", 
+                                        "Pintura", "3D", "Cinematico", "Minimalista", "Fantasia"],
+                                value="Ninguno",
+                                label="Estilo artístico"
+                            )
+                        
+                        with gr.Row():
+                            add_quality = gr.Checkbox(label="Añadir tokens de calidad", value=True)
+                            add_negative = gr.Checkbox(label="Generar negative prompt", value=True)
+                        
+                        enhance_btn = gr.Button("🚀 Mejorar Prompt", variant="primary")
                     
-                    with gr.Row():
-                        model_select = gr.Dropdown(
-                            choices=["Stable Diffusion 1.5", "Stable Diffusion XL", 
-                                    "FLUX.1", "Midjourney", "DALL-E 3"],
-                            value="Stable Diffusion XL",
-                            label="Modelo destino"
+                    with gr.Column(scale=1):
+                        output_prompt = gr.Textbox(
+                            label="✅ Prompt mejorado",
+                            lines=4,
+                            interactive=False
                         )
                         
-                        style_select = gr.Dropdown(
-                            choices=["Ninguno", "Fotorealista", "Anime", "Arte Digital", 
-                                    "Pintura", "3D", "Cinematico", "Minimalista", "Fantasia"],
-                            value="Ninguno",
-                            label="Estilo artístico"
+                        output_negative = gr.Textbox(
+                            label="❌ Negative prompt",
+                            lines=2,
+                            interactive=False
                         )
-                    
-                    with gr.Row():
-                        add_quality = gr.Checkbox(label="Añadir tokens de calidad", value=True)
-                        add_negative = gr.Checkbox(label="Generar negative prompt", value=True)
-                    
-                    enhance_btn = gr.Button("🚀 Mejorar Prompt", variant="primary")
+                        
+                        model_info = gr.Markdown("")
                 
-                with gr.Column(scale=1):
-                    output_prompt = gr.Textbox(
-                        label="✅ Prompt mejorado",
-                        lines=4,
-                        interactive=False
-                    )
-                    
-                    output_negative = gr.Textbox(
-                        label="❌ Negative prompt",
-                        lines=2,
-                        interactive=False
-                    )
-                    
-                    model_info = gr.Markdown("")
-            
-            with gr.Row():
-                with gr.Column():
-                    analysis_output = gr.Markdown(label="Análisis")
-                with gr.Column():
-                    variations_output = gr.Markdown(label="Variaciones sugeridas")
-            
-            enhance_btn.click(
-                fn=enhance_prompt,
-                inputs=[input_prompt, model_select, style_select, add_quality, add_negative],
-                outputs=[output_prompt, output_negative, analysis_output, variations_output, model_info]
-            )
-        
-        # ===== TAB 2: CORRECTOR =====
-        with gr.TabItem("🔧 Corrector de Errores"):
-            gr.Markdown("### Detecta y corrige errores comunes en tus prompts")
-            
-            with gr.Row():
-                with gr.Column():
-                    fix_input = gr.Textbox(
-                        label="Prompt con posibles errores",
-                        placeholder="Pega aquí un prompt que quieras revisar...",
-                        lines=4
-                    )
-                    fix_btn = gr.Button("🔍 Analizar y Corregir", variant="primary")
+                with gr.Row():
+                    with gr.Column():
+                        analysis_output = gr.Markdown(label="Análisis")
+                    with gr.Column():
+                        variations_output = gr.Markdown(label="Variaciones sugeridas")
                 
-                with gr.Column():
-                    fix_output = gr.Textbox(
-                        label="Prompt corregido",
-                        lines=4,
-                        interactive=False
-                    )
-                    fixes_log = gr.Markdown("")
+                enhance_btn.click(
+                    fn=enhance_prompt,
+                    inputs=[input_prompt, model_select, style_select, add_quality, add_negative],
+                    outputs=[output_prompt, output_negative, analysis_output, variations_output, model_info]
+                )
             
-            fix_btn.click(
-                fn=fix_prompt,
-                inputs=[fix_input],
-                outputs=[fix_output, fixes_log]
-            )
-        
-        # ===== TAB 3: CONSTRUCTOR =====
-        with gr.TabItem("🏗️ Constructor de Prompts"):
-            gr.Markdown("### Construye un prompt paso a paso")
+            # ===== TAB 2: CORRECTOR =====
+            with gr.TabItem("🔧 Corrector de Errores"):
+                gr.Markdown("### Detecta y corrige errores comunes en tus prompts")
+                
+                with gr.Row():
+                    with gr.Column():
+                        fix_input = gr.Textbox(
+                            label="Prompt con posibles errores",
+                            placeholder="Pega aquí un prompt que quieras revisar...",
+                            lines=4
+                        )
+                        fix_btn = gr.Button("🔍 Analizar y Corregir", variant="primary")
+                    
+                    with gr.Column():
+                        fix_output = gr.Textbox(
+                            label="Prompt corregido",
+                            lines=4,
+                            interactive=False
+                        )
+                        fixes_log = gr.Markdown("")
+                
+                fix_btn.click(
+                    fn=fix_prompt,
+                    inputs=[fix_input],
+                    outputs=[fix_output, fixes_log]
+                )
             
-            with gr.Row():
-                with gr.Column():
-                    build_subject = gr.Textbox(
-                        label="Sujeto principal *",
-                        placeholder="Ej: a young woman with red hair, a futuristic city..."
-                    )
+            # ===== TAB 3: CONSTRUCTOR =====
+            with gr.TabItem("🏗️ Constructor de Prompts"):
+                gr.Markdown("### Construye un prompt paso a paso")
+                
+                with gr.Row():
+                    with gr.Column():
+                        build_subject = gr.Textbox(
+                            label="Sujeto principal *",
+                            placeholder="Ej: a young woman with red hair, a futuristic city..."
+                        )
+                        
+                        with gr.Row():
+                            build_style = gr.Dropdown(
+                                choices=["Ninguno", "Clasico", "Impresionista", "Surrealista", 
+                                        "Pop Art", "Art Nouveau", "Concept Art", "Anime", 
+                                        "Fantasia", "Sci-fi", "Horror"],
+                                value="Ninguno",
+                                label="Estilo artístico"
+                            )
+                            
+                            build_lighting = gr.Dropdown(
+                                choices=["Ninguno", "Dramatica", "Suave", "Natural", 
+                                        "Estudio", "Neon", "Cinematica"],
+                                value="Ninguno",
+                                label="Iluminación"
+                            )
+                        
+                        with gr.Row():
+                            build_mood = gr.Dropdown(
+                                choices=["Ninguno", "Oscuro", "Alegre", "Melancolico", 
+                                        "Epico", "Sereno", "Intenso"],
+                                value="Ninguno",
+                                label="Ambiente/Mood"
+                            )
+                            
+                            build_camera = gr.Dropdown(
+                                choices=["Ninguno", "Cercano", "Lejano", "Retrato", 
+                                        "Aereo", "Bajo", "Alto"],
+                                value="Ninguno",
+                                label="Ángulo/Encuadre"
+                            )
+                        
+                        build_quality = gr.CheckboxGroup(
+                            choices=["Alta definición (8K)", "Muy detallado", "Masterpiece",
+                                    "Fotografía profesional", "Arte de alta calidad"],
+                            label="Calidad"
+                        )
+                        
+                        build_extra = gr.Textbox(
+                            label="Detalles adicionales",
+                            placeholder="Añade cualquier detalle extra..."
+                        )
+                        
+                        build_btn = gr.Button("🔨 Construir Prompt", variant="primary")
                     
-                    with gr.Row():
-                        build_style = gr.Dropdown(
-                            choices=["Ninguno", "Clasico", "Impresionista", "Surrealista", 
-                                    "Pop Art", "Art Nouveau", "Concept Art", "Anime", 
-                                    "Fantasia", "Sci-fi", "Horror"],
-                            value="Ninguno",
-                            label="Estilo artístico"
+                    with gr.Column():
+                        build_output = gr.Textbox(
+                            label="Prompt construido",
+                            lines=6,
+                            interactive=False
                         )
                         
-                        build_lighting = gr.Dropdown(
-                            choices=["Ninguno", "Dramatica", "Suave", "Natural", 
-                                    "Estudio", "Neon", "Cinematica"],
-                            value="Ninguno",
-                            
+                        gr.Markdown("""### 💡 Tips para cada componente:
+
+- **Sujeto**: Sé específico. "young woman with curly red hair" > "woman"
+- **Estilo**: Define la estética visual del resultado
+- **Iluminación**: Afecta dramáticamente el mood de la imagen
+- **Ambiente**: Establece el tono emocional
+- **Encuadre**: Controla la composición de la imagen""")
+                
+                build_btn.click(
+                    fn=build_prompt,
+                    inputs=[build_subject, build_style, build_lighting, build_mood, 
+                           build_camera, build_quality, build_extra],
+                    outputs=[build_output]
+                )
+            
+            # ===== TAB 4: ANALIZADOR =====
+            with gr.TabItem("📊 Analizador"):
+                gr.Markdown("### Analiza la calidad y estructura de tu prompt")
+                
+                with gr.Row():
+                    with gr.Column():
+                        analyze_input = gr.Textbox(
+                            label="Prompt a analizar",
+                            placeholder="Pega cualquier prompt para obtener un análisis detallado...",
+                            lines=4
+                        )
+                        analyze_btn = gr.Button("📊 Analizar", variant="primary")
+                    
+                    with gr.Column():
+                        analyze_output = gr.Markdown("")
+                
+                analyze_btn.click(
+                    fn=analyze_only,
+                    inputs=[analyze_input],
+                    outputs=[analyze_output]
+                )
+            
+            # ===== TAB 5: EJEMPLOS Y GUÍA =====
+            with gr.TabItem("📚 Guía y Ejemplos"):
+                gr.Markdown("### Aprende a crear mejores prompts")
+                
+                with gr.Row():
+                    with gr.Column():
+                        example_style = gr.Dropdown(
+                            choices=["Fotorealista", "Anime", "Arte Digital", 
+                                    "Pintura", "3D", "Cinematico"],
+                            value="Fotorealista",
+                            label="Selecciona un estilo"
+                        )
+                        show_examples_btn = gr.Button("Ver ejemplos", variant="secondary")
+                    
+                    with gr.Column():
+                        examples_output = gr.Markdown("")
+                
+                show_examples_btn.click(
+                    fn=get_style_examples,
+                    inputs=[example_style],
+                    outputs=[examples_output]
+                )
+                
+                gr.Markdown("""---
+
+## 📖 Guía Rápida de Prompts
+
+### Estructura recomendada: