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"))
 
dataset_id = "019dee79-fc81-7e16-8675-148618943166"
 
# Retrieve dataset metadata
dataset = ouro.datasets.retrieve(dataset_id)
print(dataset.name, dataset.visibility)
print(dataset.metadata)

# Get column definitions for the underlying table
columns = ouro.datasets.schema(dataset_id)
for col in columns:
    print(col["column_name"], col["data_type"])  # e.g., age integer, name text

# Option 1: Query data by dataset ID (returns Pandas DataFrame)
df = ouro.datasets.query(dataset_id)
print(df.head())
 
# Option 2: Load data by table name (faster for large datasets)
table_name = dataset.metadata["table_name"]  # e.g., "dft_vs_mlip_permanent_magnet_benchmark_specification"
df = ouro.datasets.load(table_name)
print(len(df))

import pandas as pd
 
# Update dataset metadata
updated = ouro.datasets.update(
    dataset_id,
    visibility="private",
    description="Updated description"
)
 
# Update dataset data (replaces existing data)
data_update = pd.DataFrame([
    {"name": "Charlie", "age": 33},
    {"name": "Diana", "age": 28},
])
updated = ouro.datasets.update(dataset_id, data=data_update)