Spaces:

Really-amin
/

Hoghoghi

Paused

App Files Files Community

Really-amin commited on Aug 4

Commit

8af52bf

verified ·

1 Parent(s): 272dc7c

Update start.sh

Browse files

Files changed (1) hide show

start.sh +245 -124

start.sh CHANGED Viewed

@@ -1,146 +1,267 @@
 #!/bin/bash
-# Start script for Legal Dashboard
-# Suitable for Hugging Face Spaces and Docker environments
-echo "🚀 Starting Legal Dashboard..."
-# Set default environment variables
-export PYTHONPATH=/app
-export PYTHONUNBUFFERED=1
-export DATABASE_DIR=${DATABASE_DIR:-/app/data}
-export LOG_LEVEL=${LOG_LEVEL:-INFO}
 # Create necessary directories
-echo "📁 Creating directories..."
-mkdir -p /app/data /app/cache /app/logs /app/uploads /app/backups /tmp/app_fallback
-# Set permissions if possible (ignore errors)
-chmod -R 755 /app/data /app/cache /app/logs /app/uploads /app/backups 2>/dev/null || true
-chmod -R 777 /tmp/app_fallback 2>/dev/null || true
-# Function to check if port is available
-check_port() {
-    local port=${1:-8000}
-    if lsof -Pi :$port -sTCP:LISTEN -t >/dev/null 2>&1; then
-        echo "⚠️  Port $port is already in use"
-        return 1
     else
-        echo "✅ Port $port is available"
-        return 0
     fi
 }
-# Function to wait for dependencies
-wait_for_deps() {
-    echo "⏳ Checking dependencies..."
-    # Check if we can create database
-    python3 -c "
-import sqlite3
-import os
-db_path = os.environ.get('DATABASE_DIR', '/app/data') + '/test.db'
-try:
-    conn = sqlite3.connect(db_path)
-    conn.close()
-    os.remove(db_path)
-    print('✅ Database access OK')
-except Exception as e:
-    print(f'⚠️  Database access issue: {e}')
-" || echo "Database check completed with warnings"
 }
-# Check environment
-echo "🔍 Environment check..."
-echo "  - Python: $(python3 --version)"
-echo "  - Working directory: $(pwd)"
-echo "  - Database dir: $DATABASE_DIR"
-echo "  - User: $(whoami)"
-echo "  - UID: $(id -u)"
-# Wait for dependencies
-wait_for_deps
-# Check port availability
-PORT=${PORT:-8000}
-if ! check_port $PORT; then
-    echo "🔄 Trying alternative port..."
-    PORT=$((PORT + 1))
-    check_port $PORT || PORT=7860  # HF Spaces default
-fi
-echo "🌐 Using port: $PORT"
-# Health check function
-health_check() {
-    local max_attempts=30
-    local attempt=1
-    echo "🏥 Starting health check..."
-    while [ $attempt -le $max_attempts ]; do
-        if curl -fs http://localhost:$PORT/health >/dev/null 2>&1; then
-            echo "✅ Application is healthy!"
-            return 0
         fi
-        echo "⏳ Health check attempt $attempt/$max_attempts..."
-        sleep 2
-        attempt=$((attempt + 1))
     done
-    echo "❌ Health check failed after $max_attempts attempts"
-    return 1
 }
-# Start application
-echo "🎯 Starting application on port $PORT..."
-# For Hugging Face Spaces
-if [ "$SPACE_ID" != "" ]; then
-    echo "🤗 Running in Hugging Face Spaces environment"
-    # Use gradio or fastapi depending on setup
-    if [ -f "app.py" ]; then
-        echo "📱 Starting with Gradio interface..."
-        python3 app.py
     else
-        echo "🚀 Starting FastAPI server..."
-        uvicorn app.main:app --host 0.0.0.0 --port $PORT --workers 1
     fi
