first commit

routers/images.py

@@ -0,0 +1,34 @@
+from fastapi import APIRouter, HTTPException
+from pydantic import BaseModel
+
+from services import sdbackend as sd_service
+from services.memory import get_last_assistant_message_id, update_message_image
+
+router = APIRouter(prefix="/images", tags=["images"])
+
+
+class GenerateRequest(BaseModel):
+    session_id: str
+    prompt: str
+
+
+@router.get("/health")
+async def sd_health():
+    ok = await sd_service.check_sd()
+    return {"sd_available": ok, "url": sd_service.SD_BASE_URL}
+
+
+@router.post("/generate")
+async def generate_image(req: GenerateRequest):
+    if not req.prompt.strip():
+        raise HTTPException(status_code=400, detail="Пустой промпт")
+
+    rel, err = await sd_service.generate_from_full_prompt(req.prompt)
+    if not rel:
+        raise HTTPException(status_code=502, detail=err or "SD backend недоступен")
+
+    msg_id = await get_last_assistant_message_id(req.session_id)
+    if msg_id:
+        await update_message_image(msg_id, rel)
+
+    return {"image_path": f"/static/{rel}", "status": "ok"}