From 650f8d10db2a720e76e1f94ba300d52802daa0ba Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Artem=20=D0=9Cukhametgaliev?= <airmareing@gmail.com>
Date: Sat, 25 May 2024 21:11:51 +0300
Subject: [PATCH] Update main.py

---
 main.py | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/main.py b/main.py
index be946ec..3f7f5b8 100644
--- a/main.py
+++ b/main.py
@@ -1,7 +1,8 @@
-from fastapi import FastAPI
+from fastapi import FastAPI, Request
 from transformers import pipeline
 from pydantic import BaseModel
-
+from fastapi.responses import JSONResponse
+from fastapi.exceptions import RequestValidationError
 
 class Item(BaseModel):
     text: str
@@ -11,7 +12,12 @@ class Item(BaseModel):
 app = FastAPI()
 classifier = pipeline("sentiment-analysis")
 
-
+@app.exception_handler(RequestValidationError)
+async def validation_exception_handler(request: Request, exc: RequestValidationError):
+    return JSONResponse(
+        status_code=400,
+        content={"detail": "Invalid request body"},
+    )
 
 @app.get("/")
 def root():