#!/usr/bin/env python3
"""
Debug script to test HF Spaces environment
"""
import os
import sys
import subprocess
import time
import requests
from pathlib import Path

def check_files():
    """Check if required files exist"""
    print("🔍 Checking required files...")
    files_to_check = [
        "dispatcher.py",
        "worker.py", 
        "utils.py",
        "latent_stats.json",
        "static/index.html",
        "requirements.txt"
    ]
    
    for file in files_to_check:
        if os.path.exists(file):
            print(f"✅ {file} exists")
        else:
            print(f"❌ {file} missing")
    
    print(f"📁 Current directory: {os.getcwd()}")
    print(f"📋 Directory contents:")
    for item in os.listdir('.'):
        print(f"   {item}")

def test_imports():
    """Test if imports work"""
    print("\n🔍 Testing imports...")
    try:
        import torch
        print(f"✅ torch {torch.__version__}")
    except Exception as e:
        print(f"❌ torch failed: {e}")
    
    try:
        from utils import initialize_model
        print("✅ utils imports work")
    except Exception as e:
        print(f"❌ utils import failed: {e}")
    
    try:
        import fastapi
        print(f"✅ fastapi {fastapi.__version__}")
    except Exception as e:
        print(f"❌ fastapi failed: {e}")

def test_dispatcher():
    """Test dispatcher startup"""
    print("\n🔍 Testing dispatcher...")
    try:
        # Try to start dispatcher
        proc = subprocess.Popen([
            sys.executable, "dispatcher.py", "--port", "7860"
        ], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
        
        # Wait a bit
        time.sleep(3)
        
        # Check if it's running
        if proc.poll() is None:
            print("✅ Dispatcher process is running")
            
            # Test HTTP request
            try:
                response = requests.get("http://localhost:7860/", timeout=5)
                if response.status_code == 200:
                    print("✅ Dispatcher HTTP response OK")
                else:
                    print(f"⚠️ Dispatcher HTTP status: {response.status_code}")
            except Exception as e:
                print(f"❌ Dispatcher HTTP test failed: {e}")
                
        else:
            print("❌ Dispatcher process died")
            stdout, stderr = proc.communicate()
            print(f"📋 Stdout: {stdout.decode()}")
            print(f"📋 Stderr: {stderr.decode()}")
        
        # Clean up
        proc.terminate()
        proc.wait()
        
    except Exception as e:
        print(f"❌ Failed to test dispatcher: {e}")

def main():
    print("🚀 HF Spaces Debug Script")
    print("=" * 50)
    
    check_files()
    test_imports()
    test_dispatcher()
    
    print("\n📊 Environment info:")
    print(f"Python: {sys.version}")
    print(f"Platform: {sys.platform}")
    print(f"CWD: {os.getcwd()}")
    
    # Check for GPU
    try:
        import torch
        if torch.cuda.is_available():
            print(f"🎮 CUDA available: {torch.cuda.device_count()} devices")
        else:
            print("💻 Running on CPU")
    except:
        print("❓ Cannot check GPU status")

if __name__ == "__main__":
    main()