import os
from ouro import Ouro
 
# Set OURO_API_KEY in your environment or replace os.environ.get("OURO_API_KEY")
ouro = Ouro(api_key=os.environ.get("OURO_API_KEY"))
 
# Option 1: Retrieve by route ID
route_id = "25945f7c-d83a-45ba-8c40-30b69d9a3df7"
route = ouro.routes.retrieve(route_id)
 
# Option 2: Retrieve by route identifier (username/route-name)
route_identifier = "mmoderwell/explore-a-chemical-system-with-ggen"
route = ouro.routes.retrieve(route_identifier)
 
print(route.name, route.visibility)
print(route.metadata)

# Retrieve the route
route = ouro.routes.retrieve("mmoderwell/explore-a-chemical-system-with-ggen")
 
# Use the route with a request body
response = route.use(
    body={
        'system': 'example_string',
        'max_atoms': 20,
        'max_steps': 400,
        'min_atoms': 2,
        'num_trials': 15,
        'e_above_hull': 0.15,
        'skip_existing': False,
        'relax_optimizer': 'fire',
        'require_all_elements': False,
        'include_phase_diagram': True
    }
)
print(response)

# Retrieve the route
route = ouro.routes.retrieve("mmoderwell/explore-a-chemical-system-with-ggen")
 
# Read all actions (request/response history) for this route
actions = route.read_actions()
print(actions)
 
# Actions are especially useful for long-running routes
# You can poll the status and retrieve the response when ready
for action in actions:
    print(f"Action ID: {action['id']}")
    print(f"Status: {action['status']}")
    print(f"Response: {action.get('response_data')}")