import os
from dotenv import load_dotenv

def set_environment(verbose=True) -> dict:
    env_path = os.path.abspath(".env")

    # Load .env only if it exists
    if os.path.exists(env_path):
        load_dotenv(dotenv_path=env_path, override=True)
        if verbose:
            print(f"✅ Loaded .env from: {env_path}")
    else:
        if verbose:
            print("⚠️ No .env file found. Relying on system environment variables.")

    required_keys = [
        "OPENAI_API_KEY",
        "OPENROUTER_API_KEY",
        "GEMINI_API_KEY",
        "GROQ_API_KEY"
    ]
    missing_keys = [key for key in required_keys if not os.getenv(key)]

    if verbose:
        if missing_keys:
            print(f"⚠️ Missing environment variables: {missing_keys}")
        else:
            print("✅ All required API keys loaded successfully!")

    return {key: os.getenv(key) for key in required_keys}