feat: sentiment engine

src/blackbox/sentiment.py

@@ -0,0 +1,31 @@
+from typing import Any, Coroutine
+
+from fastapi import Request, Response, status
+from fastapi.responses import JSONResponse
+
+from sentiment_engine.sentiment_engine import SentimentEngine
+from .blackbox import Blackbox
+
+
+class Sentiment(Blackbox):
+
+    def __init__(self) -> None:
+        self.engine = SentimentEngine('resources/sentiment_engine/models/paimon_sentiment.onnx')
+
+    def valid(self, data: any) -> bool:
+        return isinstance(data, str)
+    
+    def processing(self, text: any) -> int:
+        return int(self.engine.infer(text))
+    
+    async def fast_api_handler(self, request) -> Response:
+        try:
+            data = await request.json()
+        except:
+            return JSONResponse(content={"error": "json parse error"}, status_code=status.HTTP_400_BAD_REQUEST)
+        text = data.get("text")
+        if text is None:
+            return JSONResponse(content={"error": "text is required"}, status_code=status.HTTP_400_BAD_REQUEST)
+        sentiment = self.processing(text)
+        return JSONResponse(content={"sentiment": sentiment }, status_code=status.HTTP_200_OK)
+