diff --git a/main.py b/main.py index 0055fd0..709c5c9 100644 --- a/main.py +++ b/main.py @@ -129,7 +129,7 @@ async def upload_image(file: UploadFile, request: Request, token: str = Depends( raise HTTPException(status_code=401, detail="Unauthorized") if not file: raise HTTPException(status_code=400, detail="No file uploaded") - if not file.filename.lower().endswith(('.jpg', '.jpeg', '.png', '.gif')): + if not file.filename.lower().endswith(('.jpg', '.jpeg', '.png', '.gif')) or file.content_type not in ["image/jpeg", "image/png", "image/gif"]: raise HTTPException(status_code=400, detail="Unsupported file type") try: content = await file.read()