Upload app.py with huggingface_hub

app.py

@@ -33,13 +33,21 @@ logger = logging.getLogger(__name__)
 # CONFIGURATION
 # ============================================================================
 
-MODELS_DIR = Path("./trained_models")
+# Get the directory where app.py is located
+APP_DIR = Path(__file__).parent.absolute()
+
+MODELS_DIR = APP_DIR / "trained_models"
 MODELS_DIR.mkdir(exist_ok=True)
 
-DATASETS_DIR = Path("./datasets")
+DATASETS_DIR = APP_DIR / "datasets"
 DATASETS_DIR.mkdir(exist_ok=True)
 
-NOTEBOOKS_DIR = Path("./notebooks")
+NOTEBOOKS_DIR = APP_DIR / "notebooks"
+
+# Log paths for debugging
+logger.info(f"APP_DIR: {APP_DIR}")
+logger.info(f"NOTEBOOKS_DIR: {NOTEBOOKS_DIR}")
+logger.info(f"NOTEBOOKS_DIR exists: {NOTEBOOKS_DIR.exists()}")
 
 # Model types available for training
 MODEL_TYPES = {
@@ -108,20 +116,22 @@ def execute_notebook(notebook_name: str, progress=gr.Progress()) -> Tuple[str, s
     output_path = NOTEBOOKS_DIR / f"output_{notebook_name}"
     
     if not notebook_path.exists():
-        return f"Error: Notebook not found: {notebook_name}", ""
+        # Debug: list what's actually in the directory
+        available = list(NOTEBOOKS_DIR.glob("*.ipynb")) if NOTEBOOKS_DIR.exists() else []
+        return f"Error: Notebook not found: {notebook_path}\nAvailable: {available}\nDir exists: {NOTEBOOKS_DIR.exists()}", ""
     
     progress(0.1, desc="Starting notebook execution...")
     
     try:
-        # Execute notebook using nbconvert
+        # Execute notebook using nbconvert with absolute paths
         cmd = [
             sys.executable, "-m", "nbconvert",
             "--to", "notebook",
             "--execute",
-            "--output", str(output_path.name),
+            "--output", str(output_path.absolute()),
             "--ExecutePreprocessor.timeout=600",
             "--ExecutePreprocessor.kernel_name=python3",
-            str(notebook_path)
+            str(notebook_path.absolute())
         ]
         
         progress(0.3, desc="Executing cells...")