Update app.py

app.py

@@ -66,22 +66,23 @@ if file is not None:
         unsafe_allow_html=True
     )
 
+    # Add the selectbox to filter sentiments
+    filter_sentiment = st.selectbox("Filter Sentiments", ["All", "POSITIVE", "NEGATIVE"])
+    
+    # Filter the dataframe based on the selected sentiment
+    if filter_sentiment != "All":
+        df = df[df['sentiment'] == filter_sentiment]
+    
     # Create HTML table with no border and centered text
     table_html = (df.style
                   .set_properties(**{'text-align': 'left','font-size': '14px'})
                   .set_table_styles([{'selector': 'th', 'props': [('border', '0px')]},
                                      {'selector': 'td', 'props': [('border', '0px')]}])
-                  .to_html(escape=False))
-    
-    # Add the selectbox to filter sentiments
-    filter_sentiment = st.selectbox("Filter Sentiments", ["All"] + list(Sentiment.keys()))
-    
-    # Filter the dataframe based on the selected sentiment
-    if filter_sentiment != "All":
-        df = df[df['Sentiment'].str.contains(filter_sentiment)]
+                  .to_html(index=False, escape=False))
     
     # Display the table and the selectbox widget beside the title
     st.write(table_html, unsafe_allow_html=True)
 
+
 else:
     st.write("Please upload a CSV file.")