Update utils/error_handling.py

utils/error_handling.py

@@ -1,84 +1,56 @@
-import functools
-import traceback
-import streamlit as st
-from utils.logging import log_error, get_logger
-
-class DataError(Exception):
-    """Custom exception for data-related errors"""
-    pass
-
-class ValidationError(Exception):
-    """Custom exception for validation errors"""
-    pass
-
-class ConfigError(Exception):
-    """Custom exception for configuration errors"""
-    pass
-
-def handle_data_exceptions(func):
-    """Decorator to handle data-related exceptions"""
-    @functools.wraps(func)
-    def wrapper(*args, **kwargs):
-        try:
-            return func(*args, **kwargs)
-        except (DataError, ValidationError) as e:
-            log_error(f"Data error in {func.__name__}", error=e)
-            st.error(f"Data processing error: {str(e)}")
-            return None
-        except Exception as e:
-            log_error(f"Unexpected error in {func.__name__}", error=e)
-            st.error(f"An unexpected error occurred: {str(e)}")
-            return None
-    return wrapper
-
-def handle_ui_exceptions(func):
-    """Decorator to handle UI-related exceptions"""
-    @functools.wraps(func)
-    def wrapper(*args, **kwargs):
-        try:
-            return func(*args, **kwargs)
-        except Exception as e:
-            log_error(f"UI error in {func.__name__}", error=e)
-            st.error(f"Interface error: {str(e)}")
-            return None
-    return wrapper
-
-def safe_execute(func, *args, **kwargs):
-    """Safely execute a function with error handling"""
-    try:
-        return func(*args, **kwargs)
-    except Exception as e:
-        log_error(f"Error executing {func.__name__}", error=e)
-        return None
+import logging
+import sys
+from datetime import datetime
+from pathlib import Path
+
+# Global logger configuration
+LOG_FORMAT = "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
+LOG_LEVEL = logging.INFO
+
+def setup_logging(log_level=LOG_LEVEL):
+    """Setup logging configuration"""
+    logging.basicConfig(
+        level=log_level,
+        format=LOG_FORMAT,
+        handlers=[
+            logging.StreamHandler(sys.stdout),
+        ]
+    )
+
+def get_logger(name):
+    """Get a logger instance"""
+    return logging.getLogger(name)
+
+def log_error(logger, error_msg, exception=None, context=None):
+    """Log error with additional context"""
+    full_msg = f"{error_msg}"
+    if context:
+        full_msg += f" | Context: {context}"
+    if exception:
+        full_msg += f" | Exception: {str(exception)}"
+    
+    logger.error(full_msg)
 
-def validate_input(data, required_fields=None):
-    """Validate input data"""
-    if not data:
-        raise ValidationError("Input data is empty or None")
+def log_info(logger, msg, context=None):
+    """Log info with additional context"""
+    full_msg = f"{msg}"
+    if context:
+        full_msg += f" | Context: {context}"
     
-    if required_fields:
-        missing_fields = [field for field in required_fields if field not in data]
-        if missing_fields:
-            raise ValidationError(f"Missing required fields: {', '.join(missing_fields)}")
+    logger.info(full_msg)
+
+def log_warning(logger, msg, context=None):
+    """Log warning with additional context"""
+    full_msg = f"{msg}"
+    if context:
+        full_msg += f" | Context: {context}"
     
-    return True
+    logger.warning(full_msg)
 
-def handle_file_operations(func):
-    """Decorator to handle file operation exceptions"""
-    @functools.wraps(func)
-    def wrapper(*args, **kwargs):
-        try:
-            return func(*args, **kwargs)
-        except FileNotFoundError as e:
-            log_error(f"File not found in {func.__name__}", error=e)
-            st.error("File not found. Please check the file path.")
-            return None
-        except PermissionError as e:
-            log_error(f"Permission error in {func.__name__}", error=e)
-            st.error("Permission denied. Please check file permissions.")
-            return None
-        except Exception as e:
-            log_error(f"File operation error in {func.__name__}", error=e)
-            st.error(f"File operation failed: {str(e)}")
-            return None
-    return wrapper
+def log_debug(logger, msg, context=None):
+    """Log debug with additional context"""
+    full_msg = f"{msg}"
+    if context:
+        full_msg += f" | Context: {context}"
+    
+    logger.debug(full_msg)