-else
-    # Standard Docker/local environment
-    echo "🐳 Running in standard environment"
-    # Start with uvicorn
-    if command -v uvicorn >/dev/null 2>&1; then
-        echo "🚀 Starting with uvicorn..."
-        uvicorn app.main:app \
-            --host 0.0.0.0 \
-            --port $PORT \
-            --workers 1 \
-            --access-log \
-            --log-level info &
-        # Store PID
-        APP_PID=$!
-        echo "📝 Application PID: $APP_PID"
-        # Wait a bit then check health
-        sleep 5
-        if health_check; then
-            echo "🎉 Application started successfully!"
-            wait $APP_PID
-        else
-            echo "💥 Application failed to start properly"
-            kill $APP_PID 2>/dev/null
-            exit 1
-        fi
     else
-        echo "❌ uvicorn not found, trying with python..."
-        python3 -c "
-import uvicorn
-uvicorn.run('app.main:app', host='0.0.0.0', port=$PORT, workers=1)
-        "
     fi
-fi

 #!/bin/bash
+# Legal Dashboard Startup Script
+# This script helps setup and start the Legal Dashboard application
+set -e
+echo "🏛️ Legal Dashboard - Starting Setup..."
+# Colors for output
+RED='\033[0;31m'
+GREEN='\033[0;32m'
+YELLOW='\033[1;33m'
+BLUE='\033[0;34m'
+NC='\033[0m' # No Color
+# Function to print colored output
+print_status() {
+    echo -e "${GREEN}✅ $1${NC}"
+}
+print_warning() {
+    echo -e "${YELLOW}⚠️  $1${NC}"
+}
+print_error() {
+    echo -e "${RED}❌ $1${NC}"
+}
+print_info() {
+    echo -e "${BLUE}ℹ️  $1${NC}"
+}
+# Check if Docker is installed
+check_docker() {
+    if ! command -v docker &> /dev/null; then
+        print_error "Docker is not installed. Please install Docker first."
+        exit 1
+    fi
+    if ! command -v docker-compose &> /dev/null && ! docker compose version &> /dev/null; then
+        print_error "Docker Compose is not installed. Please install Docker Compose first."
+        exit 1
+    fi
+    print_status "Docker is available"
+}
 # Create necessary directories
+create_directories() {
+    print_info "Creating necessary directories..."
+    mkdir -p data
+    mkdir -p cache
+    mkdir -p logs
+    mkdir -p uploads
+    mkdir -p backups
+    mkdir -p logs/nginx
+    # Set permissions
+    chmod 755 data cache logs uploads backups
+    chmod 755 logs/nginx
+    print_status "Directories created and permissions set"
+}
+# Check if frontend directory exists
+check_frontend() {
+    if [ ! -d "frontend" ]; then
+        print_warning "Frontend directory not found. Creating basic structure..."
+        mkdir -p frontend
+        # Create basic index.html if it doesn't exist
+        if [ ! -f "frontend/index.html" ]; then
+            print_info "Creating basic index.html..."
+            cat > frontend/index.html << 'EOF'
+<!DOCTYPE html>
+<html>
+<head>
+    <title>Legal Dashboard</title>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+</head>
+<body>
+    <h1>Legal Dashboard is Starting...</h1>
+    <p>Please wait while the application loads.</p>
+    <script>
+        setTimeout(() => {
+            window.location.reload();
+        }, 5000);
+    </script>
+</body>
+</html>
+EOF
+        fi
+    fi
+    print_status "Frontend directory checked"
+}
+# Setup environment file
+setup_environment() {
+    if [ ! -f ".env" ]; then
+        print_info "Creating .env file..."
+        cat > .env << 'EOF'
+# Legal Dashboard Environment Configuration
+JWT_SECRET_KEY=your-super-secret-jwt-key-change-this-in-production-$(date +%s)
+DATABASE_DIR=/app/data
+DATABASE_PATH=/app/data/legal_dashboard.db
+LOG_LEVEL=INFO
+ENVIRONMENT=production
+PYTHONPATH=/app
+REDIS_URL=redis://redis:6379/0
+EOF
+        print_status ".env file created"
     else
+        print_status ".env file already exists"
     fi
 }
+# Build and start services
+start_services() {
+    print_info "Building and starting services..."
+    # Stop any existing containers
+    docker-compose down 2>/dev/null || docker compose down 2>/dev/null || true
+    # Build and start
+    if command -v docker-compose &> /dev/null; then
+        docker-compose up --build -d
+    else
+        docker compose up --build -d
+    fi
+    print_status "Services started"
 }
