Update app.py

app.py

@@ -1,118 +1,103 @@
+import gradio as gr
 from fastapi import FastAPI, BackgroundTasks, HTTPException
 from pydantic import BaseModel
 from typing import Optional
+import uvicorn
 
-from generation import generate_image_from_prompt
+# --- INTENTA IMPORTAR LA GENERACIÓN, SI FALLA USA UN MOCK (Para que no se rompa el Space) ---
+try:
+    from generation import generate_image_from_prompt
+except ImportError:
+    print("⚠️ ADVERTENCIA: No se encontró 'generation.py'. Usando modo simulación.")
+    def generate_image_from_prompt(prompt, negative_prompt, model_name, seed):
+        return None, "Error: Falta el archivo generation.py en el Space"
 
+# --- CONFIGURACIÓN DE FASTAPI ---
 app = FastAPI(title="Sofia AI Backend")
 
-
 class MessageRequest(BaseModel):
     platform: str
     message: str
     user_id: str
     timestamp: Optional[str] = None
 
-
 class ImageGenerationRequest(BaseModel):
     prompt_type: Optional[str] = None
     custom_prompt: Optional[str] = None
     model: str = "black-forest-labs/FLUX.1-dev"
 
-
+# --- DEFINICIÓN DE PROMPTS DE SOFÍA (TU ESTRATEGIA) ---
+PROMPT_MAP = {
+    "lifestyle": {
+        "prompt": "foto selfie profesional con iPhone de Sofia Rivera, hermosa mujer latina cubanoamericana de 25 años, cabello largo oscuro y ondulado, sonrisa cálida, departamento de Miami con vista al mar, golden hour, estética influencer Instagram",
+        "negative": "borroso, de baja calidad, distorsionado, deformado, feo, mala anatomía",
+    },
+    "fitness": {
+        "prompt": "selfie de cuerpo completo en el espejo de Sofia Rivera, influencer latina fitness, cuerpo atlético tonificado, sujetador deportivo negro, mallas de cintura alta, gimnasio moderno con espejos, iluminación natural, estética fitness Instagram",
+        "negative": "borroso, de baja calidad, distorsionado, malas proporciones",
+    },
+    "premium_boudoir": {
+        "prompt": "selfie en el dormitorio de Sofia Rivera, influencer latina de 25 años, lencería de encaje blanco, luz suave de la mañana a través de cortinas transparentes, cama lujosa con sábanas de seda, expresión sensual y segura, estilo boudoir de buen gusto, fotografía profesional",
+        "negative": "explícito, borroso, de baja calidad, distorsionado",
+    },
+    "fashion": {
+        "prompt": "foto de estilo urbano de Sofia Rivera, influencer de moda latina, outfit moderno de Miami, gafas de sol de diseñador, pose natural y segura, fondo urbano, golden hour, estética moda Instagram",
+        "negative": "borroso, de baja calidad, mala iluminación",
+    },
+    "beach": {
+        "prompt": "foto de estilo de vida de Sofia Rivera en la playa, influencer latina, Miami Beach al atardecer, atuendo casual de playa, expresión natural feliz, vibraciones tropicales, contenido lifestyle Instagram",
+        "negative": "borroso, de baja calidad, distorsionado",
+    },
+}
+
+# --- ENDPOINTS DE API (PARA AUTOMATIZACIÓN FUTURA) ---
 @app.get("/health")
 async def health():
     return {"status": "ok", "service": "sofia-ai-backend"}
 
-
 @app.post("/webhook/message")
 async def webhook_message(body: MessageRequest, background_tasks: BackgroundTasks):
-    # Aquí luego conectarás lógica real (guardar en cola, n8n, etc.)
-    background_tasks.add_task(
-        lambda: print(f"[Message] {body.platform}: {body.message} (user: {body.user_id})")
-    )
+    background_tasks.add_task(lambda: print(f"[Message] {body.platform}: {body.message}"))
     return {"status": "queued"}
 
-
-@app.post("/api/generate")
-async def api_generate(body: ImageGenerationRequest):
-    # 1) Validar entrada
-    if not body.custom_prompt and not body.prompt_type:
-        raise HTTPException(status_code=400, detail="prompt_type or custom_prompt required")
-
-    # 2) Construir prompt final (basado en tipos de Sofia Rivera)
-    if body.custom_prompt:
-        final_prompt = body.custom_prompt
-        negative_prompt = ""
+# --- INTERFAZ VISUAL (PARA QUE GENERES TÚ MANUALMENTE) ---
+def ui_generate(style_selection, custom_text):
+    # Seleccionar prompt
+    if style_selection == "Personalizado":
+        final_prompt = custom_text
+        negative_prompt = "borroso, mala calidad"
     else:
