added get_tasks to the class

model.py

@@ -210,40 +210,42 @@ class BertClassifier(LabelStudioMLBase):
             # Extract training data
             texts, labels = [], []
             
-            # Get annotations from Label Studio
             try:
                 # Get interface info
                 from_name, to_name, value = self.label_interface.get_first_tag_occurence('Choices', 'Text')
                 
                 # Get tasks from Label Studio
-                tasks = self.label_interface.get_tasks()
+                tasks = self.get_tasks()
                 logger.info(f"Found {len(tasks)} tasks")
                 
                 for task in tasks:
                     try:
                         # Get text from task
-                        text = task.get('data', {}).get(value)
+                        text = task['data'].get(value)
                         if not text:
+                            logger.warning(f"No text found in task {task.get('id')}")
                             continue
                             
                         # Get annotations
                         annotations = task.get('annotations', [])
                         if not annotations:
+                            logger.warning(f"No annotations found for task {task.get('id')}")
                             continue
                             
                         for annotation in annotations:
                             try:
                                 # Get choices from result
-                                result = annotation.get('result', [])
-                                if not result:
+                                results = annotation.get('result', [])
+                                if not results:
+                                    logger.warning(f"No results found in annotation for task {task.get('id')}")
                                     continue
                                     
-                                for r in result:
-                                    if r.get('from_name') == from_name and r.get('to_name') == to_name:
-                                        choices = r.get('value', {}).get('choices', [])
+                                for result in results:
+                                    if result.get('from_name') == from_name and result.get('to_name') == to_name:
+                                        choices = result.get('value', {}).get('choices', [])
                                         if choices:
                                             label = choices[0]
-                                            logger.info(f"Successfully extracted: Text='{text}', Label='{label}'")
+                                            logger.info(f"Successfully extracted: Text='{text[:50]}...', Label='{label}'")
                                             texts.append(text)
                                             labels.append(label)
                                             break
@@ -258,6 +260,7 @@ class BertClassifier(LabelStudioMLBase):
                         
             except Exception as e:
                 logger.error(f"Error getting tasks: {str(e)}")
+                logger.error("Full error details:", exc_info=True)
 
             logger.info(f"Prepared {len(texts)} examples for training")