run_docker.py

#!/usr/bin/env python3
"""
Docker Runner for Text Classification API
Build and run the API using Docker
"""

import subprocess
import sys
import os

def main():
    """Run the Text Classification API with Docker"""
    print("🐳 Starting Text Classification API with Docker...")

    # Check if docker-compose.yml exists in api folder
    api_dir = os.path.join(os.path.dirname(__file__), "api")
    docker_compose_path = os.path.join(api_dir, "docker-compose.yml")
    if not os.path.exists(docker_compose_path):
        print("❌ docker-compose.yml not found in api directory!")
        sys.exit(1)

    # Change to api directory for docker-compose commands
    os.chdir(api_dir)

    try:
        # Build and start the services
        print("🔨 Building Docker images...")
        subprocess.run(["docker-compose", "build"], check=True)

        print("🚀 Starting Docker containers...")
        print("📡 API will be available at: http://localhost:8000")
        print("📖 API documentation: http://localhost:8000/docs")
        print("🔍 Health check: http://localhost:8000/health")
        print("Press Ctrl+C to stop the containers")
        print("-" * 50)

        subprocess.run(["docker-compose", "up"], check=True)

    except KeyboardInterrupt:
        print("\n🛑 Stopping Docker containers...")
        try:
            subprocess.run(["docker-compose", "down"], check=False)
        except:
            pass
        print("👋 Docker containers stopped")
    except subprocess.CalledProcessError as e:
        print(f"❌ Docker command failed: {e}")
        sys.exit(1)

if __name__ == "__main__":
    main()