Update app.py

app.py

@@ -51,14 +51,10 @@ def getPrediction(input):
     return result, confidence_score
 
 def getSentiment(idx):
-   match idx:
-        case 0:
-            return "Negative"
-        case 1:
-            return "Positive"
-        case default:
-            return "Neutral"
+   return {0: "Negative", 1: "Positive", 2: "Neutral"}.get(idx, "Neutral")
        
+# Streamlit UI
+st.title("Sentiment Analysis")
 text = st.text_area("Enter Text...")
 
 if text:
@@ -67,3 +63,13 @@ if text:
     response = {"prediction": getSentiment(prediction[0]) + " Statement",
                 "confidence": "{:.2f}".format(float(confidence_score[0] * 100)) + "%"}  # Adjust as necessary for output formatting
     st.json(response)
+
+# Add a POST endpoint
+def api_predict(text):
+    prediction, confidence_score = getPrediction(text)
+    return {
+        "prediction": getSentiment(prediction) + " Statement",
+        "confidence": f"{confidence_score * 100:.2f}%"
+    }
+
+st.experimental_set_query_params(api_predict=api_predict)