#!/usr/bin/env python3
"""
CLI для агента с рефлексией и try/except retry на LangGraph
"""

import argparse
import sys
from graph import build_reflection_graph
from state import ReflectState

DEMO_QUESTION = "Объясни студенту разницу между tool и resource в MCP"

def run_agent(question: str, max_rounds: int = 2, max_retries: int = 3, verbose: bool = True):
    """Запускает агента с рефлексией и retry"""
    if verbose:
        print("\n" + "="*80)
        print(f"🤖 ЗАПУСК АГЕНТА С РЕФЛЕКСИЕЙ И TRY/EXCEPT RETRY")
        print(f"📋 Вопрос: {question}")
        print(f"🔄 Макс. раундов рефлексии: {max_rounds}")
        print(f"🔄 Макс. попыток при ошибках: {max_retries}")
        print("="*80)
    
    # Создаём граф
    app = build_reflection_graph(max_rounds=max_rounds, max_retries=max_retries)
    
    # Состояние
    initial_state: ReflectState = {
        "question": question,
        "draft": "",
        "critique": "",
        "verdict": "needs_revision",
        "round": 1,
        "max_rounds": max_rounds,
        "error": None
    }
    
    # Запуск
    final_state = app.invoke(initial_state)
    
    # Вывод
    if verbose:
        print("\n" + "="*80)
        print("✅ ФИНАЛЬНЫЙ РЕЗУЛЬТАТ")
        print("="*80)
        print(f"📝 Итоговый ответ (раунд {final_state['round']}):")
        print("-"*80)
        print(final_state["draft"])
        print("-"*80)
        
        if final_state.get("verdict") == "ok":
            print("🎉 Ответ одобрен критиком!")
        elif final_state.get("error"):
            print(f"⚠️ Завершено с ошибкой: {final_state['error']}")
        else:
            print(f"⚠️ Достигнут лимит раундов ({max_rounds})")
    
    return final_state

def interactive_mode():
    """Интерактивный режим"""
    print("\n🎮 Интерактивный режим")
    print("Введите 'exit' для выхода\n")
    
    while True:
        question = input("Ваш вопрос: ").strip()
        if question.lower() == 'exit':
            break
        if not question:
            continue
        
        max_rounds = input("Макс. раундов рефлексии (по умолч. 2): ").strip()
        max_rounds = int(max_rounds) if max_rounds.isdigit() else 2
        
        max_retries = input("Макс. попыток при ошибках (по умолч. 3): ").strip()
        max_retries = int(max_retries) if max_retries.isdigit() else 3
        
        run_agent(question, max_rounds, max_retries, verbose=True)
        print("\n" + "─"*80 + "\n")

def main():
    parser = argparse.ArgumentParser(
        description="LangGraph агент с рефлексией и try/except retry",
        formatter_class=argparse.RawDescriptionHelpFormatter,
        epilog="""
Примеры:
  python cli.py --demo                    # Запуск на демо-вопросе
  python cli.py -q "Что такое AI?"        # Свой вопрос
  python cli.py -q "..." --max-rounds 3   # 3 раунда рефлексии
  python cli.py -q "..." --max-retries 5  # 5 попыток при ошибках
  python cli.py --interactive             # Интерактивный режим
        """
    )
    
    parser.add_argument("-q", "--question", help="Вопрос для агента")
    parser.add_argument("--max-rounds", type=int, default=2, help="Макс. раундов рефлексии (по умолч. 2)")
    parser.add_argument("--max-retries", type=int, default=3, help="Макс. попыток при ошибках (по умолч. 3)")
    parser.add_argument("--demo", action="store_true", help="Запустить демо-вопрос")
    parser.add_argument("-i", "--interactive", action="store_true", help="Интерактивный режим")
    parser.add_argument("--quiet", action="store_true", help="Тихий режим (только ответ)")
    
    args = parser.parse_args()
    
    if args.interactive:
        interactive_mode()
    elif args.demo or (not args.question):
        run_agent(DEMO_QUESTION, args.max_rounds, args.max_retries, verbose=not args.quiet)
    elif args.question:
        run_agent(args.question, args.max_rounds, args.max_retries, verbose=not args.quiet)
    else:
        parser.print_help()

if __name__ == "__main__":
    main()