Update shadow_routes.py

shadow_routes.py

@@ -130,9 +130,8 @@
 #         traceback.print_exc()
 #         return JSONResponse(status_code=500, content={"error": str(exc)})
 
-
 from fastapi import APIRouter, File, UploadFile, Form
-from fastapi.responses import Response
+from fastapi.responses import Response, JSONResponse
 import shadow_generator
 from typing import Optional
 import io
@@ -142,33 +141,38 @@ router = APIRouter()
 @router.post("/generate-shadow")
 async def generate_shadow_route(
     file: UploadFile = File(...),
+    style: Optional[str] = Form("cast"),        # NEW: allow drop / cast / soft / bent
     softness: Optional[float] = Form(28.0),
     opacity: Optional[float] = Form(0.7),
     color: Optional[str] = Form("#000000"),
-    direction: Optional[float] = Form(0.0) # Placeholder for direction
+    direction: Optional[float] = Form(0.0),
+    distance: Optional[float] = Form(80.0),     # NEW: needed for cast shadows
 ):
     """
-    This endpoint receives an image and shadow parameters,
-    and returns the image with a generated shadow.
+    Generate a shadowed image from input.
+    Styles: 'drop', 'cast', 'angled', 'soft'
     """
     image_bytes = await file.read()
 
     params = {
+        "style": style,
         "softness": softness,
         "opacity": opacity,
         "color": color,
-        "direction": direction, # To be implemented
+        "direction": direction,
+        "distance": distance,
     }
 
     try:
         output_image_bytes = shadow_generator.generate_shadow_rgba(image_bytes, params)
         return Response(content=output_image_bytes, media_type="image/png")
     except Exception as e:
-        return {"error": str(e)}
+        return JSONResponse(status_code=500, content={"error": str(e)})
+
 
 @router.on_event("startup")
 async def startup_event():
     """
-    Initialize the model on startup.
+    Initialize model once on server start.
     """
     shadow_generator.initialize_once()