#!/usr/bin/env python3
"""
Direct upload script to Hugging Face Spaces.
This script will guide you through creating and uploading your Memory Chat application.
"""

import os
import sys
import subprocess
import webbrowser
from pathlib import Path

def check_prerequisites():
    """Check if prerequisites are installed."""
    print("🔍 Checking prerequisites...")

    # Check if git is installed
    try:
        result = subprocess.run(['git', '--version'], capture_output=True, text=True)
        if result.returncode == 0:
            print("✅ Git is installed")
        else:
            print("❌ Git is not installed. Please install Git first.")
            return False
    except FileNotFoundError:
        print("❌ Git is not installed. Please install Git first.")
        return False

    # Check if files exist
    required_files = [
        'app.py',
        'memory_manager.py',
        'chat_interface.py',
        'config.py',
        'requirements.txt',
        'README.md'
    ]

    missing_files = []
    for file in required_files:
        if not os.path.exists(file):
            missing_files.append(file)

    if missing_files:
        print(f"❌ Missing files: {', '.join(missing_files)}")
        print("Please ensure all required files are in the current directory.")
        return False
    else:
        print("✅ All required files found")

    return True

def create_space():
    """Guide user through creating a Space on Hugging Face."""
    print("\n🌐 Creating Hugging Face Space...")

    print("1. Open your web browser and go to: https://huggingface.co/spaces")
    print("2. Sign in to your Hugging Face account")
    print("3. Click 'Create new Space' (top right)")
    print("4. Choose a Space name (e.g., 'your-username/memory-chat')")
    print("5. Select 'Gradio' as the Space type")
    print("6. Choose 'Public' visibility")
    print("7. Click 'Create Space'")

    input("\nPress Enter after you've created your Space...")

def get_space_url():
    """Get the Space URL from user."""
    print("\n📍 Please enter your Space URL:")
    print("It should look like: https://huggingface.co/spaces/your-username/your-space-name")
    space_url = input("Space URL: ").strip()

    if not space_url.startswith("https://huggingface.co/spaces/"):
        print("❌ Invalid URL format. Please enter a valid Hugging Face Space URL.")
        return get_space_url()

    return space_url

def clone_space_repo(space_url):
    """Clone the Space repository."""
    print(f"\n📥 Cloning Space repository: {space_url}")

    # Extract the repo name from URL
    repo_name = space_url.split('/')[-1]
    username = space_url.split('/')[-2]

    clone_url = f"https://huggingface.co/spaces/{username}/{repo_name}"

    try:
        subprocess.run(['git', 'clone', clone_url], check=True)
        return repo_name
    except subprocess.CalledProcessError as e:
        print(f"❌ Failed to clone repository: {e}")
        return None

def copy_files_to_repo(repo_name):
    """Copy files to the cloned repository."""
    print(f"\n📁 Copying files to {repo_name}...")

    files_to_copy = [
        ('app.py', 'app.py'),
        ('memory_manager.py', 'memory_manager.py'),
        ('chat_interface.py', 'chat_interface.py'),
        ('config.py', 'config.py'),
        ('requirements.txt', 'requirements.txt'),
        ('README.md', 'README.md')
    ]

    for src, dst in files_to_copy:
        try:
            subprocess.run(['cp', src, f"{repo_name}/{dst}"], check=True)
            print(f"✅ Copied {src} -> {repo_name}/{dst}")
        except subprocess.CalledProcessError as e:
            print(f"❌ Failed to copy {src}: {e}")
            return False

    return True

def commit_and_push(repo_name, username):
    """Commit and push changes to the Space."""
    print(f"\n🚀 Committing and pushing changes...")

    try:
        # Change to repo directory
        os.chdir(repo_name)

        # Add files
        subprocess.run(['git', 'add', '.'], check=True)

        # Commit
        subprocess.run(['git', 'commit', '-m', 'Initial commit: Memory Chat application'], check=True)

        # Push
        subprocess.run(['git', 'push', 'origin', 'main'], check=True)

        print("✅ Successfully pushed to Hugging Face Spaces!")
        print(f"🌐 Your Space will be available at: https://huggingface.co/spaces/{username}/{repo_name}")

        return True

    except subprocess.CalledProcessError as e:
        print(f"❌ Failed to commit/push: {e}")
        return False

def main():
    """Main deployment function."""
    print("🚀 Memory Chat - Hugging Face Spaces Deployment")
    print("=" * 50)

    # Check prerequisites
    if not check_prerequisites():
        sys.exit(1)

    # Create Space on Hugging Face
    create_space()

    # Get Space URL
    space_url = get_space_url()
    username = space_url.split('/')[-2]

    # Clone the Space repository
    repo_name = clone_space_repo(space_url)
    if not repo_name:
        sys.exit(1)

    # Copy files
    if not copy_files_to_repo(repo_name):
        sys.exit(1)

    # Commit and push
    if commit_and_push(repo_name, username):
        print("\n🎉 Deployment successful!")
        print("Your Space will build automatically and be available shortly.")
        print("Check the 'Logs' tab in your Space for build status.")
    else:
        print("\n❌ Deployment failed. Please check the error messages above.")

if __name__ == "__main__":
    main()