Update app.py

app.py

@@ -66,31 +66,36 @@ if file is not None:
         unsafe_allow_html=True
     )
 
+
     # 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
-    filtered_df['sentiment'] = filtered_df['sentiment'].apply(lambda x: f'<span style="color: {sentiment_colors[x]}">{x}</span>')
+    df['sentiment'] = 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 = (filtered_df.style
+    table_html = (df.style
                   .set_properties(**{'text-align': 'center'})
-                  .set_table_styles(
-                        [{'selector': 'th', 'props': [('border', '0px')]}, {'selector': 'td', 'props': [('border', '0px')]}])
+                  .set_table_styles([{'selector': 'th', 'props': [('border', '0px')]},
+                                     {'selector': 'td', 'props': [('border', '0px')]}])
                   .to_html(escape=False))
     
-    # Display the table
+    # Add the title of the table
+    st.title("Sentiment Analysis Results")
+    
+    # Create a space in the sidebar
+    empty_slot = st.sidebar.empty()
+    
+    # Add the selectbox to the space created above
+    filter_sentiment = empty_slot.selectbox("Filter Sentiment", ["All"] + list(sentiment_colors.keys()))
+    
+    # Filter the dataframe based on the selected sentiment
+    if filter_sentiment != "All":
+        df = df[df['sentiment'].str.contains(filter_sentiment)]
+    
+    # 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.")