from data_comparator import KSTCoordiComparator

def test_sheet_filtering():
    comparator = KSTCoordiComparator('data/sample-data.xlsx')
    if comparator.load_data():
        print("=== SHEET FILTERING TEST ===")
        
        # Test All Sheets
        summary_all = comparator.get_comparison_summary()
        print(f"All Sheets:")
        print(f"  Available sheets: {summary_all['sheet_names']}")
        print(f"  Matched items: {summary_all['matched_items_count']}")
        print(f"  KST only: {summary_all['mismatches']['kst_only_count']}")
        print(f"  Coordi only: {summary_all['mismatches']['coordi_only_count']}")
        print()
        
        # Test TH URGENT only
        summary_th = comparator.get_comparison_summary('TH URGENT')
        print(f"TH URGENT only:")
        print(f"  Matched items: {summary_th['matched_items_count']}")
        print(f"  KST only: {summary_th['mismatches']['kst_only_count']}")
        print(f"  Coordi only: {summary_th['mismatches']['coordi_only_count']}")
        print()
        
        # Test US URGENT only
        summary_us = comparator.get_comparison_summary('US URGENT')
        print(f"US URGENT only:")
        print(f"  Matched items: {summary_us['matched_items_count']}")
        print(f"  KST only: {summary_us['mismatches']['kst_only_count']}")
        print(f"  Coordi only: {summary_us['mismatches']['coordi_only_count']}")
        print()
        
        print("✓ Sheet filtering functionality working!")
        print("✓ Web GUI ready at http://localhost:8080")
        print("✓ Dropdown will show: All Sheets, TH URGENT, US URGENT")

if __name__ == "__main__":
    test_sheet_filtering()