Update app/api/image_router.py

app/api/image_router.py

@@ -178,29 +178,25 @@ async def virtual_try_on(
 @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.
+    Generate shoe product images from a raw prompt.
+    Streams the first generated image back.
     """
-    logger.info("Received request to /generate-shoe: %s", request.style[:80])
+    logger.info("Received request to /generate-shoe: %s", request.prompt[:80])
 
-    if not request.style:
-        raise HTTPException(status_code=400, detail="`style` is required.")
+    if not request.prompt:
+        raise HTTPException(status_code=400, detail="`prompt` field is required.")
 
     try:
-        gen_text, images = generate_shoe_images(
-            style=request.style,
-            color=request.color,
-            perspective=request.perspective,
-            variations=request.variations or 1,
-        )
+        # Call service with the raw prompt
+        gen_text, images = generate_shoe_images(prompt=request.prompt)
 
         if images and len(images) > 0:
-            # Stream the first image (consistent with other endpoints)
+            # Stream the first image
             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
+            # Handle failure / text-only response
             detail = {"success": False, "notes": "No image generated."}
             if gen_text:
                 detail["model_text"] = gen_text