Update app.py

app.py

@@ -31,6 +31,7 @@ def load_model(model_type):
     try:
         # Clear any existing cached data
         gc.collect()
+        torch.cuda.empty_cache()
         
         device = "cpu"  # Force CPU usage
         
@@ -150,7 +151,7 @@ def generate_summary(text, model, tokenizer):
                 "num_beams": 4,
                 "length_penalty": 2.0,
                 "early_stopping": True,
-                "no_repeat_ngram_size": 3  # Prevent repetition of phrases
+                "no_repeat_ngram_size": 3
             }
         )
     
@@ -229,7 +230,33 @@ def main():
                 # Individual Summaries Section
                 st.header("📝 Individual Paper Summaries")
                 
-                # Replace the display section in main() with this:
+                # Generate summaries if not already done
+                if st.session_state.summaries is None:
+                    try:
+                        with st.spinner("Generating individual paper summaries..."):
+                            # Load summarization model
+                            model, tokenizer = load_model("summarize")
+                            
+                            # Generate summaries for each abstract
+                            summaries = []
+                            progress_bar = st.progress(0)
+                            
+                            for idx, abstract in enumerate(df['Abstract']):
+                                summary = generate_summary(abstract, model, tokenizer)
+                                summaries.append(summary)
+                                progress_bar.progress((idx + 1) / len(df))
+                            
+                            # Store summaries in session state
+                            st.session_state.summaries = summaries
+                            
+                            # Cleanup
+                            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  # Reset to allow retry
+                
                 # Display summaries with improved sorting
                 if st.session_state.summaries is not None:
                     col1, col2 = st.columns(2)