Add ML request model

Author: njouanin

backend/balanceteshaters/routers/ml_model.py

@@ -0,0 +1,28 @@
+from datetime import datetime
+
+from pydantic import BaseModel
+
+
+class Author(BaseModel):
+    name: str
+    accountHref: str
+
+
+class Comment(BaseModel):
+    textContent: str
+    publishedAt: datetime
+    scrapedAt: datetime
+    screenshotData: str
+    nbLikes: int
+    replies: list["Comment"]
+
+
+class Post(BaseModel):
+    url: str
+    publishedAt: datetime
+    scrapedAt: datetime
+    textContent: str
+    postId: str
+    socialNetwork: str
+    author: Author
+    comments: list[Comment]

backend/balanceteshaters/routers/ml_router.py

@@ -1,10 +1,11 @@
+from balanceteshaters.routers.ml_model import Post
 from fastapi import APIRouter
 
 router = APIRouter()
 
 
 @router.post("/post")
-def ml_post(publication: str):
+def ml_post(publication: Post):
     results = []
     print(results)
     return {"result": results}