docker_entrypoint.py

#!/usr/bin/env python3
"""Docker entrypoint — supports API / CLI / Gradio modes."""

import argparse
import sys
from pathlib import Path

sys.path.insert(0, str(Path(__file__).resolve().parent / "src"))


def main():
    parser = argparse.ArgumentParser(description="PPTGen-v2 Docker Entrypoint")
    parser.add_argument("--mode", choices=["api", "cli", "gradio", "test"], default="api")
    parser.add_argument("--pdf", type=str, help="PDF path for CLI mode")
    parser.add_argument("--scene", choices=["a", "b"], default="a")
    parser.add_argument("--port", type=int, default=7860)
    args = parser.parse_args()

    if args.mode == "test":
        import pytest
        code = pytest.main(["-v", "tests/"])
        sys.exit(code)

    if args.mode == "cli":
        from orchestrator.pipeline import Pipeline
        pipeline = Pipeline()
        pdf_path = args.pdf or "data/attention_is_all_you_need.pdf"
        result = pipeline.run({"pdf_file": pdf_path})
        print(f"✅ Pipeline complete: {result.get('status')}")
        print(f"📄 Output: {result.get('pptx_path')}")
        sys.exit(0)

    if args.mode == "api":
        print(f"🚀 API mode on port {args.port}")
        try:
            from fastapi import FastAPI
            import uvicorn
            app = FastAPI(title="PPTGen-v2 API")
            uvicorn.run(app, host="0.0.0.0", port=args.port)
        except ImportError:
            print("⚠️  fastapi/uvicorn not installed. Install with: pip install fastapi uvicorn")
            sys.exit(1)
        sys.exit(0)

    if args.mode == "gradio":
        from web.app import main as launch_ui
        launch_ui()
        sys.exit(0)


if __name__ == "__main__":
    main()