Update app.py

app.py

@@ -169,17 +169,36 @@ def main():
                             f"Upload Document to Case: {case['title']}", 
                             key=f"upload_{case['id']}"
                         )
+                        # 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
+                                        "metadata": metadata,
+                                        "chunks": chunks
                                     }
                                     case_manager.add_document(case["id"], doc_data)
-                                    st.success(f"Document '{uploaded_file.name}' added successfully!")
+                                    st.success(f"Document '{uploaded_file.name}' processed and added successfully!")
                                 except Exception as e:
                                     st.error(f"Error processing document: {str(e)}")
             else: