| import pandas as pd |
| from pathlib import Path |
|
|
| |
| BASE_DIR = Path(__file__).resolve().parent |
|
|
| HOSPITALS = { |
| "Mayo Clinic": BASE_DIR / "hospitals" / "mayo_clinic_data.csv", |
| "Cleveland Clinic": BASE_DIR / "hospitals" / "cleveland_clinic_data.csv", |
| "Johns Hopkins Hospital": BASE_DIR / "hospitals" / "johns_hopkins_hospital_data.csv", |
| "Massachusetts General Hospital": BASE_DIR / "hospitals" / "massachusetts_general_hospital_data.csv", |
| "UCLA Medical Center": BASE_DIR / "hospitals" / "ucla_medical_center_data.csv", |
| "Cedars-Sinai Medical Center": BASE_DIR / "hospitals" / "cedars_sinai_medical_center_data.csv", |
| "NewYork-Presbyterian Hospital": BASE_DIR / "hospitals" / "newyork_presbyterian_hospital_data.csv", |
| "Northwestern Memorial Hospital": BASE_DIR / "hospitals" / "northwestern_memorial_hospital_data.csv", |
| "UCSF Medical Center": BASE_DIR / "hospitals" / "ucsf_medical_center_data.csv", |
| "Houston Methodist Hospital": BASE_DIR / "hospitals" / "houston_methodist_hospital_data.csv", |
| } |
|
|
| def load_services_data(path) -> pd.DataFrame: |
| """Load price data from a CSV file.""" |
| df = pd.read_csv(path) |
| return df[[ |
| "intent", |
| "description", |
| "gross_charge", |
| "negotiated_rate", |
| ]] |
|
|
| def list_services(services_data: pd.DataFrame) -> list[str]: |
| return services_data["description"].tolist() |
|
|
