Update app.py

app.py

@@ -359,6 +359,27 @@ 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 if they exist
                 if st.session_state.summaries is not None:
                     col1, col2 = st.columns(2)
                     with col1:
@@ -380,6 +401,7 @@ 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
@@ -391,21 +413,11 @@ def main():
                     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':
+                    if sort_column in ['Times Cited', '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)
-                    else:
-                        # Keep original order for other columns after filtering		
-                        # Keep original order for other columns after filtering
-                            sorted_df = filtered_df
-                    
                     # Show number of filtered results
                     if len(sorted_df) != len(display_df):
                         st.write(f"Showing {len(sorted_df)} of {len(display_df)} papers")