Update app.py

app.py

@@ -1,141 +1,80 @@
-import gradio as gr
-import os
-import random
-from datetime import datetime
-from huggingface_hub import InferenceClient
-from PIL import Image
+import gradio as gr  
+import os  
+import random  
+from datetime import datetime  
+from huggingface_hub import InferenceClient  
+  
+# Configuración  
+OUTPUT_DIR = "generated_images"  
+os.makedirs(OUTPUT_DIR, exist_ok=True)  
+client = InferenceClient()  
+  
+# Identidad de Sofía Rivera  
+SOFIA_CORE = (  
+    "sofia rivera, a beautiful 25-year-old latina cuban-american woman, "  
+    "long dark wavy hair, mesmerizing hazel eyes, toned athletic fitness body, "  
+    "natural radiant skin, miami influencer aesthetic"  
+)  
+  
+CATEGORIES = {  
+    "Fitness - Gym": {  
+        "prompt": f"full body mirror selfie of {SOFIA_CORE}, wearing black sports bra and leggings, modern miami gym background, 8k, photorealistic",  
+        "desc": "### 🏋️ Pilar Fitness\nIdeal para Lunes, Miércoles y Viernes."  
+    },  
+    "Lifestyle - Beach": {  
+        "prompt": f"candid photo of {SOFIA_CORE}, tropical outfit, miami beach sunset, cinematic lighting, vibrant colors",  
+        "desc": "### 🌴 Pilar Lifestyle\nIdeal para Martes y Jueves."  
+    },  
+    "Lifestyle - Luxury": {  
+        "prompt": f"portrait of {SOFIA_CORE}, luxury miami apartment, morning light, natural makeup, authentic vibe",  
+        "desc": "### ☕ Pilar Personal\nIdeal para Sábados."  
+    },  
+    "Premium - Elegant": {  
+        "prompt": f"tasteful photography of {SOFIA_CORE}, elegant black lace, soft moody lighting, professional aesthetic",  
+        "desc": "### 💎 Pilar Premium\nIdeal para Domingos."  
+    }  
+}  
+  
+def generate(category_name, tweak, model_id, steps, cfg, seed):  
+    try:  
+        cat = CATEGORIES[category_name]  
+        prompt = f"{cat['prompt']}, {tweak}" if tweak else cat['prompt']  
+        seed = random.randint(0, 2**32 - 1) if seed == -1 else int(seed)  
+          
+        image = client.text_to_image(  
+            prompt=prompt,  
+            model=model_id,  
+            num_inference_steps=int(steps),  
+            guidance_scale=float(cfg),  
+            seed=seed  
+        )  
+        return image, f"✅ Generada: {category_name} | Seed: {seed}"  
+    except Exception as e:  
+        return None, f"❌ Error: {str(e)}"  
+  
+def create_interface():  
+    with gr.Blocks(theme=gr.themes.Soft()) as demo:  
+        gr.Markdown("# 🌴 Sofia Rivera AI Workspace")  
+        with gr.Tab("🎨 Generador"):  
+            with gr.Row():  
+                with gr.Column(scale=1):  
+                    cat_drop = gr.Dropdown(choices=list(CATEGORIES.keys()), label="Pilar", value="Fitness - Gym")  
+                    cat_info = gr.Markdown(CATEGORIES["Fitness - Gym"]["desc"])  
+                    tweak = gr.Textbox(label="Detalles extra", placeholder="ej: wearing sunglasses...")  
+                    with gr.Accordion("Avanzado", open=False):  
+                        model = gr.Dropdown(choices=["black-forest-labs/FLUX.1-dev", "black-forest-labs/FLUX.1-schnell"], label="Modelo", value="black-forest-labs/FLUX.1-dev")  
+                        steps = gr.Slider(1, 50, 28, label="Pasos")  
+                        cfg = gr.Slider(1, 20, 7, label="Realismo")  
+                        seed = gr.Number(-1, label="Seed")  
+                    btn = gr.Button("🚀 Generar", variant="primary")  
+                with gr.Column(scale=2):  
+                    out_img = gr.Image(label="Sofía Rivera")  
+                    out_log = gr.Textbox(label="Estado", interactive=False)  
+          
+        cat_drop.change(lambda x: CATEGORIES[x]["desc"], inputs=[cat_drop], outputs=[cat_info])  
+        btn.click(generate, [cat_drop, tweak, model, steps, cfg, seed], [out_img, out_log])  
+    return demo  
+  
+if __name__ == "__main__":  
+    create_interface().launch()  
 