+# Wait for services to be ready
+wait_for_services() {
+    print_info "Waiting for services to be ready..."
+    # Wait for Redis
+    echo -n "Waiting for Redis..."
+    for i in {1..30}; do
+        if docker exec legal_dashboard_redis redis-cli ping > /dev/null 2>&1; then
+            echo " Ready!"
+            break
         fi
+        echo -n "."
+        sleep 1
     done
+    # Wait for FastAPI
+    echo -n "Waiting for FastAPI..."
+    for i in {1..60}; do
+        if curl -fs http://localhost:8000/health > /dev/null 2>&1; then
+            echo " Ready!"
+            break
+        fi
+        echo -n "."
+        sleep 1
+    done
+    print_status "All services are ready"
 }
+# Show status
+show_status() {
+    echo ""
+    echo "🎉 Legal Dashboard is now running!"
+    echo ""
+    echo "📱 Access URLs:"
+    echo "   Main Dashboard: http://localhost"
+    echo "   Direct Access:  http://localhost:8000"
+    echo "   API Documentation: http://localhost:8000/api/docs"
+    echo ""
+    echo "🔧 Management Commands:"
+    echo "   View logs:      docker-compose logs -f legal-dashboard"
+    echo "   Stop services:  docker-compose down"
+    echo "   Restart:        docker-compose restart"
+    echo ""
+    echo "📊 Service Status:"
+    # Check service status
+    if docker ps --format "table {{.Names}}\t{{.Status}}" | grep legal_dashboard_app > /dev/null; then
+        print_status "Legal Dashboard App: Running"
     else
+        print_error "Legal Dashboard App: Not Running"
     fi
+    if docker ps --format "table {{.Names}}\t{{.Status}}" | grep legal_dashboard_redis > /dev/null; then
+        print_status "Redis: Running"
+    else
+        print_error "Redis: Not Running"
+    fi
+    if docker ps --format "table {{.Names}}\t{{.Status}}" | grep legal_dashboard_nginx > /dev/null; then
+        print_status "Nginx: Running"
     else
+        print_warning "Nginx: Not Running (This is optional)"
     fi
+}
+# Main execution
+main() {
+    echo "🚀 Starting Legal Dashboard Setup..."
+    echo "=================================="
+    check_docker
+    create_directories
+    check_frontend
+    setup_environment
+    start_services
+    wait_for_services
+    show_status
+    echo ""
+    echo "✨ Setup completed successfully!"
+    echo "You can now access the Legal Dashboard at http://localhost"
+}
+# Handle script arguments
+case "${1:-}" in
+    "stop")
+        print_info "Stopping Legal Dashboard..."
+        docker-compose down 2>/dev/null || docker compose down 2>/dev/null
+        print_status "Legal Dashboard stopped"
+        ;;
+    "restart")
+        print_info "Restarting Legal Dashboard..."
+        docker-compose restart 2>/dev/null || docker compose restart 2>/dev/null
+        print_status "Legal Dashboard restarted"
+        ;;
+    "logs")
+        print_info "Showing logs..."
+        docker-compose logs -f legal-dashboard 2>/dev/null || docker compose logs -f legal-dashboard 2>/dev/null
+        ;;
+    "status")
+        show_status
+        ;;
+    "clean")
+        print_warning "Cleaning up all containers and data..."
+        read -p "Are you sure? This will delete all data (y/N): " -n 1 -r
+        echo
+        if [[ $REPLY =~ ^[Yy]$ ]]; then
+            docker-compose down -v 2>/dev/null || docker compose down -v 2>/dev/null
+            docker system prune -f
+            rm -rf data cache logs uploads backups
+            print_status "Cleanup completed"
+        fi
+        ;;
+    "")
+        main
+        ;;
+    *)
+        echo "Usage: $0 [stop|restart|logs|status|clean]"
+        echo ""
+        echo "Commands:"
+        echo "  (no args)  - Start the Legal Dashboard"
+        echo "  stop       - Stop all services"
+        echo "  restart    - Restart all services"
+        echo "  logs       - Show application logs"
+        echo "  status     - Show service status"
+        echo "  clean      - Clean up all containers and data (DESTRUCTIVE)"
+        exit 1
+        ;;
+esac