Add ability to dismiss processed sample documents

- Added close button to dismiss processed documents
- Enhanced sample document styling and indicators
- Added proper state management for active documents
- Improved logic for document processing flow
- Added helpful tooltips and styled buttons

app.py

@@ -1205,8 +1205,18 @@ with main_tab1:
     if 'sample_document' in st.session_state and st.session_state.sample_document is not None:
         # Use the sample document
         uploaded_file = st.session_state.sample_document
-        # Add a notice about using sample document
-        st.success(f"Using sample document: {uploaded_file.name}")
+        # Add a notice about using sample document with better style
+        st.markdown(
+            f"""
+            <div style="background-color: #D4EDDA; color: #155724; padding: 10px; 
+                 border-radius: 4px; border-left: 5px solid #155724; margin-bottom: 10px;">
+                <div style="display: flex; justify-content: space-between; align-items: center;">
+                    <span style="font-weight: bold;">Sample Document: {uploaded_file.name}</span>
+                </div>
+            </div>
+            """, 
+            unsafe_allow_html=True
+        )
         
         # Set auto-process flag in session state if this is a newly loaded sample
         if st.session_state.sample_just_loaded:
@@ -1269,11 +1279,23 @@ with main_tab1:
             # Container for success message (will be filled after processing)
             # No extra spacing needed as it will be managed programmatically
             metadata_placeholder = st.empty()
+            
+            # If there's a document already processed and active, show a button to close it
+            if st.session_state.processed_document_active:
+                if st.button("✕ Close Current Document", key="close_document_button_top", help="Clear current document and start over"):
+                    # Clear the processed document flag
+                    st.session_state.processed_document_active = False
+                    # Rerun to reset the page
+                    st.rerun()
         
         # Check if we need to auto-process a sample document
         if 'auto_process_sample' not in st.session_state:
             st.session_state.auto_process_sample = False
         
+        # Initialize processed_document_active if needed
+        if 'processed_document_active' not in st.session_state:
+            st.session_state.processed_document_active = False
+            
         # Results section - process if button clicked or auto-process flag is set
         process_now = process_button or st.session_state.auto_process_sample
         
@@ -1281,7 +1303,12 @@ with main_tab1:
         if st.session_state.auto_process_sample:
             st.info("Automatically processing sample document...")
             
-        if process_now:
+        # Process if: 
+        # 1. Button is clicked (overrides active document flag)
+        # 2. Auto processing is triggered and no active document is displayed
+        should_process = process_button or (st.session_state.auto_process_sample and not st.session_state.processed_document_active)
+        
+        if should_process:
             # Reset auto-process flag to avoid processing on next rerun
             if st.session_state.auto_process_sample:
                 st.session_state.auto_process_sample = False
@@ -2033,8 +2060,43 @@ with main_tab1:
                     # Close document content div
                     st.markdown('</div>', unsafe_allow_html=True)
                 
-                # Show a compact success message without extra container space
-                metadata_placeholder.success("**Document processed successfully**")
+                # Initialize processed_document_active in session state if needed
+                if 'processed_document_active' not in st.session_state:
+                    st.session_state.processed_document_active = True
+                else:
+                    # Set to True when new document is processed
+                    st.session_state.processed_document_active = True
+                
+                # Add CSS for styling close buttons
+                st.markdown("""
+                <style>
+                .close-button {
+                    background-color: #e7e7e7;
+                    color: #666;
+                    border: none;
+                    border-radius: 4px;
+                    padding: 0.25rem 0.5rem;
+                    font-size: 0.8rem;
+                    cursor: pointer;
+                    transition: all 0.2s ease;
+                }
+                .close-button:hover {
+                    background-color: #d7d7d7;
+                    color: #333;
+                }
+                </style>
+                """, unsafe_allow_html=True)
+                
+                # Display success message with close button for dismissing processed documents
+                success_cols = st.columns([5, 1])
+                with success_cols[0]:
+                    metadata_placeholder.success("**Document processed successfully**")
+                with success_cols[1]:
+                    if st.button("✕ Close", key="close_document_button", help="Clear current document and start over"):
+                        # Clear the session state
+                        st.session_state.processed_document_active = False
+                        # Rerun to reset the page
+                        st.rerun()
                     
                 # Store the result in the previous results list
                 # Add timestamp to result for history tracking