data_registry.py

# data_registry.py
import os
import pandas as pd
from typing import Dict, Optional

class DataRegistry:
    def __init__(self):
        self._frames: Dict[str, pd.DataFrame] = {}

    def add_path(self, path: str) -> bool:
        name = os.path.basename(path)
        ext = os.path.splitext(name)[1].lower()
        if ext == ".csv":
            df = pd.read_csv(path)
            self._frames[name] = df
            return True
        return False

    def names(self):
        return list(self._frames.keys())

    def get(self, name_or_path: str) -> Optional[pd.DataFrame]:
        key = os.path.basename(name_or_path)
        return self._frames.get(key)

    def put_df(self, key: str, df):
        self._frames[key] = df