refactor and add tests, v0.2.0

src/embeddingbuddy/ui/callbacks/data_processing.py

@@ -0,0 +1,61 @@
+import numpy as np
+from dash import callback, Input, Output, State
+from ...data.processor import DataProcessor
+
+
+class DataProcessingCallbacks:
+    
+    def __init__(self):
+        self.processor = DataProcessor()
+        self._register_callbacks()
+    
+    def _register_callbacks(self):
+        
+        @callback(
+            Output('processed-data', 'data'),
+            Input('upload-data', 'contents'),
+            State('upload-data', 'filename')
+        )
+        def process_uploaded_file(contents, filename):
+            if contents is None:
+                return None
+            
+            processed_data = self.processor.process_upload(contents, filename)
+            
+            if processed_data.error:
+                return {'error': processed_data.error}
+            
+            return {
+                'documents': [self._document_to_dict(doc) for doc in processed_data.documents],
+                'embeddings': processed_data.embeddings.tolist()
+            }
+        
+        @callback(
+            Output('processed-prompts', 'data'),
+            Input('upload-prompts', 'contents'),
+            State('upload-prompts', 'filename')
+        )
+        def process_uploaded_prompts(contents, filename):
+            if contents is None:
+                return None
+            
+            processed_data = self.processor.process_upload(contents, filename)
+            
+            if processed_data.error:
+                return {'error': processed_data.error}
+            
+            return {
+                'prompts': [self._document_to_dict(doc) for doc in processed_data.documents],
+                'embeddings': processed_data.embeddings.tolist()
+            }
+    
+    @staticmethod
+    def _document_to_dict(doc):
+        return {
+            'id': doc.id,
+            'text': doc.text,
+            'embedding': doc.embedding,
+            'category': doc.category,
+            'subcategory': doc.subcategory,
+            'tags': doc.tags
+        }