from huggingface_hub import InferenceClient
from dotenv import load_dotenv
import os

load_dotenv()
token = os.getenv("HF_API_TOKEN")

prompt = "A roman mercenary"

candidates = [
    "stabilityai/stable-diffusion-2-1",
    "runwayml/stable-diffusion-v1-5",
    "prompthero/openjourney",
    "CompVis/stable-diffusion-v1-4",
    "Segmind/SSD-1B"
]

print("--- Searching for Free Models ---")
for model in candidates:
    print(f"\nTesting: {model}...")
    client = InferenceClient(model=model, token=token)
    try:
        # Standard conservative params
        image = client.text_to_image(prompt, guidance_scale=7.5) 
        fname = f"test_{model.replace('/', '_')}.png"
        image.save(fname)
        print(f"✅ SUCCESS! {model} is FREE and working.")
        # Stop after first success to save time/credits? 
        # No, let's find the best one.
    except Exception as e:
        print(f"❌ FAILED: {model} - {e}")