Update app.py

app.py

@@ -1,4 +1,4 @@
-# Essential imports
+# Core imports
 import os
 import logging
 import torch
@@ -13,6 +13,28 @@ from datetime import datetime
 from dataclasses import dataclass, field
 from dotenv import load_dotenv
 
+# Initialize logging
+logging.basicConfig(
+    level=logging.INFO,
+    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s'
+)
+logger = logging.getLogger(__name__)
+
+# Optional imports with error handling
+try:
+    import wandb
+    WANDB_AVAILABLE = True
+except ImportError:
+    logger.warning("wandb not installed, tracking disabled")
+    WANDB_AVAILABLE = False
+
+try:
+    from datasets import load_dataset
+    DATASETS_AVAILABLE = True
+except ImportError:
+    logger.warning("datasets not installed, demo dataset loading disabled")
+    DATASETS_AVAILABLE = False
+
 # Retrieve secrets securely from environment variables
 kaggle_username = os.getenv("KAGGLE_USERNAME")
 kaggle_key = os.getenv("KAGGLE_KEY")