Update api.py

feat: reuse sofia-rivera api structure in backend.

api.py

@@ -1,25 +1,46 @@
-from fastapi import FastAPI
+from fastapi import FastAPI, BackgroundTasks, HTTPException
 from pydantic import BaseModel
-import uvicorn
 
-app = FastAPI()
+app = FastAPI(title="Sofia AI Backend")
 
-class GenerateRequest(BaseModel):
-    prompt: str
+class MessageRequest(BaseModel):
+    platform: str
+    message: str
+    user_id: str
+    timestamp: str | None = None
+
+class ImageGenerationRequest(BaseModel):
+    prompt_type: str
+    custom_prompt: str | None = None
+    model: str = "FLUX.1-schnell"
 
 @app.get("/health")
-def health_check():
-    return {"status": "ok"}
+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})")
+    )
+    return {"status": "queued"}
+
+@app.post("/api/generate")
+async def api_generate(body: ImageGenerationRequest):
+    # TODO: aquí conectaremos con tu lógica real de generación (FLUX/SDXL)
+    # Por ahora devolvemos stub
+    if not body.custom_prompt and not body.prompt_type:
+        raise HTTPException(status_code=400, detail="prompt_type or custom_prompt required")
 
-@app.post("/generate-image")
-def generate_image(req: GenerateRequest):
-    # Aquí luego conectamos con tu lógica real de generación (FLUX / SD)
     return {
-        "status": "received",
-        "prompt": req.prompt,
-        "image_url": "https://example.com/fake-image-url"
+        "status": "pending",
+        "message": "Image generation not yet implemented",
+        "prompt_type": body.prompt_type,
+        "model": body.model,
     }
 
 if __name__ == "__main__":
-    # Hugging Face espera que el servidor escuche en 0.0.0.0:7860
+    import uvicorn
+    # Importante: usar puerto 7860 para Hugging Face Spaces
     uvicorn.run(app, host="0.0.0.0", port=7860)