Update app.py

app.py

@@ -1,6 +1,9 @@
 from fastapi import FastAPI
 from pydantic import BaseModel
-from vaderSentiment.vaderSentiment import SentimentIntensityAnalyzer
+
+import joblib
+import pandas as pd
+import maven_text_preprocessing
 
 app = FastAPI()
 
@@ -13,5 +16,17 @@ def greet_json():
 
 @app.post("/sentiment")
 def sentiment_analysis(payload: SentimentRequest):
-    analyzer = SentimentIntensityAnalyzer()
-    return analyzer.polarity_scores(payload.message)
+    
+    model = joblib.load("logistic_model.joblib")
+    vectorizer = joblib.load("vectorizer.joblib")
+
+    dirty_text_series = pd.Series([payload.message])
+    clean_text = maven_text_preprocessing.clean_and_normalize(dirty_text_series)
+
+    X = vectorizer.transform(clean_text)  # ⚠️ transform, NOT fit_transform
+
+    predictions = model.predict(X)
+    pred_prob = model.predict_proba(X)
+
+    return [predictions.tolist()[0],pred_prob.tolist()[0][0],pred_prob.tolist()[0][1]]
+