Update describepdf/config.py

describepdf/config.py

@@ -14,9 +14,41 @@ import pathlib
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - [%(module)s] - %(message)s')
 logger = logging.getLogger('describepdf')
 
+def _resolve_prompts_directory() -> pathlib.Path:
+"""
+Resolve the path to the prompts directory with multiple fallback strategies.
+Returns:
+    pathlib.Path: Path to the prompts directory
+"""
+# List of potential paths to check
+potential_paths = [
+    # Current file's parent directory
+    pathlib.Path(__file__).parent.parent / "prompts",
+    
+    # Relative to the current working directory
+    pathlib.Path.cwd() / "prompts",
+    
+    # Absolute path fallback (useful in deployment)
+    pathlib.Path("/app/prompts"),
+    pathlib.Path("/workspace/prompts"),
+    
+    # Hugging Face Spaces specific path
+    pathlib.Path("/home/user/app/prompts")
+]
+
+# Try each path
+for path in potential_paths:
+    if path.is_dir():
+        logger.info(f"Prompts directory found at: {path}")
+        return path
+
+# If no path is found
+logger.error("Could not locate prompts directory. Using a temporary fallback.")
+return pathlib.Path(__file__).parent / "prompts"
+
 # Directory containing prompt templates (making path absolute by using current file location)
 SCRIPT_DIR = pathlib.Path(__file__).parent.parent.absolute()
-PROMPTS_DIR = "prompts"
+PROMPTS_DIR = _resolve_prompts_directory()
 
 # Default configuration values
 DEFAULT_CONFIG: Dict[str, Any] = {