medextract / app /models.py
harsh-dev's picture
Add
ec563fd
raw
history blame contribute delete
510 Bytes
import pandas as pd
from app.config import settings
df = None
icd_lookup = {}
def load_icd_dataset():
global df, icd_lookup
try:
df = pd.read_csv(settings.ICD_DATA_PATH) # CSV with TestName, ICDCode
icd_lookup = dict(zip(df["TestName"], df["ICDCode"]))
print("✅ Loaded ICD dataset")
except FileNotFoundError:
df = None
icd_lookup = {}
print("⚠️ No ICD dataset found. Please provide ICD CSV")
# Load immediately on import
load_icd_dataset()