Update app.py

app.py

@@ -402,44 +402,44 @@ def main():
                 
                 # 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",
-                                    ["A to Z", "Z to A"],
-                                    horizontal=True
-                                ) == "A to Z"
-                            elif sort_column == 'Times Cited':
-                                ascending = st.radio(
-                                    "Sort order",
-                                    ["Most cited", "Least cited"],
-                                    horizontal=True
-                                ) == "Least cited"
-                            else:
-                                ascending = True  # Default for other columns
-                        
-                        # Create display dataframe
-                        display_df = df.copy()
-                        display_df['Summary'] = st.session_state.summaries
-                        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)
-                        
+                    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':
-                            # Sort alphabetically
-                            sorted_df = filtered_df.sort_values(by=sort_column, ascending=ascending)
+                            ascending = st.radio(
+                                "Sort order",
+                                ["A to Z", "Z to A"],
+                                horizontal=True
+                            ) == "A to Z"
+                        elif sort_column == 'Times Cited':
+                            ascending = st.radio(
+                                "Sort order",
+                                ["Most cited", "Least cited"],
+                                horizontal=True
+                            ) == "Least cited"
                         else:
-                            # Keep original order for other columns after filtering		
-                		  	# Keep original order for other columns after filtering
-                                sorted_df = filtered_df
+                            ascending = True  # Default for other columns
+                    
+                    # Create display dataframe
+                    display_df = df.copy()
+                    display_df['Summary'] = st.session_state.summaries
+                    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)
+                    
+                    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):
@@ -478,33 +478,33 @@ def main():
                     
                     # Display papers using the filtered and sorted dataframe
                     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>Times Cited:</strong> {row['Times Cited']}<br>
-                                        <strong>DOI:</strong> {row['DOI'] if pd.notna(row['DOI']) else 'None'}
-                                    </div>
+                        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>Times Cited:</strong> {row['Times Cited']}<br>
+                                    <strong>DOI:</strong> {row['DOI'] if pd.notna(row['DOI']) else 'None'}
                                 </div>
-                                """, unsafe_allow_html=True)
+                            </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 spacing between papers
-                            st.markdown("<div style='margin-bottom: 20px;'></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 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():