from gradio_client import Client
import json

print("Connecting to harvesthealth/tiny_factory...")
client = Client("harvesthealth/tiny_factory")

print("Calling /generate_personas...")
try:
    result = client.predict(
        business_description="A revolutionary health tech startup focusing on personalized nutrition.",
        customer_profile="Health-conscious professionals aged 25-45 living in urban areas.",
        num_personas=1,
        blablador_api_key=None,
        api_name="/generate_personas"
    )
    print("API call successful!")
    print(json.dumps(result, indent=2))
except Exception as e:
    print(f"API call failed: {e}")