main .py

import argparse
import logging
from auto_quantum_genius.config_loader import load_config
from auto_quantum_genius.quantum_engine import QuantumEngine

logger = logging.getLogger(__name__)

def parse_args():
    parser = argparse.ArgumentParser(
        description="Auto-Quantum-Genius: symbolic analysis with quantum heuristics"
    )
    parser.add_argument("--config", required=True, help="Path to YAML config file")
    parser.add_argument(
        "--mode", choices=["train", "infer"], default="infer",
        help="Operation mode"
    )
    return parser.parse_args()

def main():
    logging.basicConfig(level=logging.INFO)
    args = parse_args()
    cfg = load_config(args.config)
    engine = QuantumEngine(cfg)
    
    if args.mode == "train":
        result = engine.train()
    else:
        result = engine.infer()

    logger.info("Completed %s step; result summary: %s", args.mode, result)

if __name__ == "__main__":
    main()