chatbot / entrypoint.sh
Moderator404's picture
Create entrypoint.sh
0764160 verified
Raw
History Blame Contribute Delete
1.42 kB
#!/bin/bash
set -e
echo "Starting TMC Chatbot on Hugging Face Spaces..."
# Disable telemetry
export ANONYMIZED_TELEMETRY=False
export CHROMA_TELEMETRY_ENABLED=false
export CHROMADB_TELEMETRY_DISABLED=true
export POSTHOG_DISABLED=true
# Create necessary directories
mkdir -p /app/data/transformers_cache
mkdir -p /app/data/huggingface_cache
mkdir -p /app/data/vector_db
mkdir -p /app/data/sentence_transformers_cache
# Set database path
export DATABASE_PATH=${DATABASE_PATH:-/app/data/tmc_customer_service.db}
# Initialize database if needed
if [ ! -f "$DATABASE_PATH" ]; then
echo "Database not found. Creating tables and default users..."
python -c "
from scripts.init_database import init_sample_data
init_sample_data()
print('Database initialized with default users.')
"
else
# Check if users exist
USER_COUNT=$(python -c "import sqlite3; conn = sqlite3.connect('$DATABASE_PATH'); cur = conn.cursor(); cur.execute('SELECT COUNT(*) FROM users'); print(cur.fetchone()[0]); conn.close()" 2>/dev/null || echo "0")
if [ "$USER_COUNT" = "0" ]; then
echo "Database exists but has no users. Adding default users..."
python -c "
from scripts.init_database import init_sample_data
init_sample_data()
print('Default users added.')
"
else
echo "Database already initialized with $USER_COUNT user(s)."
fi
fi
echo "Starting Flask application on port 7860..."
exec python app.py