Update data_registry.py

data_registry.py

@@ -1,17 +1,27 @@
 # data_registry.py
+import os
 import pandas as pd
+from typing import Dict, Optional
 
 class DataRegistry:
     def __init__(self):
-        self._data={}
+        self._frames: Dict[str, pd.DataFrame] = {}
 
-    def add_path(self, path: str):
-        if path.endswith(".csv"):
-            self._data[path]=pd.read_csv(path)
-        # future: add PDF/TXT/MD parsing
-
-    def get(self, name: str):
-        return self._data.get(name)
+    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._data.keys())
+        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