Update utils/case_manager.py

utils/case_manager.py

@@ -5,6 +5,7 @@ from typing import List, Dict, Optional
 from pathlib import Path
 import shutil
 
+
 class CaseManager:
     def __init__(self, base_path: str = "data/cases"):
         """Initialize CaseManager with a base directory to store cases."""
@@ -31,11 +32,11 @@ class CaseManager:
         """Create a new case and save it to storage."""
         case_id = datetime.now().strftime('%Y%m%d_%H%M%S')
         case_path = self.base_path / case_id
-        
+
         # Create case directory structure
         case_path.mkdir(exist_ok=True)
         (case_path / 'documents').mkdir(exist_ok=True)
-        
+
         # Prepare case data
         case_data = {
             'id': case_id,
@@ -47,11 +48,11 @@ class CaseManager:
             'status': 'active',
             'documents': []
         }
-        
+
         # Save case metadata
         with open(case_path / 'metadata.json', 'w') as f:
             json.dump(case_data, f, indent=2)
-        
+
         self.cases[case_id] = case_data
         return case_id
 
@@ -64,23 +65,22 @@ class CaseManager:
         return self.cases.get(case_id)
 
     def add_document(self, case_id: str, document_data: Dict):
-    """Add a document to a case."""
-    if case_id not in self.cases:
-        raise ValueError(f"Case with ID {case_id} not found.")
-
-    # Update document metadata
-    document_data['id'] = document_data.get('id', datetime.now().strftime('%Y%m%d_%H%M%S'))
-    document_data['added_at'] = document_data.get('added_at', datetime.now().isoformat())
+        """Add a document to a case."""
+        if case_id not in self.cases:
+            raise ValueError(f"Case with ID {case_id} not found.")
 
-    # Add document to case
-    case = self.cases[case_id]
-    case['documents'].append(document_data)
-    case['updated_at'] = datetime.now().isoformat()
+        # Update document metadata
+        document_data['id'] = document_data.get('id', datetime.now().strftime('%Y%m%d_%H%M%S'))
+        document_data['added_at'] = document_data.get('added_at', datetime.now().isoformat())
 
-    # Save updated case metadata
-    with open(self.base_path / case_id / 'metadata.json', 'w') as f:
-        json.dump(case, f, indent=2)
+        # Add document to case
+        case = self.cases[case_id]
+        case['documents'].append(document_data)
+        case['updated_at'] = datetime.now().isoformat()
 
+        # Save updated case metadata
+        with open(self.base_path / case_id / 'metadata.json', 'w') as f:
+            json.dump(case, f, indent=2)
 
     def list_documents(self, case_id: str) -> List[Dict]:
         """List all documents in a case."""
@@ -92,21 +92,21 @@ class CaseManager:
         """Remove a document from a case."""
         if case_id not in self.cases:
             return False
-        
+
         case = self.cases[case_id]
         document = next((doc for doc in case['documents'] if doc['id'] == document_id), None)
-        
+
         if not document:
             return False
-        
+
         # Remove from case metadata
         case['documents'] = [doc for doc in case['documents'] if doc['id'] != document_id]
         case['updated_at'] = datetime.now().isoformat()
-        
+
         # Save updated case metadata
         with open(self.base_path / case_id / 'metadata.json', 'w') as f:
             json.dump(case, f, indent=2)
-        
+
         # Remove the physical file
         try:
             doc_path = Path(document['file_path'])