# /// script
# requires-python = ">=3.10"
# dependencies = [
#     "torch>=2.0.0",
#     "diffusers>=0.25.0",
#     "transformers>=4.35.0",
#     "accelerate>=0.24.0",
#     "peft>=0.7.0",
#     "huggingface-hub>=0.20.0",
#     "safetensors>=0.4.0",
#     "Pillow>=10.0.0",
#     "numpy>=1.24.0",
#     "tqdm>=4.66.0",
# ]
# ///

import os
import sys
print("="*70)
print("FLUX LoRA Training v9 - Token Fix")
print("="*70)

# Get token - try multiple sources
token = os.environ.get("HF_TOKEN")
if not token or token == "$HF_TOKEN":
    # Try to read from file if env var not set properly
    try:
        with open("/secrets/HF_TOKEN") as f:
            token = f.read().strip()
    except:
        pass

if not token or token == "$HF_TOKEN":
    print("ERROR: HF_TOKEN not set correctly")
    print(f"  Value: {token}")
    print("  Env vars:", list(os.environ.keys()))
    sys.exit(1)

print(f"Token loaded: {token[:10]}...")

# Set token for huggingface_hub
os.environ["HF_TOKEN"] = token
os.environ["HUGGINGFACE_TOKEN"] = token

# Now import and login
from huggingface_hub import login, HfApi
login(token=token, add_to_git_credential=False)
print("✅ Login successful")

from diffusers import FluxPipeline
print("✅ FluxPipeline imported")

print("\n✅ SUCCESS: All imports work!")