"""
Test script for NLP Text Analysis Bot
Developer: RSK World
Website: https://rskworld.in
Email: help@rskworld.in
Phone: +91 93305 39277
Year: 2026
"""

from nlp_pipeline import NLPPipeline
import json

def test_analysis():
    """
    Test the NLP pipeline with sample text
    Developer: RSK World - https://rskworld.in
    """
    # Initialize pipeline
    pipeline = NLPPipeline()
    
    # Sample text for testing
    sample_text = """
    Apple Inc. is an American multinational technology company headquartered in Cupertino, California.
    The company was founded by Steve Jobs, Steve Wozniak, and Ronald Wayne in 1976.
    Apple is known for its innovative products like the iPhone, iPad, and MacBook.
    The company has a strong presence in markets worldwide and continues to lead in technology innovation.
    """
    
    print("Testing NLP Text Analysis Bot...")
    print("=" * 50)
    print(f"Input Text: {sample_text.strip()}")
    print("=" * 50)
    
    # Run analysis
    results = pipeline.analyze(sample_text)
    
    # Display results
    print("\n📊 ANALYSIS RESULTS")
    print("=" * 50)
    
    # Summary
    summary = results.get('summary', {})
    print(f"\n📈 Summary:")
    print(f"  Word Count: {summary.get('word_count', 0)}")
    print(f"  Sentence Count: {summary.get('sentence_count', 0)}")
    print(f"  Sentiment: {summary.get('sentiment_label', 'N/A')}")
    print(f"  Entities Found: {summary.get('entity_count', 0)}")
    
    # Sentiment
    sentiment = results.get('sentiment', {})
    print(f"\n😊 Sentiment Analysis:")
    print(f"  Label: {sentiment.get('label', 'N/A')}")
    print(f"  Score: {sentiment.get('score', 0):.3f}")
    
    # Entities
    entities = results.get('entities', {})
    print(f"\n🏷️  Named Entities ({entities.get('total_entities', 0)} found):")
    for entity in entities.get('entities', [])[:10]:
        print(f"  - {entity.get('text')} ({entity.get('label')})")
    
    # Semantic
    semantic = results.get('semantic', {})
    print(f"\n🧠 Semantic Understanding:")
    print(f"  Topics: {', '.join(semantic.get('topics', [])[:5])}")
    print(f"  Keywords: {', '.join(semantic.get('keywords', [])[:10])}")
    
    print("\n" + "=" * 50)
    print("✅ Analysis Complete!")
    
    # Save results to JSON
    with open('test_results.json', 'w', encoding='utf-8') as f:
        json.dump(results, f, indent=2, ensure_ascii=False)
    print("📄 Results saved to test_results.json")

if __name__ == '__main__':
    test_analysis()