Update utils/case_manager_interface.py

utils/case_manager_interface.py

@@ -75,6 +75,7 @@ class CaseManagerInterface:
                 st.divider()
 
     def _render_document_manager(self):
+        """Render document management interface with file preview."""
         # Case selection
         cases = self.case_manager.get_all_cases()
         if not cases:
@@ -134,45 +135,82 @@ class CaseManagerInterface:
             # Document list
             if documents:
                 st.subheader("Documents")
-                for doc in documents:
-                    with st.container():
-                        col1, col2, col3, col4 = st.columns([3, 1, 1, 1])
-                        
-                        with col1:
-                            st.markdown(f"**{doc['title']}**")
+                
+                # Add tabs for different document views
+                doc_tab1, doc_tab2 = st.tabs(["List View", "Grid View"])
+                
+                with doc_tab1:
+                    for doc in documents:
+                        with st.container():
+                            # Create a clickable document title that shows preview
+                            if st.markdown(f"[📄 {doc['title']}](#{doc['id']})", help="Click to preview"):
+                                with st.expander("Document Preview", expanded=True):
+                                    # Add tabs for different preview modes
+                                    preview_tab1, preview_tab2, preview_tab3 = st.tabs(["Content", "Metadata", "Chunks"])
+                                    
+                                    with preview_tab1:
+                                        st.text_area("Document Content", doc.get('text', 'No content available'), 
+                                                   height=300, key=f"content_{doc['id']}")
+                                    
+                                    with preview_tab2:
+                                        st.json(doc.get('metadata', {}))
+                                    
+                                    with preview_tab3:
+                                        chunks = doc.get('chunks', [])
+                                        for idx, chunk in enumerate(chunks):
+                                            st.markdown(f"**Chunk {idx + 1}**")
+                                            st.text_area(f"Chunk Content", 
+                                                       chunk.get('text', 'No content available'),
+                                                       height=100,
+                                                       key=f"chunk_{doc['id']}_{idx}")
+                            
+                            col1, col2, col3 = st.columns([1, 1, 1])
+                            
+                            with col1:
+                                if st.button("📥 Download", key=f"download_{doc['id']}"):
+                                    try:
+                                        st.download_button(
+                                            "Click to Download",
+                                            doc.get('text', ''),
+                                            file_name=doc['title'],
+                                            mime='text/plain',
+                                            key=f"download_button_{doc['id']}"
+                                        )
+                                    except Exception as e:
+                                        st.error(f"Error downloading: {str(e)}")
+                            
+                            with col2:
+                                if st.button("🔄 Reprocess", key=f"reprocess_{doc['id']}"):
+                                    try:
+                                        # Implement reprocessing logic
+                                        st.info("Reprocessing document...")
+                                    except Exception as e:
+                                        st.error(f"Error reprocessing: {str(e)}")
+                            
+                            with col3:
+                                if st.button("🗑️ Delete", key=f"delete_{doc['id']}"):
+                                    if st.warning("Are you sure you want to delete this document?"):
+                                        try:
+                                            self.vector_store.delete_document(doc['id'])
+                                            self.case_manager.remove_document(selected_case['id'], doc['id'])
+                                            st.success(f"Deleted: {doc['title']}")
+                                            st.rerun()
+                                        except Exception as e:
+                                            st.error(f"Error deleting: {str(e)}")
+                            
+                            st.divider()
+                
+                with doc_tab2:
+                    # Grid view of documents
+                    cols = st.columns(3)
+                    for idx, doc in enumerate(documents):
+                        with cols[idx % 3]:
+                            st.markdown(f"### 📄 {doc['title']}")
                             st.markdown(f"Added: {doc.get('added_at', 'Unknown')}")
-                        
-                        with col2:
-                            if st.button("👁️ View", key=f"view_{doc['id']}"):
-                                with st.expander("Document Content", expanded=True):
-                                    st.text_area("Content", doc.get('text', 'No content available'), height=300)
-                        
-                        with col3:
-                            if st.button("📥 Download", key=f"download_{doc['id']}"):
-                                try:
-                                    # Get document content
-                                    doc_content = doc.get('text', '')
-                                    # Provide download
-                                    st.download_button(
-                                        "Download",
-                                        doc_content,
-                                        file_name=doc['title'],
-                                        mime='text/plain'
-                                    )
-                                except Exception as e:
-                                    st.error(f"Error downloading: {str(e)}")
-                        
-                        with col4:
-                            if st.button("🗑️", key=f"del_doc_{doc['id']}"):
-                                try:
-                                    self.vector_store.delete_document(doc['id'])
-                                    self.case_manager.remove_document(selected_case['id'], doc['id'])
-                                    st.success(f"Deleted: {doc['title']}")
-                                    st.rerun()
-                                except Exception as e:
-                                    st.error(f"Error deleting: {str(e)}")
-                        
-                        st.divider()
+                            st.markdown(f"Size: {len(doc.get('text', ''))}")
+                            if st.button("Preview", key=f"grid_preview_{doc['id']}"):
+                                with st.expander("Document Preview", expanded=True):
+                                    st.text_area("Content", doc.get('text', ''), height=200)
             else:
                 st.info("No documents in this case yet.")