from services.classifier import classify_garment
import requests
import os
import shutil

def test_robustness():
    print("Testing Classifier Robustness...")
    
    # 1. Test with "Jeans.jpg" (Should be Bottom via Fallback if API fails)
    img_url = "https://upload.wikimedia.org/wikipedia/commons/thumb/5/5a/Jeans.jpg/440px-Jeans.jpg"
    filename = "Jeans.jpg" # Capitalized to test case insensitivity
    
    with open(filename, "wb") as f:
        f.write(requests.get(img_url).content)
        
    print(f"\n1. input: {filename}")
    cat, desc, tags = classify_garment(filename)
    print(f"   -> Result: Category={cat}, Desc={desc}, Source={tags}")
    
    if cat == "bottom":
        print("✅ SUCCESS: Detected matching bottom (Logic Works)")
    else:
        print("❌ FAILURE: Did not detect bottom")

    # 2. Test with "black_tshirt.jpg" (Should be Top)
    filename2 = "black_tshirt.jpg"
    shutil.copy(filename, filename2) # Just copy content, name matters
    
    print(f"\n2. input: {filename2}")
    cat, desc, tags = classify_garment(filename2)
    print(f"   -> Result: Category={cat}, Desc={desc}, Source={tags}")
    
    if cat == "top":
         print("✅ SUCCESS: Detected matching top")
    else:
         print("❌ FAILURE: Failed top detection")

if __name__ == "__main__":
    test_robustness()