feat: added post id to request to keep track of posts

app.py

@@ -9,20 +9,25 @@ sentiment_analyzer = pipeline("sentiment-analysis", model="nlptown/bert-base-mul
 
 class SentimentRequest(BaseModel):
     text: str
+    post_id: str
 
 @app.post("/sentiment")
 async def analyze_sentiment(request: SentimentRequest):
     result = sentiment_analyzer(request.text)
-    return {"label": result[0]["label"], "score": result[0]["score"]}
+    return {"post_id": request.post_id, "label": result[0]["label"], "score": result[0]["score"]}
 
+class SentimentPost(BaseModel):
+    post_id: str
+    text: str
 
 class SentimentBatchRequest(BaseModel):
-    texts: List[str]
+    posts: List[SentimentPost]
 
 @app.post("/sentiment/batch")
 async def analyze_sentiment_batch(request: SentimentBatchRequest):
-    results = sentiment_analyzer(request.texts)
-    return [{"text": text, "label": res["label"], "score": res["score"]} for text, res in zip(request.texts, results)]
+    texts = [post.text for post in request.posts]
+    results = sentiment_analyzer(texts)
+    return [{"text": post.text, "post_id": post.post_id, "label": res["label"], "score": res["score"]} for post, res in zip(request.posts, results)]
 
 @app.get("/")
 def greet_json():