-        prompt_map = {
-            "lifestyle": {
-                "prompt": (
-                    "foto selfie profesional con iPhone de Sofia Rivera, hermosa mujer latina "
-                    "cubanoamericana de 25 años, cabello largo oscuro y ondulado, sonrisa cálida, "
-                    "departamento de Miami con vista al mar, golden hour, estética influencer Instagram"
-                ),
-                "negative": "borroso, de baja calidad, distorsionado, deformado, feo, mala anatomía",
-            },
-            "fitness": {
-                "prompt": (
-                    "selfie de cuerpo completo en el espejo de Sofia Rivera, influencer latina fitness, "
-                    "cuerpo atlético tonificado, sujetador deportivo negro, mallas de cintura alta, "
-                    "gimnasio moderno con espejos, iluminación natural, estética fitness Instagram"
-                ),
-                "negative": "borroso, de baja calidad, distorsionado, malas proporciones",
-            },
-            "premium_boudoir": {
-                "prompt": (
-                    "selfie en el dormitorio de Sofia Rivera, influencer latina de 25 años, lencería de encaje "
-                    "blanco, luz suave de la mañana a través de cortinas transparentes, cama lujosa con sábanas "
-                    "de seda, expresión sensual y segura, estilo boudoir de buen gusto, fotografía profesional"
-                ),
-                "negative": "explícito, borroso, de baja calidad, distorsionado",
-            },
-            "fashion": {
-                "prompt": (
-                    "foto de estilo urbano de Sofia Rivera, influencer de moda latina, outfit moderno de Miami, "
-                    "gafas de sol de diseñador, pose natural y segura, fondo urbano, golden hour, estética moda Instagram"
-                ),
-                "negative": "borroso, de baja calidad, mala iluminación",
-            },
-            "beach": {
-                "prompt": (
-                    "foto de estilo de vida de Sofia Rivera en la playa, influencer latina, Miami Beach al atardecer, "
-                    "atuendo casual de playa, expresión natural feliz, vibraciones tropicales, contenido lifestyle Instagram"
-                ),
-                "negative": "borroso, de baja calidad, distorsionado",
-            },
-        }
-
-        if body.prompt_type not in prompt_map:
-            raise HTTPException(status_code=400, detail=f"Unknown prompt_type: {body.prompt_type}")
-
-        final_prompt = prompt_map[body.prompt_type]["prompt"]
-        negative_prompt = prompt_map[body.prompt_type]["negative"]
-
-    # 3) Llamar al motor de generación REAL (generation.py)
+        data = PROMPT_MAP.get(style_selection)
+        final_prompt = data["prompt"]
+        negative_prompt = data["negative"]
+    
+    # Llamar a la generación
     image_path, status = generate_image_from_prompt(
         prompt=final_prompt,
         negative_prompt=negative_prompt,
-        model_name=body.model,
-        seed=None,
+        model_name="black-forest-labs/FLUX.1-dev",
+        seed=None
     )
-
-    if image_path is None:
-        raise HTTPException(status_code=500, detail=status)
-
-    return {
-        "status": "completed",
-        "prompt": final_prompt,
-        "model": body.model,
-        "image_path": image_path,
-        "status_message": status,
-    }
-
+    
+    return image_path, final_prompt
+
+# Crear la Interfaz con Gradio
+with gr.Blocks(title="Sofia Rivera - Generador") as demo:
+    gr.Markdown("# 📸 Sofia AI Studio")
+    with gr.Row():
+        with gr.Column():
+            estilo = gr.Dropdown(choices=list(PROMPT_MAP.keys()) + ["Personalizado"], label="Estilo de Foto", value="lifestyle")
+            custom_prompt = gr.Textbox(label="Prompt Personalizado (si eliges arriba 'Personalizado')", placeholder="Escribe aquí...")
+            btn = gr.Button("Generar Foto", variant="primary")
+        with gr.Column():
+            resultado = gr.Image(label="Resultado")
+            debug_info = gr.Textbox(label="Prompt Usado", interactive=False)
+            
+    btn.click(fn=ui_generate, inputs=[estilo, custom_prompt], outputs=[resultado, debug_info])
+
+# MONTAR GRADIO SOBRE FASTAPI
+# Esto hace que la ruta principal "/" muestre la interfaz visual
+app = gr.mount_gradio_app(app, demo, path="/")
 
 if __name__ == "__main__":
-    import uvicorn
-
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    uvicorn.run(app, host="0.0.0.0", port=7860)