Improve cohesion and consistency in the codebase

requirements.txt

@@ -72,7 +72,3 @@ language-tool-python>=2.7.1  # For grammar checking
 
 # Domain-agnostic processing
 textacy>=0.12.0
-
-# Add these lines to requirements.txt
-beautifulsoup4>=4.9.3
-lxml>=4.9.0  # Optional but recommended parser for bs4

src/cluster_manager.py

@@ -6,6 +6,7 @@ class ClusterManager:
     def __init__(self, config):
         self.config = config
         self.clusterer = None
+        self.method = self.config['clustering']['method']
 
     def perform_clustering(self, embeddings):
         """Perform clustering on embeddings and return labels."""

src/data_loader.py

@@ -8,6 +8,14 @@
 from sentence_transformers import SentenceTransformer
 
 class DataLoader:
+    """
+    DataLoader class to handle loading and processing of datasets.
+
+    Attributes:
+        config (Dict[str, Any]): Configuration dictionary for data loading.
+        logger (logging.Logger): Logger for logging information and errors.
+        batch_size (int): Batch size for data loading.
+    """
     def __init__(self, config: Dict[str, Any]):
         """Initialize DataLoader with configuration"""
         self.config = config
@@ -20,7 +28,12 @@ def __init__(self, config: Dict[str, Any]):
             self.config['data']['scisummnet_path'] = str(project_root / self.config['data']['scisummnet_path'])
 
     def load_all_datasets(self) -> Dict[str, pd.DataFrame]:
-        """Load all configured datasets."""
+        """
+        Load all configured datasets.
+
+        Returns:
+            Dict[str, pd.DataFrame]: Dictionary of loaded datasets.
+        """
         datasets = {}
 
         # Load XL-Sum dataset if enabled
@@ -70,7 +83,15 @@ def load_all_datasets(self) -> Dict[str, pd.DataFrame]:
         return datasets
 
     def load_scisummnet(self, path: str) -> Optional[pd.DataFrame]:
-        """Load ScisummNet dataset from local directory"""
+        """
+        Load ScisummNet dataset from local directory.
+
+        Args:
+            path (str): Path to the ScisummNet dataset.
+
+        Returns:
+            Optional[pd.DataFrame]: DataFrame containing the loaded data.
+        """
         try:
             self.logger.info(f"Loading ScisummNet dataset from {path}...")
             data = []
@@ -130,7 +151,12 @@ def load_scisummnet(self, path: str) -> Optional[pd.DataFrame]:
             return None
 
     def load_scisummnet_dataset(self) -> pd.DataFrame:
-        """Load ScisummNet dataset."""
+        """
+        Load ScisummNet dataset.
+
+        Returns:
+            pd.DataFrame: DataFrame containing the loaded data.
+        """
         try:
             self.logger.info(f"Loading ScisummNet dataset from {self.config['data']['scisummnet_path']}...")
 
@@ -180,4 +206,4 @@ def load_scisummnet_dataset(self) -> pd.DataFrame:
 
         except Exception as e:
             self.logger.error(f"Error loading ScisummNet dataset: {e}")
-            return None
+            return None

src/main.py

@@ -40,7 +40,6 @@
 
 import torch
 import multiprocessing
-from concurrent.futures import ThreadPoolExecutor, ProcessPoolExecutor
 from utils.style_selector import determine_cluster_style, get_style_parameters
 from utils.metrics_utils import calculate_cluster_variance, calculate_lexical_diversity, calculate_cluster_metrics
 from datasets import load_dataset
@@ -203,6 +202,9 @@ def get_optimal_batch_size():
     return 16  # Default for CPU
 
 def main():
+    """
+    Main function to run the entire pipeline.
+    """
     # Initialize logger first
     logger = logging.getLogger(__name__)
 

src/main_with_training.py

@@ -180,4 +180,4 @@ def generate_summaries(cluster_texts: Dict[str, List[str]], config: Dict) -> Lis
     return summaries
 
 if __name__ == "__main__":
-    main() 
+    main()