Update app.py

app.py

@@ -235,34 +235,7 @@ def main():
                 # Individual Summaries Section
                 st.header("📝 Individual Paper Summaries")
                 
-                if st.session_state.summaries is None:
-                    try:
-                        with st.spinner("Generating summaries..."):
-                            # Load summarization model
-                            model, tokenizer = load_model("summarize")
-                            
-                            # Process abstracts with real-time updates
-                            summaries = []
-                            progress_bar = st.progress(0)
-                            summary_display = st.empty()
-                            
-                            for i, (_, row) in enumerate(df.iterrows()):
-                                summary = generate_summary(row['Abstract'], model, tokenizer)
-                                summaries.append(summary)
-                                
-                                # Update progress and show current summary
-                                progress = (i + 1) / len(df)
-                                progress_bar.progress(progress)
-                                summary_display.write(f"Processing paper {i+1}/{len(df)}:\n{row['Article Title']}")
-                                
-                            st.session_state.summaries = summaries
-                            
-                            # Cleanup first model
-                            cleanup_model(model, tokenizer)
-                    
-                    except Exception as e:
-                        st.error(f"Error generating summaries: {str(e)}")
-                
+                # Replace the display section in main() with this:
                 # Display summaries with improved sorting
                 if st.session_state.summaries is not None:
                     col1, col2 = st.columns(2)
@@ -278,24 +251,65 @@ def main():
                     display_df['Publication Year'] = display_df['Publication Year'].astype(int)
                     sorted_df = display_df.sort_values(by=sort_column, ascending=ascending)
                     
-                    # Apply custom formatting
+                    # Apply custom styling
                     st.markdown("""
                     <style>
-                    .stDataFrame {
-                        font-size: 16px;
+                    .paper-info {
+                        border: 1px solid #ddd;
+                        padding: 15px;
+                        margin-bottom: 20px;
+                        border-radius: 5px;
+                    }
+                    .paper-section {
+                        margin-bottom: 10px;
+                    }
+                    .section-header {
+                        font-weight: bold;
+                        color: #555;
+                        margin-bottom: 8px;
                     }
-                    .stDataFrame td {
-                        white-space: normal !important;
-                        padding: 8px !important;
+                    .paper-title {
+                        margin-top: 5px;
+                        margin-bottom: 10px;
+                    }
+                    .paper-meta {
+                        font-size: 0.9em;
+                        color: #666;
+                    }
+                    .doi-link {
+                        color: #0366d6;
                     }
                     </style>
                     """, unsafe_allow_html=True)
                     
-                    st.dataframe(
-                        sorted_df[['Article Title', 'Authors', 'Source Title', 
-                                 'Publication Year', 'DOI', 'Summary']],
-                        hide_index=True
-                    )
+                    # Display papers in side-by-side layout
+                    for _, row in sorted_df.iterrows():
+                        paper_info_cols = st.columns([1, 1])
+                        
+                        with paper_info_cols[0]:  # PAPER column
+                            st.markdown('<div class="paper-section"><div class="section-header">PAPER</div>', unsafe_allow_html=True)
+                            st.markdown(f"""
+                            <div class="paper-info">
+                                <div class="paper-title">{row['Article Title']}</div>
+                                <div class="paper-meta">
+                                    <strong>Authors:</strong> {row['Authors']}<br>
+                                    <strong>Source:</strong> {row['Source Title']}<br>
+                                    <strong>Publication Year:</strong> {row['Publication Year']}<br>
+                                    <strong>DOI:</strong> {row['DOI'] if pd.notna(row['DOI']) else 'None'}
+                                </div>
+                            </div>
+                            """, unsafe_allow_html=True)
+
+                        with paper_info_cols[1]:  # SUMMARY column
+                            st.markdown('<div class="paper-section"><div class="section-header">SUMMARY</div>', unsafe_allow_html=True)
+                            st.markdown(f"""
+                            <div class="paper-info">
+                                {row['Summary']}
+                            </div>
+                            """, unsafe_allow_html=True)
+                        
+                        # Add some spacing between papers
+                        st.markdown("<div style='margin-bottom: 20px;'></div>", unsafe_allow_html=True)
                 
                 # Question-focused Summary Section (only if question provided)
                 if question.strip():