Fix storage initialization with better fallback handling

storage.py

@@ -1,8 +1,12 @@
 import os
 import shutil
-from azure.storage.blob import BlobServiceClient
 from abc import ABC, abstractmethod
 import json
+try:
+    from azure.storage.blob import BlobServiceClient
+    AZURE_AVAILABLE = True
+except ImportError:
+    AZURE_AVAILABLE = False
 
 class StorageInterface(ABC):
 
@@ -194,15 +198,29 @@ class BlobStorage(StorageInterface):
 class StorageFactory:
     @staticmethod
     def get_storage() -> StorageInterface:
-        storage_type = os.getenv('STORAGE_TYPE', 'local').lower()
+        storage_type = os.getenv('STORAGE_TYPE', 'local')
+        
+        # Always return LocalStorage if STORAGE_TYPE is 'local'
         if storage_type == 'local':
             return LocalStorage()
-        elif storage_type == 'blob':
+            
+        # Handle Azure storage with fallback
+        if storage_type in ['azure', 'blob']:
+            # If Azure SDK isn't available, fall back to local
+            if not AZURE_AVAILABLE:
+                print("Warning: Azure Storage SDK not available, falling back to local storage")
+                return LocalStorage()
+                
+            # Try to get Azure credentials
             connection_string = os.getenv('AZURE_STORAGE_CONNECTION_STRING')
             container_name = os.getenv('AZURE_STORAGE_CONTAINER_NAME')
+            
+            # If credentials are missing, fall back to local
             if not connection_string or not container_name:
-                raise ValueError("Azure Blob Storage connection string and container name must be set")
+                print("Warning: Azure credentials not found, falling back to local storage")
+                return LocalStorage()
+                
             return BlobStorage(connection_string, container_name)
-        else:
-            raise ValueError(f"Unsupported storage type: {storage_type}")
+            
+        raise ValueError(f"Unsupported storage type: {storage_type}")