#!/usr/bin/env python3
"""
Test script for the Gradio UI functionality
"""

import sys
import os

# Add paths for imports
sys.path.append(os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(__file__))), 'hf_space'))
sys.path.append(os.path.dirname(os.path.dirname(__file__)))

def test_mcp_client():
    """Test the MCP client integration"""
    print("🧪 Testing MCP Client Integration...")
    
    try:
        from mcp_client import find_best_spots
        
        # Test with a simple location
        print("   Testing: Málaga, Spain")
        result = find_best_spots("Málaga, Spain", max_distance_km=50, top_n=3)
        
        if result.get("ok"):
            spots = result.get("results", [])
            ai_summary = result.get("ai_summary", "")
            
            print(f"   ✅ Found {len(spots)} surf spots")
            if ai_summary:
                print(f"   🤖 AI Summary: {ai_summary[:100]}...")
            
            for i, spot in enumerate(spots[:2], 1):
                print(f"   {i}. {spot['name']} - Score: {spot['score']}/100")
                
            return True
        else:
            print(f"   ❌ Error: {result.get('error')}")
            return False
            
    except Exception as e:
        print(f"   💥 Exception: {e}")
        return False

def test_ui_functions():
    """Test UI formatting functions"""
    print("\n🎨 Testing UI Functions...")
    
    try:
        from app import format_spot_results, run_surf_finder
        
        # Test with sample data
        sample_spots = [
            {
                "name": "Test Spot",
                "score": 85.5,
                "distance_km": 25.3,
                "conditions": {"wave_height": 1.5, "wind_speed": 10},
                "explain": "Good conditions for surfing with moderate waves and favorable wind direction"
            }
        ]
        
        spots_html, ai_html, reasoning = format_spot_results(sample_spots, "Test AI summary")
        
        if "Test Spot" in spots_html and "85.5" in spots_html:
            print("   ✅ Spot formatting works correctly")
        else:
            print("   ❌ Spot formatting failed")
            return False
            
        if "Test AI summary" in ai_html:
            print("   ✅ AI summary formatting works")
        else:
            print("   ❌ AI summary formatting failed")
            return False
            
        return True
        
    except Exception as e:
        print(f"   💥 Exception: {e}")
        return False

def main():
    """Run all tests"""
    print("🚀 Testing Surf Spot Finder UI Components")
    print("=" * 50)
    
    # Test MCP integration
    client_ok = test_mcp_client()
    
    # Test UI functions
    ui_ok = test_ui_functions()
    
    print("\n" + "=" * 50)
    print("📊 Test Results:")
    print(f"   MCP Client: {'✅' if client_ok else '❌'}")
    print(f"   UI Functions: {'✅' if ui_ok else '❌'}")
    
    if client_ok and ui_ok:
        print("\n🎉 All tests passed! UI is ready for deployment.")
        print("💡 The Gradio app should work correctly on HF Space.")
    else:
        print("\n⚠️  Some tests failed. Check the issues above.")
    
    return client_ok and ui_ok

if __name__ == "__main__":
    success = main()
    sys.exit(0 if success else 1)