Create routers/images.py

routers/images.py

@@ -0,0 +1,44 @@
+# routers/images.py
+from fastapi import APIRouter, Depends, HTTPException, status
+from starlette.concurrency import run_in_threadpool
+import models
+import config
+from services import generation, security
+import logging
+
+logger = logging.getLogger(__name__)
+router = APIRouter(
+    prefix="/images",
+    tags=["Image Generation"],
+    dependencies=[Depends(security.get_api_key)]
+)
+
+@router.post("/text-to-image", response_model=models.ImageResponse, status_code=status.HTTP_201_CREATED)
+async def create_image_from_text(request: models.ImageRequest):
+    """
+    Generates an image based on a text prompt using a diffusion model.
+    """
+    try:
+        logger.info(f"Received text-to-image request: prompt='{request.prompt}'")
+        image_base64 = await run_in_threadpool(
+            generation.generate_image_sync,
+            prompt=request.prompt,
+            negative_prompt=request.negative_prompt,
+            height=request.height,
+            width=request.width,
+            num_inference_steps=request.num_inference_steps,
+            guidance_scale=request.guidance_scale
+        )
+        logger.info(f"Successfully generated image for prompt: '{request.prompt}'")
+        return models.ImageResponse(
+            image_base64=image_base64,
+            prompt=request.prompt,
+            model_name=config.IMAGE_MODEL_NAME,
+            format="PNG"
+        )
+    except RuntimeError as e:
+         logger.error(f"Model runtime error during image generation: {e}", exc_info=True)
+         raise HTTPException(status_code=status.HTTP_503_SERVICE_UNAVAILABLE, detail=f"Model inference failed: {e}")
+    except Exception as e:
+        logger.error(f"Unexpected error during image generation: {e}", exc_info=True)
+        raise HTTPException(status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, detail=f"An unexpected error occurred: {e}")