Update api.py

feat: full backend api)

api.py

@@ -2,27 +2,29 @@ from fastapi import FastAPI, BackgroundTasks, HTTPException
 from pydantic import BaseModel
 from typing import Optional
 
-# ====== IMPORTA AQUÍ TU LÓGICA REAL DE GENERACIÓN ======
-# Por ejemplo, si luego copias una función desde app.py de Rivera:
-# from generation import generate_image_from_prompt
+from generation import generate_image_from_prompt
 
 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 = "FLUX.1-schnell"
+    model: str = "black-forest-labs/FLUX.1-dev"
+
 
 @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.)
@@ -31,43 +33,86 @@ async def webhook_message(body: MessageRequest, background_tasks: BackgroundTask
     )
     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 (ejemplo simple; aquí replicas tu lógica de Sofia Rivera)
+    # 2) Construir prompt final (basado en tipos de Sofia Rivera)
     if body.custom_prompt:
         final_prompt = body.custom_prompt
+        negative_prompt = ""
     else:
-        # Aquí puedes mapear prompt_type → prompt predefinido igual que en app.py de Rivera
-        # p.ej. lifestyle, fitness, boudoir, fashion, beach, etc.
         prompt_map = {
-            "lifestyle": "Sofia in a modern lifestyle scenario, Instagram aesthetic, high quality, 4k",
-            "fitness": "Sofia in a fitness photoshoot at the gym, strong lighting, detailed, 4k",
-            "premium_boudoir": "Sofia in a premium boudoir photoshoot, cinematic lighting, soft focus, 4k",
-            "fashion": "Sofia in a fashion editorial photoshoot, Vogue style, 4k",
-            "beach": "Sofia on the beach at golden hour, cinematic, 4k",
+            "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]
 
-    # 3) Llamar a tu motor de generación REAL
-    # Aquí es donde debes enchufar la función que uses en app.py de Rivera.
-    # Ejemplo placeholder:
-    # image_url = generate_image_from_prompt(final_prompt, model=body.model)
+        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)
+    image_path, status = generate_image_from_prompt(
+        prompt=final_prompt,
+        negative_prompt=negative_prompt,
+        model_name=body.model,
+        seed=None,
+    )
 
-    image_url = "https://example.com/fake-image-url"  # <-- Sustituir por la URL real cuando integremos el motor
+    if image_path is None:
+        raise HTTPException(status_code=500, detail=status)
 
     return {
         "status": "completed",
         "prompt": final_prompt,
         "model": body.model,
-        "image_url": image_url,
+        "image_path": image_path,
+        "status_message": status,
     }
 
+
 if __name__ == "__main__":
     import uvicorn
+
     uvicorn.run(app, host="0.0.0.0", port=7860)