Update app.py

app.py

@@ -312,27 +312,6 @@ def create_filter_controls(df, sort_column):
     elif sort_column == 'Article Title':
         # Only alphabetical sorting, no filtering
         pass
-
-
-    elif sort_column == 'Times Cited':
-        # Cited count range slider
-        cited_min = int(df['Times Cited'].min())
-        cited_max = int(df['Times Cited'].max())
-        col1, col2 = st.columns(2)
-        with col1:
-            start_cited = st.number_input('From Cited Count', 
-                min_value=cited_min, 
-                max_value=cited_max,
-                value=cited_min)
-        with col2:
-            end_cited = st.number_input('To Cited Count', 
-                min_value=cited_min, 
-                max_value=cited_max,
-                value=cited_max)
-        filtered_df = filtered_df[
-            (filtered_df['Times Cited'] >= start_cited) & 
-            (filtered_df['Times Cited'] <= end_cited)
-        ]
     
     return filtered_df
 
@@ -380,34 +359,12 @@ def main():
                 st.header("📝 Individual Paper Summaries")
                 
                 # Generate summaries if not already done
-                if st.session_state.summaries is None:
-                    try:
-                        with st.spinner("Generating individual paper summaries..."):
-                            model, tokenizer = load_model("summarize")
-                            summaries = []
-                            progress_bar = st.progress(0)
-                            
-                            for idx, abstract in enumerate(df['Abstract']):
-                                summary = improve_summary_generation(abstract, model, tokenizer)
-                                summaries.append(summary)
-                                progress_bar.progress((idx + 1) / len(df))
-                            
-                            st.session_state.summaries = summaries
-                            cleanup_model(model, tokenizer)
-                            progress_bar.empty()
-                            
-                    except Exception as e:
-                        st.error(f"Error generating summaries: {str(e)}")
-                        st.session_state.processing_started = False
-                
-                # Display summaries with improved sorting and filtering
                 if st.session_state.summaries is not None:
                     col1, col2 = st.columns(2)
                     with col1:
                         sort_options = ['Article Title', 'Authors', 'Publication Year', 'Source Title', 'Times Cited']
                         sort_column = st.selectbox("Sort/Filter by:", sort_options)
                     with col2:
-                        # Only show A-Z/Z-A option for Article Title
                         if sort_column == 'Article Title':
                             ascending = st.radio(
                                 "Sort order",
@@ -417,9 +374,9 @@ def main():
                         elif sort_column == 'Times Cited':
                             ascending = st.radio(
                                 "Sort order",
-                                ["Most cited", "Least cited"],
+                                ["Most cited first", "Least cited first"],
                                 horizontal=True
-                            ) == "Least cited"
+                            ) == "Least cited first"
                         else:
                             ascending = True  # Default for other columns
                     
@@ -429,10 +386,18 @@ def main():
                     display_df['Publication Year'] = display_df['Publication Year'].astype(int)
                     display_df.rename(columns={'Times Cited, All Databases': 'Times Cited'}, inplace=True)
                     display_df['Times Cited'] = display_df['Times Cited'].fillna(0).astype(int)
-                
+                    
                     # Apply filters
                     filtered_df = create_filter_controls(display_df, sort_column)
                     
+                    # Apply sorting
+                    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
+                    
                     if sort_column == 'Article Title':
                         # Sort alphabetically
                         sorted_df = filtered_df.sort_values(by=sort_column, ascending=ascending)