Update app.py

app.py

@@ -312,6 +312,7 @@ def create_filter_controls(df, sort_column):
     elif sort_column == 'Article Title':
         # Only alphabetical sorting, no filtering
         pass
+
     
     return filtered_df
 
@@ -379,7 +380,7 @@ def main():
                         st.error(f"Error generating summaries: {str(e)}")
                         st.session_state.processing_started = False
                 
-                # Display summaries if they exist
+                # Display summaries with improved sorting and filtering
                 if st.session_state.summaries is not None:
                     col1, col2 = st.columns(2)
                     with col1:
@@ -401,7 +402,6 @@ def main():
                         else:
                             ascending = True  # Default for other columns
                     
-                    # Create display dataframe
                     # Create display dataframe
                     display_df = df.copy()
                     display_df['Summary'] = st.session_state.summaries
@@ -413,7 +413,9 @@ def main():
                     filtered_df = create_filter_controls(display_df, sort_column)
                     
                     # Apply sorting
-                    if sort_column in ['Times Cited', 'Article Title']:
+                    if sort_column == 'Times Cited':
+                        sorted_df = filtered_df.sort_values(by=sort_column, ascending=ascending)
+                    elif sort_column == 'Article Title':
                         sorted_df = filtered_df.sort_values(by=sort_column, ascending=ascending)
                     else:
                         sorted_df = filtered_df