src/utils.py

# utils.py

def get_all_models_from_database(db):
    models = []
    for tier, categories in db.items():
        for category, items in categories.items():
            for model in items:
                display_name = f"{model['name']} | {model['size']} | {category} | {tier}"
                models.append({
                    "display": display_name,
                    "name": model["name"],
                    "size": model["size"],
                    "description": model["description"],
                    "cost_a100": model.get("cost(A100)", "₹0"),
                    "cost_h100": model.get("cost(H100)", "₹0"),
                    "tier": tier,
                    "category": category
                })
    return models