Fix sample document processing and close functionality

- Fixed auto-processing of sample documents not running
- Improved Process button clarity based on document state
- Enhanced close button appearance and functionality
- Added tracking of sample document processing
- Fixed issues with document state management
- Added more descriptive button text

app.py

@@ -1203,6 +1203,8 @@ with main_tab1:
         st.session_state.sample_just_loaded = False
     if 'processed_document_active' not in st.session_state:
         st.session_state.processed_document_active = False
+    if 'sample_document_processed' not in st.session_state:
+        st.session_state.sample_document_processed = False
         
     # Use uploaded_file or sample_document if available
     if 'sample_document' in st.session_state and st.session_state.sample_document is not None:
@@ -1224,6 +1226,8 @@ with main_tab1:
         # Set auto-process flag in session state if this is a newly loaded sample
         if st.session_state.sample_just_loaded:
             st.session_state.auto_process_sample = True
+            # Mark that this is a sample document being processed
+            st.session_state.sample_document_processed = True
             st.session_state.sample_just_loaded = False
             
         # Clear sample document after use to avoid interference with future uploads
@@ -1242,7 +1246,11 @@ with main_tab1:
         
         # Process button - flush left with similar padding as file browser
         with left_col:
-            process_button = st.button("Process Document")
+            # Make the button more clear about its function
+            if st.session_state.processed_document_active:
+                process_button = st.button("Process Document Again")
+            else:
+                process_button = st.button("Process Document")
             
             # Empty container for progress indicators - will be filled during processing
             # Positioned right after the process button for better visibility
@@ -1289,17 +1297,18 @@ with main_tab1:
         
         # processed_document_active is already initialized at the top of the function
             
-        # Results section - process if button clicked or auto-process flag is set
-        process_now = process_button or st.session_state.auto_process_sample
+        # We'll determine processing logic below
         
-        # Show a message if auto-processing
-        if st.session_state.auto_process_sample:
+        # Check if this is an auto-processing situation
+        auto_processing = st.session_state.auto_process_sample and not st.session_state.processed_document_active
+            
+        # Show a message if auto-processing is happening
+        if auto_processing:
             st.info("Automatically processing sample document...")
             
-        # 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)
+        # Determine if we should process the document
+        # Either process button was clicked OR auto-processing is happening
+        should_process = process_button or auto_processing
         
         if should_process:
             # Reset auto-process flag to avoid processing on next rerun
@@ -2081,9 +2090,25 @@ with main_tab1:
                 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"):
+                    # Use a more visually distinctive close button
+                    st.markdown("""
+                    <style>
+                    div[data-testid="stButton"] button {
+                        background-color: #f8f9fa;
+                        border: 1px solid #dee2e6;
+                        padding: 0.25rem 0.5rem;
+                        font-size: 0.875rem;
+                        border-radius: 0.2rem;
+                    }
+                    </style>
+                    """, unsafe_allow_html=True)
+                    
+                    if st.button("✕ Close Document", key="close_document_button", help="Clear current document and start over"):
                         # Clear the session state
                         st.session_state.processed_document_active = False
+                        # Reset any active document data
+                        if 'current_result' in st.session_state:
+                            del st.session_state.current_result
                         # Rerun to reset the page
                         st.rerun()
                     
@@ -2092,6 +2117,12 @@ with main_tab1:
                 result_copy = result.copy()
                 result_copy['timestamp'] = datetime.now().strftime("%Y-%m-%d %H:%M")
                 
+                # Store if this was a sample document
+                if 'sample_document_processed' in st.session_state and st.session_state.sample_document_processed:
+                    result_copy['sample_document'] = True
+                    # Reset the flag
+                    st.session_state.sample_document_processed = False
+                
                 # Generate more descriptive file name for the result
                 original_name = Path(result.get('file_name', uploaded_file.name)).stem