Update app.py

app.py

@@ -171,36 +171,41 @@ def main():
                         )
                         # In the Case Manager tab section, where documents are uploaded
                         if uploaded_file:
-                            with st.spinner("Processing document..."):
-                                try:
-                                    # Process document
-                                    text, chunks, metadata = doc_processor.process_and_tag_document(uploaded_file)
-                                    
-                                    # Add document to vector store
-                                    for chunk in chunks:
-                                        vector_store.add_document(
-                                            doc_id=metadata['doc_id'],
-                                            text=chunk['text'],
-                                            metadata={
-                                                'doc_id': metadata['doc_id'],
-                                                'chunk_id': chunk['chunk_id'],
-                                                'title': uploaded_file.name,
-                                                'type': metadata.get('document_type', 'unknown')
-                                            }
-                                        )
-                                    
-                                    # Add document to case
-                                    doc_data = {
-                                        "id": metadata['doc_id'],
-                                        "title": uploaded_file.name,
-                                        "text": text,
-                                        "metadata": metadata,
-                                        "chunks": chunks
-                                    }
-                                    case_manager.add_document(case["id"], doc_data)
-                                    st.success(f"Document '{uploaded_file.name}' processed and added successfully!")
-                                except Exception as e:
-                                    st.error(f"Error processing document: {str(e)}")
+                            if uploaded_file:
+                                with st.spinner("Processing document..."):
+                                    try:
+                                        # Process document
+                                        text, chunks, metadata = doc_processor.process_and_tag_document(uploaded_file)
+                                        
+                                        # Add each chunk to vector store
+                                        for chunk in chunks:
+                                            try:
+                                                vector_store.add_document(
+                                                    doc_id=f"{metadata['doc_id']}_chunk_{chunk['chunk_id']}",
+                                                    text=chunk['text'],
+                                                    metadata={
+                                                        'doc_id': metadata['doc_id'],
+                                                        'chunk_id': chunk['chunk_id'],
+                                                        'title': uploaded_file.name,
+                                                        'type': metadata.get('document_type', 'unknown')
+                                                    }
+                                                )
+                                            except Exception as e:
+                                                st.warning(f"Error adding chunk to vector store: {str(e)}")
+                                        
+                                        # Add document to case
+                                        doc_data = {
+                                            "id": metadata['doc_id'],
+                                            "title": uploaded_file.name,
+                                            "text": text,
+                                            "metadata": metadata,
+                                            "chunks": chunks
+                                        }
+                                        case_manager.add_document(case["id"], doc_data)
+                                        st.success(f"Document '{uploaded_file.name}' processed and added successfully!")
+                                        
+                                    except Exception as e:
+                                        st.error(f"Error processing document: {str(e)}")
             else:
                 st.info("No cases created yet. Use the form above to create your first case.")
         except Exception as e: