Prepare for Hugging Face Spaces deployment with simplified configuration

.env.example

@@ -1,4 +1,10 @@
-OPENAI_API_KEY=
-LANGCHAIN_API_KEY=
-LANGCHAIN_TRACING_V2='true'
-LANGCHAIN_PROJECT=
+# API Keys
+OPENAI_API_KEY=your-openai-api-key
+LANGCHAIN_API_KEY=your-langchain-api-key
+
+# Configuration
+LANGCHAIN_TRACING_V2=true
+LANGCHAIN_PROJECT=MentalHealthCounselorPOC
+
+# Deployment
+HF_SPACES=true

Dockerfile

@@ -1,13 +1,15 @@
-FROM python:3.13-slim
-
-RUN useradd -m -u 1000 user
-USER user
-ENV PATH="/home/user/.local/bin:$PATH"
+FROM python:3.9-slim
 
 WORKDIR /app
 
+# Install git and git-lfs for downloading large files (if needed)
+RUN apt-get update && \
+    apt-get install -y git git-lfs build-essential && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
 # Copy requirements file
-COPY --chown=user requirements.txt .
+COPY requirements.txt .
 
 # Install dependencies
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
@@ -17,9 +19,12 @@ RUN python -c "import nltk; nltk.download('punkt'); nltk.download('wordnet'); nl
 
 # Create necessary directories
 RUN mkdir -p data/users data/sessions data/conversations data/feedback
+RUN mkdir -p mental_health_model_artifacts/chroma_db
 
 # Copy application files
-COPY --chown=user . /app
+COPY app.py .
+COPY .env.example .env
+COPY api_mental_health.py .
 
 # Expose the port Hugging Face Spaces expects
 EXPOSE 7860

README.md

@@ -1,21 +1,29 @@
 ---
 title: Mental Health Counselor API
 emoji: 🧠
-colorFrom: purple
-colorTo: indigo
+colorFrom: indigo
+colorTo: purple
 sdk: docker
+app_port: 7860
 pinned: false
 ---
 
 # Mental Health Counselor API
 
-This is a backend API for a mental health counseling application. It provides endpoints for analyzing patient messages, suggesting counselor responses, and managing counseling sessions.
+This is the backend API for a mental health counseling application. It provides endpoints for analyzing patient messages, suggesting counselor responses, and managing counseling sessions.
 
 ## API Endpoints
 
-- `/`: Root endpoint showing API status
-- `/health`: Health check endpoint
-- And many more specialized endpoints from the full API
+- `/` - Root endpoint showing API status
+- `/health` - Health check endpoint
+- `/metadata` - API metadata and information
+
+## Technology Stack
+
+- FastAPI framework
+- Scikit-Learn based ML models
+- NLTK for NLP processing
+- Vector database for semantic search
 
 ## Deployment
 
@@ -24,3 +32,10 @@ This API is deployed on Hugging Face Spaces using Docker.
 ## Frontend
 
 The frontend for this application is deployed separately on Vercel.
+
+## Repository Structure
+
+- `app.py` - Main application entry point
+- `api_mental_health.py` - Core API functionality
+- `data/` - Data storage directories
+- `mental_health_model_artifacts/` - ML model files

app.py

@@ -1,42 +1,78 @@
-from fastapi import FastAPI
+from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel
 import os
+from dotenv import load_dotenv
 import logging
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
+# Load environment variables
+load_dotenv()
+
 # Initialize FastAPI app
 app = FastAPI(title="Mental Health Counselor API")
 
-# Create necessary directories
-os.makedirs("data/users", exist_ok=True)
-os.makedirs("data/sessions", exist_ok=True)
-os.makedirs("data/conversations", exist_ok=True)
-os.makedirs("data/feedback", exist_ok=True)
+# Initialize global storage
+DATA_DIR = os.path.join(os.path.dirname(__file__), "data")
+os.makedirs(DATA_DIR, exist_ok=True)
+os.makedirs(os.path.join(DATA_DIR, "users"), exist_ok=True)
+os.makedirs(os.path.join(DATA_DIR, "sessions"), exist_ok=True)
+os.makedirs(os.path.join(DATA_DIR, "conversations"), exist_ok=True)
+os.makedirs(os.path.join(DATA_DIR, "feedback"), exist_ok=True)
+
+# Simple health check route
+@app.get("/")
+async def root():
+    return {
+        "status": "ok",
+        "message": "Mental Health Counselor API is running",
+        "api_version": "1.0.0",
+        "backend_info": "FastAPI on Hugging Face Spaces"
+    }
 
-# Define a simple health check route
+# Health check endpoint
 @app.get("/health")
 async def health_check():
     return {"status": "ok", "message": "Mental Health Counselor API is running"}
 
-# Define a simple root route
-@app.get("/")
-async def root():
+# Metadata endpoint
+@app.get("/metadata")
+async def get_metadata():
     return {
-        "app": "Mental Health Counselor API",
-        "status": "running",
+        "api_version": "1.0.0",
         "endpoints": [
+            "/",
             "/health",
-            "/api-docs"
-        ]
+            "/metadata"
+        ],
+        "provider": "Mental Health Counselor API on Hugging Face Spaces",
+        "deployment_type": "Hugging Face Spaces Docker",
+        "description": "This API provides functionality for a mental health counseling application.",
+        "frontend": "Deployed separately on Vercel"
     }
 
-# Import the actual API if the file exists
+# Try to import the full API if available
 try:
-    from api_mental_health import app as full_app
-    # Merge the routes from the full app
-    app.routes.extend(full_app.routes)
-    logger.info("Loaded full API functionality")
-except ImportError:
-    logger.warning("Could not import full API functionality")
+    import api_mental_health
+    # If the import succeeds, try to add those routes
+    logger.info("Successfully imported full API module")
+    
+    # Add a placeholder for full functionality
+    @app.get("/full-api-status")
+    async def full_api_status():
+        return {
+            "status": "imported",
+            "message": "Full API module was imported successfully, but endpoints may require additional setup"
+        }
+except ImportError as e:
+    logger.warning(f"Could not import full API module: {e}")
+    
+    @app.get("/full-api-status")
+    async def full_api_status():
+        return {
+            "status": "unavailable",
+            "message": "Full API module could not be imported",
+            "error": str(e)
+        }

requirements.txt

@@ -1,12 +1,12 @@
 # Core dependencies for data processing and ML
-pandas
-numpy
-scikit-learn  # Used for ML models
-joblib
+pandas==2.0.3
+numpy==1.24.3
+scikit-learn==1.2.2
+joblib==1.3.1
 
 # NLP and sentiment analysis
-nltk
-vaderSentiment
+nltk==3.8.1
+vaderSentiment==3.3.2
 
 # Dataset downloading
 kagglehub
@@ -17,21 +17,21 @@ openai
 langchain
 langchain-openai
 langchain-chroma
-httpx  # Required for API calls
+httpx==0.24.1
 
 # API and tracing
-fastapi
-uvicorn
-pydantic
+fastapi==0.95.2
+uvicorn[standard]==0.22.0
+pydantic==1.10.8
 langsmith
-python-dotenv
+python-dotenv==1.0.0
 lightgbm
 
 # New dependencies for additional features
-python-multipart  # For file uploads
+python-multipart==0.0.6
 fastapi-cors  # For CORS support
-aiofiles  # For async file operations
-jinja2  # For template rendering
+aiofiles==23.1.0
+jinja2==3.1.2
 python-jose[cryptography]  # For JWT tokens (authentication)
 passlib[bcrypt]  # For password hashing
 pydub  # For audio processing