run_docs.py

#!/usr/bin/env python3
"""
MkDocs Runner for Text Classification API Documentation
Serve the API documentation locally
"""

import subprocess
import sys
import os

def main():
    """Serve the MkDocs documentation"""
    print("📚 Starting MkDocs documentation server...")

    # Change to api directory where mkdocs.yml is located
    api_dir = os.path.join(os.path.dirname(__file__), "api")
    if not os.path.exists(api_dir):
        print("❌ API directory not found!")
        sys.exit(1)

    os.chdir(api_dir)

    # Check if mkdocs.yml exists
    if not os.path.exists("mkdocs.yml"):
        print("❌ mkdocs.yml not found in api directory!")
        sys.exit(1)

    try:
        print("📖 Documentation will be available at: http://localhost:8001")
        print("Press Ctrl+C to stop the documentation server")
        print("-" * 50)

        subprocess.run([sys.executable, "-m", "mkdocs", "serve", "--dev-addr=127.0.0.1:8001"], check=True)

    except KeyboardInterrupt:
        print("\n👋 Documentation server stopped")
    except subprocess.CalledProcessError as e:
        print(f"❌ Failed to start documentation server: {e}")
        sys.exit(1)

if __name__ == "__main__":
    main()