#!/usr/bin/env python3
"""
BeastBullet v1.0 - Main Entry Point
Sonnet-level AI reasoning on potato hardware
"""

import sys
import argparse
from pathlib import Path

# Add code directory to path
sys.path.insert(0, str(Path(__file__).parent / 'code'))

def main():
    parser = argparse.ArgumentParser(
        description='BeastBullet v1.0 - Premise-Locked Edition',
        formatter_class=argparse.RawDescriptionHelpFormatter,
        epilog="""
Examples:
  python main.py                    # Interactive mode
  python main.py --demo             # Run demo queries
  python main.py --test             # Run test suite
  python main.py --query "What is 15%% of 240?"

Features:
  • 18 specialized expert models
  • Premise-lock validator (zero logic drift)
  • 91%% quality score, 96%% confidence
  • Runs on potato hardware (4GB RAM)

Mind it! 🎯
        """
    )
    
    parser.add_argument('--query', '-q', type=str, help='Single query to process')
    parser.add_argument('--demo', action='store_true', help='Run demo queries')
    parser.add_argument('--test', action='store_true', help='Run test suite')
    parser.add_argument('--victory', action='store_true', help='Run victory test')
    parser.add_argument('--adversarial', action='store_true', help='Run adversarial tests')
    parser.add_argument('--premise-lock', action='store_true', help='Test premise-lock validator')
    parser.add_argument('--interactive', '-i', action='store_true', help='Interactive mode (default)')
    
    args = parser.parse_args()
    
    print("="*70)
    print("🥔 BeastBullet v1.0 - Premise-Locked Edition")
    print("="*70)
    print("Sonnet-level AI reasoning on potato hardware")
    print("91% quality | 96% confidence | Zero hallucinations")
    print()
    
    # Run appropriate mode
    if args.demo:
        run_demo()
    elif args.test:
        run_all_tests()
    elif args.victory:
        run_victory_test()
    elif args.adversarial:
        run_adversarial_test()
    elif args.premise_lock:
        run_premise_lock_test()
    elif args.query:
        run_single_query(args.query)
    else:
        run_interactive()


def run_demo():
    """Run demo queries"""
    print("Running demo queries...\n")
    import demo_integration
    demo_integration.main()


def run_all_tests():
    """Run all test suites"""
    print("Running all test suites...\n")
    
    print("1. Victory Run...")
    import victory_run
    victory_run.main()
    
    print("\n2. Adversarial Tests...")
    import adversarial_test
    adversarial_test.main()
    
    print("\n3. Premise-Lock Validation...")
    import premise_lock_validator
    premise_lock_validator.test_premise_lock()


def run_victory_test():
    """Run victory test suite"""
    print("Running victory test suite...\n")
    import victory_run
    victory_run.main()


def run_adversarial_test():
    """Run adversarial stress tests"""
    print("Running adversarial stress tests...\n")
    import adversarial_test
    adversarial_test.main()


def run_premise_lock_test():
    """Test premise-lock validator"""
    print("Testing premise-lock validator...\n")
    import premise_lock_validator
    premise_lock_validator.test_premise_lock()


def run_single_query(query):
    """Process a single query"""
    print(f"Processing query: {query}\n")
    import blackboard_demo
    demo = blackboard_demo.BlackboardTinyLlamaDemo()
    demo.process_query(query)


def run_interactive():
    """Interactive mode"""
    print("Interactive Mode")
    print("Type 'quit' or 'exit' to stop")
    print("Type 'help' for commands")
    print()
    
    import blackboard_demo
    demo = blackboard_demo.BlackboardTinyLlamaDemo()
    
    while True:
        try:
            query = input("BeastBullet> ").strip()
            
            if not query:
                continue
            
            if query.lower() in ['quit', 'exit', 'q']:
                print("\nMind it! 🎯")
                break
            
            if query.lower() == 'help':
                print_help()
                continue
            
            print()
            demo.process_query(query)
            print()
            
        except KeyboardInterrupt:
            print("\n\nMind it! 🎯")
            break
        except Exception as e:
            print(f"\nError: {e}\n")


def print_help():
    """Print help message"""
    print("""
Available commands:
  help              - Show this help
  quit, exit, q     - Exit interactive mode
  
  Or just type your query!
  
Examples:
  What is 15% of 240?
  Calculate compound interest on $1000 at 5% for 3 years
  If all cats are mammals, and Fluffy is a cat, what is Fluffy?
  
Features:
  • Multi-expert collaboration via shared blackboard
  • Premise-lock validator prevents logic drift
  • Confidence calibration (honest uncertainty)
  • Zero hallucinations guaranteed
    """)


if __name__ == "__main__":
    main()