VTO / app /api /routers /v1 /multi_view.py
Akshajzclap's picture
Upload 89 files
bd55a25 verified
from fastapi import APIRouter, File, UploadFile, HTTPException, Form
from fastapi.responses import StreamingResponse
from app.services.image_generation_service import generate_image_from_files_and_prompt
from app.services.prompt_factory import MULTI_VIEW_TEMPLATE
from app.schemas.common_schemas import ModelSchema, CameraSchema,GarmentType
import io
from typing import List, Annotated
router = APIRouter()
@router.post("/multi_view", tags=["Multi_view"])
async def multi_view(
garment_images: List[UploadFile] = File(..., description="Exactly one image of the garments (e.g., dress /bag/shoes).", min_items=1, max_items=1),
):
# Validation for exactly two files (FastAPI also handles min_items/max_items for File, but explicit check is fine)
if not garment_images or len(garment_images) != 1:
raise HTTPException(status_code=400, detail="Exactly one garment images are required for outfit match.")
image_bytes_list = []
for image_file in garment_images: # No need to check if garment_images is None here
if not image_file.content_type.startswith("image/"):
raise HTTPException(status_code=400, detail=f"Invalid file type for garment: {image_file.filename}. Must be an image.")
content = await image_file.read()
image_bytes_list.append((content, image_file.filename)) # Store as tuple (bytes, filename)
prompt=MULTI_VIEW_TEMPLATE
generated_image_data = await generate_image_from_files_and_prompt(
image_files=image_bytes_list,
prompt=prompt
)
if generated_image_data:
return StreamingResponse(io.BytesIO(generated_image_data), media_type="image/png")
else:
raise HTTPException(status_code=500, detail="Failed to generate image. Check service logs.")