utils/view_conversation.py

#!/usr/bin/env python3
"""
Скрипт для швидкого перегляду розмов MAI-DX сесії в терміналі
"""

import os
import json
import sys
import argparse
from datetime import datetime
from rich.console import Console
from rich.panel import Panel
from rich.table import Table
from rich.syntax import Syntax
from rich.markdown import Markdown

# Спробуємо імпортувати Rich, якщо немає - використаємо простий вивід
try:
    from rich.console import Console
    from rich.panel import Panel
    from rich.table import Table
    from rich.syntax import Syntax
    from rich.markdown import Markdown
    RICH_AVAILABLE = True
    console = Console()
except ImportError:
    RICH_AVAILABLE = False
    print("⚠️ Rich не встановлено. Для кращого відображення: pip install rich")

def load_session(case_id, log_dir="mai_dx_logs"):
    """Завантажити конкретну сесію"""
    json_path = os.path.join(log_dir, f"{case_id}.json")
    
    if not os.path.exists(json_path):
        # Спробуємо знайти файл з частковим збігом
        for filename in os.listdir(log_dir):
            if case_id in filename and filename.endswith('.json'):
                json_path = os.path.join(log_dir, filename)
                break
        else:
            return None
            
    try:
        with open(json_path, 'r', encoding='utf-8') as f:
            return json.load(f)
    except Exception as e:
        print(f"❌ Помилка читання файлу: {e}")
        return None

def display_session_rich(session):
    """Відображення сесії з Rich форматуванням"""
    # Заголовок
    console.print(Panel.fit(
        f"[bold blue]MAI-DX Diagnostic Session[/bold blue]\n"
        f"[dim]{session['case_id']}[/dim]",
        border_style="blue"
    ))
    
    # Основна інформація
    info_table = Table(show_header=False, box=None, padding=(0, 2))
    info_table.add_column(style="bold cyan")
    info_table.add_column()
    
    info_table.add_row("📋 Випадок:", session['case_name'])
    info_table.add_row("🕐 Час:", session['timestamp'])
    info_table.add_row("🔧 Режим:", session['mode'])
    info_table.add_row("💊 Діагноз:", f"[bold yellow]{session['diagnosis']}[/bold yellow]")
    info_table.add_row("📊 Точність:", f"{session['confidence']}/5.0 {'✅' if session['confidence'] >= 3.0 else '⚠️'}")
    info_table.add_row("💰 Вартість:", f"${session['cost']} (бюджет: ${session['budget']})")
    info_table.add_row("⏱️ Тривалість:", f"{session['duration']:.1f} секунд")
    
    console.print(info_table)
    console.print()
    
    # Розмови
    conversations = session.get('conversations', [])
    if not conversations:
        console.print("[yellow]⚠️ Розмови не знайдено[/yellow]")
        return
        
    console.print(f"[bold]💬 Розмови ({len(conversations)} раундів):[/bold]\n")
    
    for i, conv in enumerate(conversations):
        # Заголовок раунду
        console.print(Panel(
            f"[bold]Раунд {conv['round_number']}[/bold]\n"
            f"Час: {conv.get('start_time', 'N/A')} - {conv.get('end_time', 'N/A')}\n"
            f"Повідомлень: {len(conv.get('messages', []))}"
            + (f"\nРішення: [yellow]{conv['decision']}[/yellow]" if conv.get('decision') else "")
            + (f"\nВартість: ${conv['cost_incurred']}" if conv.get('cost_incurred') else ""),
            title=f"[cyan]Раунд {i+1}[/cyan]",
            border_style="cyan"
        ))
        
        # Повідомлення
        for msg in conv.get('messages', []):
            agent_color = {
                'Dr. Hypothesis': 'blue',
                'Dr. Test-Chooser': 'green',
                'Dr. Challenger': 'red',
                'Dr. Stewardship': 'yellow',
                'Dr. Checklist': 'magenta',
                'Consensus Coordinator': 'cyan',
                'Judge': 'white',
                'Gatekeeper': 'bright_black'
            }.get(msg['agent_name'], 'white')
            
            msg_type_icon = {
                'reasoning': '🤔',
                'decision': '💡',
                'input': '❓',
                'output': '📊',
                'log': '📝'
            }.get(msg['message_type'], '💬')
            
            console.print(
                f"\n[bold {agent_color}]{msg_type_icon} {msg['agent_name']}[/bold {agent_color}] "
                f"[dim]({msg['message_type']})[/dim]"
            )
            
            # Обмежуємо довжину контенту для кращого відображення
            content = msg['content']
            if len(content) > 500:
                content = content[:500] + "..."
                
            console.print(Panel(
                content,
                border_style=agent_color,
                padding=(0, 1)
            ))

