Update app.py

app.py

@@ -108,91 +108,66 @@ class StreamlitDocProcessor:
     def upload_and_process_page(self):
         st.header("Upload and Process Documents")
         
-        # Add instructions
-        with st.expander("Instructions", expanded=True):
-            st.write("""
-            1. Click 'Browse files' to select documents
-            2. You can select multiple files at once
-            3. Supported formats: PDF and XML
-            4. Wait for processing to complete
-            5. Processed files will be listed below
-            """)
-        
-        # File uploader
         uploaded_files = st.file_uploader(
             "Upload PDF or XML files",
             type=['pdf', 'xml'],
-            accept_multiple_files=True,
-            help="Select one or more PDF or XML files to upload"
+            accept_multiple_files=True
         )
-
+    
         if uploaded_files:
             for uploaded_file in uploaded_files:
-                # Create progress bar and status container
-                col1, col2 = st.columns([3, 1])
-                with col1:
-                    progress_bar = st.progress(0)
-                with col2:
-                    status_text = st.empty()
+                # Create progress containers
+                progress_bar = st.progress(0)
+                status_container = st.empty()
                 
-                # Check if file is already processed
-                file_ext = os.path.splitext(uploaded_file.name)[1][1:]  # Get extension without dot
+                file_ext = os.path.splitext(uploaded_file.name)[1][1:]
                 if uploaded_file.name not in st.session_state.processed_files.get(file_ext, []):
                     try:
-                        # Create a temporary file
                         with tempfile.NamedTemporaryFile(delete=False, suffix=os.path.splitext(uploaded_file.name)[1]) as tmp_file:
                             tmp_file.write(uploaded_file.getbuffer())
                             temp_path = tmp_file.name
-
-                        # Process the file
-                        status_text.info('Processing...')
-                        progress_bar.progress(25)
+    
+                        status_container.text(f'Processing {uploaded_file.name}...')
                         
-                        result = st.session_state.processor.process_file(temp_path)
-                        progress_bar.progress(75)
+                        if file_ext == 'xml':
+                            # Add processing status updates
+                            status_container.text('Parsing XML...')
+                            progress_bar.progress(10)
+                            
+                            result = st.session_state.processor.process_file(temp_path)
+                            
+                            if result['success']:
+                                total_chunks = result['total_chunks']
+                                for i, chunk_result in enumerate(result['results']):
+                                    # Update progress for each batch
+                                    progress = min(95, int(10 + (85 * (i / total_chunks))))
+                                    progress_bar.progress(progress)
+                                    status_container.text(f'Processing chunk {i+1}/{total_chunks}...')
+                        else:
+                            # Regular PDF processing
+                            result = st.session_state.processor.process_file(temp_path)
                         
                         if result['success']:
                             if file_ext not in st.session_state.processed_files:
                                 st.session_state.processed_files[file_ext] = []
                             st.session_state.processed_files[file_ext].append(uploaded_file.name)
                             progress_bar.progress(100)
-                            status_text.success("✓ Success")
+                            status_container.success(f"Successfully processed {uploaded_file.name}")
                         else:
                             progress_bar.progress(100)
-                            status_text.error("✗ Failed")
-                            st.error(f"Failed to process {uploaded_file.name}: {result['error']}")
-
+                            status_container.error(f"Failed to process {uploaded_file.name}: {result['error']}")
+    
                     except Exception as e:
-                        status_text.error("✗ Error")
-                        st.error(f"Error processing {uploaded_file.name}: {str(e)}")
+                        status_container.error(f"Error processing {uploaded_file.name}: {str(e)}")
                     finally:
-                        # Clean up temporary file
                         try:
                             os.unlink(temp_path)
                         except:
                             pass
                 else:
-                    status_text.info("Already processed")
+                    status_container.info(f"{uploaded_file.name} has already been processed")
                     progress_bar.progress(100)
 
-        # Display processed files
-        if any(st.session_state.processed_files.values()):
-            st.subheader("Processed Files")
-            
-            col1, col2 = st.columns(2)
-            
-            with col1:
-                if st.session_state.processed_files.get('xml'):
-                    st.write("📱 XML Files:")
-                    for file in sorted(st.session_state.processed_files['xml']):
-                        st.text(f"  • {file}")
-                        
-            with col2:
-                if st.session_state.processed_files.get('pdf'):
-                    st.write("📄 PDF Files:")
-                    for file in sorted(st.session_state.processed_files['pdf']):
-                        st.text(f"  • {file}")
-
     def qa_page(self):
         st.header("Query Documents")