fix: restore local execution compatibility and fix setup.py package mapping

setup.py

@@ -3,8 +3,8 @@ from setuptools import setup, find_packages
 setup(
     name="sap-rpt1",
     version="0.1.0",
-    package_dir={"": "code"},
-    packages=find_packages(where="code"),
+    package_dir={"": "webapp"},
+    packages=find_packages(where="webapp"),
     install_requires=[
         "numpy>=1.26.4",
         "pandas>=2.2.3",

webapp/benchmark.py

@@ -24,8 +24,11 @@ os.environ.setdefault("TABPFN_ACCEPT_TERMS",     "1")
 os.environ.setdefault("TABPFN_LICENSE_ACCEPTED", "1")
 os.environ.setdefault("AGREE_TABPFN_LICENSE",    "1")
 
-# Imports are handled via absolute package paths
-from webapp.models.tabpfn_wrapper import TabPFNWrapper
+# Flexible imports to support both HF Space (root) and local execution (inside webapp)
+try:
+    from webapp.models.tabpfn_wrapper import TabPFNWrapper
+except (ImportError, ModuleNotFoundError):
+    from models.tabpfn_wrapper import TabPFNWrapper
 
 N_FOLDS   = int(os.getenv("N_FOLDS",   "3"))
 RAND      = int(os.getenv("RANDOM_STATE", "42"))
@@ -437,9 +440,9 @@ def run_benchmark(df: pd.DataFrame, target_col: str) -> dict:
     dict with keys: dataset_info, task, results, ensemble_info, recommendation
     """
     try:
-        from ensemble import select_top_models, run_voting_ensemble, run_stacking_ensemble, SKLEARN_SAFE
-    except ImportError:
         from webapp.ensemble import select_top_models, run_voting_ensemble, run_stacking_ensemble, SKLEARN_SAFE
+    except (ImportError, ModuleNotFoundError):
+        from ensemble import select_top_models, run_voting_ensemble, run_stacking_ensemble, SKLEARN_SAFE
 
     y_raw = df[target_col].copy()
     X     = df.drop(columns=[target_col]).copy()

webapp/main.py

@@ -9,8 +9,11 @@ from fastapi import FastAPI, File, UploadFile, Form, HTTPException
 from fastapi.responses import JSONResponse, FileResponse
 from fastapi.staticfiles import StaticFiles
 
-# Absolute imports based on project root
-from webapp.benchmark import run_benchmark, infer_task, BUILDERS, _prep, _encode_target
+# Flexible imports to support both HF Space (root) and local execution (inside webapp)
+try:
+    from webapp.benchmark import run_benchmark, infer_task, BUILDERS, _prep, _encode_target
+except (ImportError, ModuleNotFoundError):
+    from benchmark import run_benchmark, infer_task, BUILDERS, _prep, _encode_target
 
 # Setup logging
 logging.basicConfig(level=logging.INFO)