import sys
import os

# Add the parent directory to sys.path to allow imports from 'src'
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))

from src.model_downloader import ModelDownloader

def download_all():
    print("Starting download of all models...")
    downloader = ModelDownloader()
    models = downloader.list_available_models()
    
    for model_name in models:
        try:
            print(f"Checking/Downloading {model_name}...")
            path = downloader.download_model(model_name)
            print(f"✓ {model_name} is ready at {path}")
        except Exception as e:
            print(f"✗ Failed to download {model_name}: {e}")

    print("\nAll downloads completed.")

if __name__ == "__main__":
    download_all()