Add new files

api_mental_health.py

@@ -37,10 +37,16 @@ logger = logging.getLogger(__name__)
 # Load environment variables
 load_dotenv()
 
-# Download required NLTK data
-nltk.download('punkt')
-nltk.download('wordnet')
-nltk.download('stopwords')
+# Set NLTK data path to a directory where we have write permissions
+nltk_data_path = os.path.join(os.path.dirname(__file__), "nltk_data")
+os.makedirs(nltk_data_path, exist_ok=True)
+nltk.data.path.append(nltk_data_path)
+
+# Download required NLTK data to our custom directory
+logger.info(f"Downloading NLTK data to {nltk_data_path}")
+nltk.download('punkt', download_dir=nltk_data_path)
+nltk.download('wordnet', download_dir=nltk_data_path)
+nltk.download('stopwords', download_dir=nltk_data_path)
 
 # Initialize FastAPI app
 app = FastAPI(title="Mental Health Counselor API")

run_api.sh

@@ -0,0 +1,21 @@
+#!/bin/bash
+
+# This script handles startup for the API on Hugging Face Spaces
+echo "===== Application Startup at $(date -u '+%Y-%m-%d %H:%M:%S') ====="
+
+# Make sure our custom directories are properly readable/writable
+echo "Setting up directories with proper permissions..."
+mkdir -p data/users data/sessions data/conversations data/feedback
+mkdir -p mental_health_model_artifacts/chroma_db
+mkdir -p nltk_data
+chmod -R 777 data nltk_data mental_health_model_artifacts
+
+# Create .env file if it doesn't exist
+if [ ! -f .env ]; then
+    echo "Creating .env file from .env.example..."
+    cp .env.example .env
+fi
+
+# Start the API with uvicorn
+echo "Starting API server..."
+exec uvicorn app:app --host 0.0.0.0 --port 7860 

setup_hf_deployment.sh

@@ -1,60 +0,0 @@
-#!/bin/bash
-set -e
-
-echo "Setting up deployment to Hugging Face Spaces..."
-
-# Check if git is installed
-if ! command -v git &> /dev/null; then
-    echo "Error: git is not installed. Please install git first."
-    exit 1
-fi
-
-# Check if git-lfs is installed
-if ! command -v git-lfs &> /dev/null; then
-    echo "Error: git-lfs is not installed. Please install git-lfs first."
-    echo "You can install it with: brew install git-lfs (Mac) or sudo apt-get install git-lfs (Linux)"
-    exit 1
-fi
-
-# Initialize Git LFS
-echo "Initializing Git LFS..."
-git lfs install
-
-# Create a new orphaned branch
-echo "Creating new orphaned branch 'hf-deployment'..."
-git checkout --orphan hf-deployment
-
-# Remove all files from staging
-echo "Clearing staging area..."
-git rm -rf --cached .
-
-# Keep the .gitattributes and .gitignore files
-echo "Adding .gitattributes and .gitignore..."
-git add .gitattributes .gitignore
-
-# Make sure the LFS tracking is set up correctly
-echo "Setting up LFS tracking for large files..."
-git lfs track "*.sqlite3"
-git lfs track "*.pkl"
-git lfs track "*.bin"
-git lfs track "mental_health_model_artifacts/chroma_db/chroma.sqlite3"
-git lfs track "mental_health_model_artifacts/chroma_db/**/*.bin"
-git lfs track "mental_health_model_artifacts/**/*.pkl"
-
-# Add all files
-echo "Adding all files to staging..."
-git add .
-
-# Commit the changes
-echo "Committing changes..."
-git commit -m "Initial deployment with Git LFS enabled"
-
-echo "======================================================"
-echo "Setup complete! To push to Hugging Face Spaces, run:"
-echo "git push -u origin hf-deployment --force"
-echo ""
-echo "If this is successful and you want to make this the main branch:"
-echo "1. Go to Hugging Face Spaces and set 'hf-deployment' as the default branch"
-echo "2. Rename the branch locally: git branch -m hf-deployment main"
-echo "3. Push to origin: git push -u origin main"
-echo "======================================================"