Create utils/case_manager.py

utils/case_manager.py

@@ -0,0 +1,70 @@
+# utils/case_manager.py
+import os
+import json
+from datetime import datetime
+from typing import Dict, List, Optional
+import streamlit as st
+
+class CaseManager:
+    def __init__(self, base_path: str = "data/cases"):
+        self.base_path = base_path
+        os.makedirs(base_path, exist_ok=True)
+        self._load_cases()
+
+    def _load_cases(self):
+        """Load existing cases from storage"""
+        if 'cases' not in st.session_state:
+            st.session_state.cases = {}
+            for case_id in os.listdir(self.base_path):
+                case_path = os.path.join(self.base_path, case_id)
+                if os.path.isdir(case_path):
+                    with open(os.path.join(case_path, 'metadata.json'), 'r') as f:
+                        st.session_state.cases[case_id] = json.load(f)
+
+    def create_case(self, title: str, description: str, case_type: str) -> str:
+        """Create a new case"""
+        case_id = datetime.now().strftime('%Y%m%d_%H%M%S')
+        case_path = os.path.join(self.base_path, case_id)
+        os.makedirs(case_path, exist_ok=True)
+        os.makedirs(os.path.join(case_path, 'documents'), exist_ok=True)
+        
+        case_data = {
+            'id': case_id,
+            'title': title,
+            'description': description,
+            'case_type': case_type,
+            'created_at': datetime.now().isoformat(),
+            'documents': []
+        }
+        
+        with open(os.path.join(case_path, 'metadata.json'), 'w') as f:
+            json.dump(case_data, f)
+            
+        st.session_state.cases[case_id] = case_data
+        return case_id
+
+    def add_document(self, case_id: str, document_data: Dict) -> str:
+        """Add document to case"""
+        case = st.session_state.cases.get(case_id)
+        if not case:
+            raise ValueError(f"Case {case_id} not found")
+            
+        doc_id = datetime.now().strftime('%Y%m%d_%H%M%S')
+        document_data['id'] = doc_id
+        document_data['added_at'] = datetime.now().isoformat()
+        
+        case['documents'].append(document_data)
+        
+        case_path = os.path.join(self.base_path, case_id)
+        with open(os.path.join(case_path, 'metadata.json'), 'w') as f:
+            json.dump(case, f)
+            
+        return doc_id
+
+    def get_case(self, case_id: str) -> Optional[Dict]:
+        """Get case details"""
+        return st.session_state.cases.get(case_id)
+
+    def get_all_cases(self) -> List[Dict]:
+        """Get all cases"""
+        return list(st.session_state.cases.values())