Update app.py

app.py

@@ -19,6 +19,8 @@ import matplotlib.pyplot as plt
 import seaborn as sns
 import ast
 
+nltk.download('stopwords')
+nltk.download('wordnet')
 
 # Load the model
 def load_model():
@@ -110,7 +112,23 @@ if sidebar_selection == "Ratings Prediction":
     # Check if the submit button is clicked and the input is not empty
     if submit_button and ratings_review:
         rating_pred = ratings(ratings_review)
-        st.write(f"The predicted average rating for a product with the list of reviews above is: {rating_pred}")
+        def get_rating_category(rating):
+            if rating < 2.0:
+                return "between 1 and 2 which is Very Low"
+            elif rating < 3.0:
+                return "between 2 and 3 which is Low"
+            elif rating < 4.0:
+                return "between 3 and 4 which is Medium"
+            elif rating < 5.0:
+                return "between 4 and 5 which is High"
+            else:
+                return "5 which is Very High"
+
+        # Determine the rating category
+        rating_category = get_rating_category(rating_pred)
+
+        # Display the result
+        st.write(f"Based on the list of reviews provided, your average rating falls {rating_category}.")
     elif submit_button:
         # Display a message if the submit button is clicked but no review is provided
         st.write("Please enter a review to get a prediction.")