Update main.py

main.py

@@ -1,7 +1,14 @@
 from fastapi import FastAPI
+from pydantic import BaseModel
 
 app = FastAPI()
 
+class TextRatingRequest(BaseModel):
+    text: str
+    rating: int
+
 @app.post("/")
-def read_root():
+def read_root(text_rating: TextRatingRequest):
+    text = text_rating.text
+    rating = text_rating.rating
     return {"Hello": "World!"}