Update app.py

app.py

@@ -8,6 +8,9 @@ import nltk
 import json 
 import numpy as np
 import streamlit  as st
+from fastapi import FastAPI, HTTPException
+
+app = FastAPI()
 
 nltk.download('wordnet')
 # Load your TensorFlow model
@@ -52,6 +55,15 @@ def getPrediction(input):
 
 def getSentiment(idx):
    return {0: "Negative", 1: "Positive", 2: "Neutral"}.get(idx, "Neutral")
+
+
+@app.post("/predict")
+async def predict(text: str):
+    prediction, confidence_score = getPrediction(text)
+    return {
+        "prediction": getSentiment(prediction) + " Statement",
+        "confidence": f"{confidence_score * 100:.2f}%"
+    }
        
 # Streamlit UI
 st.title("Sentiment Analysis")