simple_test.py

#!/usr/bin/env python3
"""
Simple Test Script for Folder-based Damage Detection

This script demonstrates the simplified folder-based processing.
"""

import os
import sys
from pathlib import Path

# Add the project root to the path
sys.path.append(os.path.dirname(os.path.abspath(__file__)))

def create_sample_structure():
    """Create the simple folder structure and add instructions."""
    print("🗂️  Creating simple folder structure...")
    
    # Create folders
    input_dir = Path("input")
    output_dir = Path("output") 
    
    input_dir.mkdir(exist_ok=True)
    output_dir.mkdir(exist_ok=True)
    
    # Create simple instructions
    readme_content = """# How to Use

## Step 1: Add Images
Put your poster images in this folder:
- poster1.jpg
- poster2.png
- poster3.bmp
etc.

## Step 2: Run Detection
From the main project folder, run:
```bash
python3 pretrained_damage_detector.py --folder
```

## Step 3: Check Results
Results will appear in the output/ folder with:
- Annotated images showing damage status
- JSON reports with detailed analysis
"""
    
    with open(input_dir / "README.md", "w") as f:
        f.write(readme_content)
    
    print(f"✅ Created: {input_dir}/")
    print(f"✅ Created: {output_dir}/")
    print(f"📝 Added instructions in {input_dir}/README.md")

def test_if_images_exist():
    """Test if there are any images to process."""
    input_dir = Path("input")
    
    if not input_dir.exists():
        return False, 0
    
    image_extensions = {'.jpg', '.jpeg', '.png', '.bmp', '.tiff', '.webp'}
    image_count = 0
    
    for file_path in input_dir.iterdir():
        if file_path.is_file() and file_path.suffix.lower() in image_extensions:
            image_count += 1
    
    return image_count > 0, image_count

def run_simple_test():
    """Run a simple test of the folder processing."""
    print("\n🧪 Testing Simple Folder Processing...")
    
    has_images, count = test_if_images_exist()
    
    if not has_images:
        print("⚠️  No images found in input/ folder")
        print("📁 Please add some poster images to input/ folder and try again")
        print("\nExample:")
        print("  cp your_poster.jpg input/")
        print("  python3 pretrained_damage_detector.py --folder")
        return
    
    print(f"📸 Found {count} images in input/ folder")
    print("🚀 Running damage detection...")
    
    try:
        # Import and run the detector
        from pretrained_damage_detector import PretrainedDamageDetector
        
        detector = PretrainedDamageDetector(device='cpu')  # Use CPU for testing
        summary = detector.process_input_folder("input", "output")
        
        print("\n" + "="*40)
        print("🎉 TEST RESULTS")
        print("="*40)
        print(f"📸 Total Images: {summary['total_images']}")
        print(f"✅ Processed: {summary['processed']}")
        print(f"🔴 Damaged: {summary['damaged']}")
        print(f"🟢 Undamaged: {summary['undamaged']}")
        print(f"📊 Damage Rate: {summary['damage_rate']}")
        print(f"📄 Results saved to: output/")
        
        # List output files
        output_dir = Path("output")
        if output_dir.exists():
            files = list(output_dir.iterdir())
            print(f"\n📁 Output files ({len(files)}):")
            for file in files:
                print(f"  • {file.name}")
        
        print("\n✅ Test completed successfully!")
        
    except Exception as e:
        print(f"❌ Test failed: {e}")
        print("\nTry installing dependencies:")
        print("  pip install -r requirements.txt")

def show_usage():
    """Show simple usage instructions."""
    print("🎯 Simple Damage Detector Usage")
    print("="*35)
    print()
    print("📁 Folder Structure:")
    print("  input/   ← Put your images here")
    print("  output/  ← Results appear here")
    print()
    print("🚀 Commands:")
    print("  python3 pretrained_damage_detector.py --folder")
    print("  python3 pretrained_damage_detector.py image.jpg")
    print()
    print("🔧 Setup:")
    print("  1. Add images to input/ folder")
    print("  2. Run: python3 pretrained_damage_detector.py --folder")
    print("  3. Check results in output/ folder")
    print()
    print("That's it! Simple and clean. 🎉")

def main():
    """Main test function."""
    print("🎯 Simple Damage Detection Test")
    print("="*35)
    
    # Create folder structure
    create_sample_structure()
    
    # Show usage
    show_usage()
    
    # Test if we can run detection
    print("\n" + "="*40)
    print("🧪 TESTING")
    print("="*40)
    
    run_simple_test()

if __name__ == "__main__":
    main()