def display_session_simple(session):
    """Відображення сесії без Rich"""
    print("="*60)
    print(f"MAI-DX Diagnostic Session: {session['case_id']}")
    print("="*60)
    
    print(f"\n📋 Випадок: {session['case_name']}")
    print(f"🕐 Час: {session['timestamp']}")
    print(f"💊 Діагноз: {session['diagnosis']}")
    print(f"📊 Точність: {session['confidence']}/5.0")
    print(f"💰 Вартість: ${session['cost']} (бюджет: ${session['budget']})")
    
    conversations = session.get('conversations', [])
    if not conversations:
        print("\n⚠️ Розмови не знайдено")
        return
        
    print(f"\n💬 Розмови ({len(conversations)} раундів):")
    
    for i, conv in enumerate(conversations):
        print(f"\n{'='*40}")
        print(f"Раунд {conv['round_number']}")
        print(f"{'='*40}")
        
        if conv.get('decision'):
            print(f"Рішення: {conv['decision']}")
            
        for msg in conv.get('messages', []):
            print(f"\n[{msg['message_type']}] {msg['agent_name']}:")
            content = msg['content']
            if len(content) > 300:
                content = content[:300] + "..."
            print(content)
            print("-"*40)

def export_markdown(session, output_file=None):
    """Експорт розмов в Markdown"""
    if not output_file:
        output_file = f"{session['case_id']}_conversation.md"
        
    md_content = f"""# MAI-DX Diagnostic Session: {session['case_id']}

## 📋 Інформація про сесію

- **Випадок**: {session['case_name']}
- **Час**: {session['timestamp']}
- **Режим**: {session['mode']}
- **Діагноз**: **{session['diagnosis']}**
- **Точність**: {session['confidence']}/5.0
- **Вартість**: ${session['cost']} (бюджет: ${session['budget']})
- **Тривалість**: {session['duration']:.1f} секунд

## 💬 Розмови

"""
    
    for i, conv in enumerate(session.get('conversations', [])):
        md_content += f"\n### Раунд {conv['round_number']}\n\n"
        
        if conv.get('decision'):
            md_content += f"**Рішення**: {conv['decision']}\n\n"
            
        for msg in conv.get('messages', []):
            md_content += f"\n#### {msg['agent_name']} [{msg['message_type']}]\n\n"
            md_content += f"{msg['content']}\n\n"
            md_content += "---\n"
            
    with open(output_file, 'w', encoding='utf-8') as f:
        f.write(md_content)
        
    print(f"\n✅ Експортовано в Markdown: {output_file}")

def main():
    parser = argparse.ArgumentParser(description='Перегляд розмов MAI-DX сесії')
    parser.add_argument('case_id', help='ID сесії або його частина')
    parser.add_argument('--dir', default='mai_dx_logs', help='Директорія з логами')
    parser.add_argument('--simple', action='store_true', help='Простий вивід без Rich')
    parser.add_argument('--export-md', action='store_true', help='Експортувати в Markdown')
    parser.add_argument('--raw', action='store_true', help='Показати сирий текст')
    
    args = parser.parse_args()
    
    # Завантажуємо сесію
    session = load_session(args.case_id, args.dir)
    
    if not session:
        print(f"❌ Сесія '{args.case_id}' не знайдена в {args.dir}")
        
        # Показуємо доступні сесії
        print("\n📁 Доступні сесії:")
        for filename in sorted(os.listdir(args.dir)):
            if filename.endswith('.json') and not filename.endswith('_analytics.json'):
                print(f"  - {filename[:-5]}")
        sys.exit(1)
        
    # Показуємо сирий текст якщо потрібно
    if args.raw:
        raw_file = os.path.join(args.dir, f"{session['case_id']}_raw.txt")
        if os.path.exists(raw_file):
            with open(raw_file, 'r', encoding='utf-8') as f:
                print(f.read())
        else:
            print("❌ Файл з сирим текстом не знайдено")
        return
        
    # Експортуємо в Markdown якщо потрібно
    if args.export_md:
        export_markdown(session)
        return
        
    # Відображаємо сесію
    if args.simple or not RICH_AVAILABLE:
        display_session_simple(session)
    else:
        display_session_rich(session)

if __name__ == "__main__":
    main()