Update app/api/image_router.py

app/api/image_router.py

@@ -2,9 +2,10 @@ import logging
 from fastapi import APIRouter, HTTPException, Response, UploadFile, File, Form
 from fastapi.responses import StreamingResponse, JSONResponse
 from app.schemas import image as image_schemas
-from app.services import image_service
+from app.schemas.image import ShoeGenerateRequest
 from io import BytesIO
-from typing import Optional, Dict, Any
+from app.services import image_service
+from app.services.image_service import generate_shoe_images
 
 logger = logging.getLogger(__name__)
 
@@ -157,4 +158,42 @@ async def virtual_try_on(
         raise
     except Exception as e:
         logger.error(f"Error in /virtual-try-on: {e}", exc_info=True)
-        raise HTTPException(status_code=500, detail=str(e))
+        raise HTTPException(status_code=500, detail=str(e))
+    
+
+@router.post("/generate-shoe", response_class=StreamingResponse)
+async def generate_shoe(request: ShoeGenerateRequest):
+    """
+    Generate shoe product images from a textual description.
+    Streams the first generated image back; if no image, returns JSON with model text.
+    """
+    logger.info("Received request to /generate-shoe: %s", request.style[:80])
+
+    if not request.style:
+        raise HTTPException(status_code=400, detail="`style` is required.")
+
+    try:
+        gen_text, images = generate_shoe_images(
+            style=request.style,
+            color=request.color,
+            perspective=request.perspective,
+            variations=request.variations or 1,
+        )
+
+        if images and len(images) > 0:
+            # Stream the first image (consistent with other endpoints)
+            img_io: BytesIO = images[0]
+            img_io.seek(0)
+            return StreamingResponse(img_io, media_type="image/png")
+        else:
+            # If no image was produced, return any model text or an error summary as JSON
+            detail = {"success": False, "notes": "No image generated."}
+            if gen_text:
+                detail["model_text"] = gen_text
+            return JSONResponse(status_code=500, content=detail)
+
+    except HTTPException:
+        raise
+    except Exception as e:
+        logger.exception("Error in /generate-shoe: %s", e)
+        raise HTTPException(status_code=500, detail=str(e))