Update app.py

app.py

@@ -69,19 +69,28 @@ if file is not None:
     # Define sentiment colors
     sentiment_colors = {'POSITIVE': 'FFC845', 'NEGATIVE': '52565E'}
     
+    # Create a copy of the dataframe to use for filtering
+    filtered_df = df.copy()
+    
+    # Create selectbox to filter by sentiment
+    selected_sentiment = st.sidebar.selectbox('Filter by Sentiment', ['All'] + list(sentiment_colors.keys()))
+    
+    # If the user selects a specific sentiment, filter the dataframe
+    if selected_sentiment != 'All':
+        filtered_df = filtered_df[filtered_df['sentiment'] == selected_sentiment]
+    
     # Map sentiment to colours and apply to review column
-    df['sentiment'] = df['sentiment'].apply(lambda x: f'<span style="color: {sentiment_colors[x]}">{x}</span>')
+    filtered_df['sentiment'] = filtered_df['sentiment'].apply(lambda x: f'<span style="color: {sentiment_colors[x]}">{x}</span>')
     
     # Create HTML table with no border and centered text
-    table_html = (df.style
+    table_html = (filtered_df.style
                   .set_properties(**{'text-align': 'center'})
                   .set_table_styles(
-        [{'selector': 'th', 'props': [('border', '0px')]}, {'selector': 'td', 'props': [('border', '0px')]}])
-                  .render())
+                        [{'selector': 'th', 'props': [('border', '0px')]}, {'selector': 'td', 'props': [('border', '0px')]}])
+                  .to_html(escape=False))
     
     # Display the table
     st.write(table_html, unsafe_allow_html=True)
 
-
 else:
     st.write("Please upload a CSV file.")