-# Configuración de entorno y directorios
-OUTPUT_DIR = "generated_images"
-os.makedirs(OUTPUT_DIR, exist_ok=True)
-
-# Inicialización del cliente de inferencia
-# En Hugging Face Spaces, el token se maneja automáticamente si el Space tiene permisos
-client = InferenceClient()
-
-# --- PERFIL DE SOFIA RIVERA ---
-SOFIA_CORE = (
-    "sofia rivera, a beautiful 25-year-old latina cuban-american woman, "
-    "long dark wavy hair, mesmerizing hazel eyes, toned athletic fitness body, "
-    "natural radiant skin, miami influencer aesthetic"
-)
-
-# --- CATEGORÍAS DE CONTENIDO ---
-CATEGORIES = {
-    "Fitness - Gym Mirror": {
-        "prompt": f"full body mirror selfie of {SOFIA_CORE}, wearing professional black sports bra and matching high-waist leggings, standing in a modern high-end miami gym, mirrors reflecting gym equipment, natural window lighting, confident pose, authentic fitness instagram post, shot on iPhone 15 Pro, high detail, 8k",
-        "negative": "blurry, low quality, distorted, bad proportions, cartoon, anime, 3d render",
-        "description": "### 🏋️ Pilar Fitness (40%)\nIdeal para Lunes, Miércoles y Viernes. Enfoque en motivación y resultados."
-    },
-    "Lifestyle - Miami Beach": {
-        "prompt": f"candid lifestyle photo of {SOFIA_CORE}, wearing a stylish tropical summer outfit, standing on miami beach with turquoise ocean background, sunset golden hour lighting, palm trees, warm breeze, happy genuine smile, high fashion photography, vibrant colors, shot on professional camera, cinematic lighting",
-        "negative": "blurry, low quality, distorted, bad anatomy, overexposed, crowded beach",
-        "description": "### 🌴 Pilar Lifestyle Miami (30%)\nIdeal para Martes y Jueves. Fotos en exteriores, playa y vibras tropicales."
-    },
-    "Lifestyle - Luxury Apartment": {
-        "prompt": f"close up portrait of {SOFIA_CORE} in a luxury miami apartment, floor-to-ceiling windows with city skyline view, wearing casual white silk loungewear, soft morning light, natural makeup, holding a coffee cup, relaxed and authentic vibe, 8k resolution, photorealistic",
-        "negative": "blurry, low quality, distorted, messy room, dark lighting",
-        "description": "### ☕ Pilar Personal/BTS (20%)\nIdeal para Sábados. Contenido 'Detrás de cámaras' y rutinas diarias."
-    },
-    "Premium - Elegant Boudoir": {
-        "prompt": f"tasteful boudoir photography of {SOFIA_CORE}, wearing elegant black lace lingerie, soft moody lighting in a luxury bedroom, silk sheets, artistic shadows, confident and sultry expression, professional photography aesthetic, high class and elegant",
-        "negative": "explicit, pornographic, blurry, low quality, distorted, cheap aesthetic",
-        "description": "### 💎 Pilar Premium (10%)\nIdeal para Domingos. Teasers exclusivos con elegancia y clase."
-    }
-}
-
-def generate_sofia_image(category_name, custom_tweak, model_id, steps, cfg_scale, seed):
-    try:
-        category = CATEGORIES[category_name]
-        final_prompt = category["prompt"]
-        if custom_tweak:
-            final_prompt += f", {custom_tweak}"
-            
-        if seed == -1:
-            seed = random.randint(0, 2**32 - 1)
-
-        image = client.text_to_image(
-            prompt=final_prompt,
-            negative_prompt=category["negative"],
-            model=model_id,
-            num_inference_steps=int(steps),
-            guidance_scale=float(cfg_scale),
-            seed=int(seed)
-        )
-
-        timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
-        filename = f"sofia_{timestamp}.png"
-        filepath = os.path.join(OUTPUT_DIR, filename)
-        image.save(filepath)
-
-        return image, f"✅ Imagen generada: {filename}\nSeed: {seed}"
-    except Exception as e:
-        return None, f"❌ Error: {str(e)}"
-
-# --- INTERFAZ DE GRADIO ---
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown(f"""
-    # 🌴 Sofia Rivera - AI Influencer Workspace
-    Bienvenido al panel de control de **Sofía Rivera**. Genera contenido consistente basado en la estrategia oficial.
-    """)
-    
-    with gr.Tab("🎨 Generador"):
-        with gr.Row():
-            with gr.Column(scale=1):
-                category_dropdown = gr.Dropdown(
-                    choices=list(CATEGORIES.keys()), 
-                    label="Pilar de Contenido", 
-                    value="Fitness - Gym Mirror"
-                )
-                category_desc = gr.Markdown(CATEGORIES["Fitness - Gym Mirror"]["description"])
-                
-                custom_tweak = gr.Textbox(
-                    label="Personalizar Escena", 
-                    placeholder="ej: holding a water bottle, wearing sunglasses, laughing..."
-                )
-                
-                with gr.Accordion("Configuración Técnica", open=False):
-                    model_id = gr.Dropdown(
-                        choices=["black-forest-labs/FLUX.1-dev", "black-forest-labs/FLUX.1-schnell"],
-                        label="Modelo",
-                        value="black-forest-labs/FLUX.1-dev"
-                    )
-                    steps = gr.Slider(minimum=1, maximum=50, step=1, value=28, label="Pasos")
-                    cfg = gr.Slider(minimum=1, maximum=20, step=0.5, value=7.0, label="CFG Scale")
-                    seed = gr.Number(label="Seed (-1 = Aleatorio)", value=-1)
-
-                generate_btn = gr.Button("🚀 Generar Contenido", variant="primary")
-                
-            with gr.Column(scale=2):
-                output_image = gr.Image(label="Vista Previa de Sofía")
-                status_text = gr.Textbox(label="Log de Sistema", interactive=False)
-
-    with gr.Tab("📊 Estrategia & Perfil"):
-        gr.Markdown(f"""
-        ### 👤 Identidad de Sofía Rivera
-        - **Nombre:** Sofía Rivera
-        - **Edad:** 25 años
-        - **Etnia:** Latina (Cubana-Americana)
-        - **Ubicación:** Miami, Florida
-        - **Rasgos:** Cabello oscuro ondulado, ojos avellana, cuerpo atlético.
-        
-        ### 📅 Calendario de Publicación
-        - **Lunes/Miércoles/Viernes:** Fitness & Gym
-        - **Martes/Jueves:** Lifestyle & Miami Vibes
-        - **Sábados:** Personal & Detrás de Cámaras
-        - **Domingos:** Premium Teasers
-        """)
-
-    def update_desc(cat):
-        return CATEGORIES[cat]["description"]
-    
-    category_dropdown.change(update_desc, inputs=[category_dropdown], outputs=[category_desc])
-    
-    generate_btn.click(
-        fn=generate_sofia_image,
-        inputs=[category_dropdown, custom_tweak, model_id, steps, cfg, seed],
-        outputs=[output_image, status_text]
-    )
-
-if __name__ == "__main__":
-    demo.launch()