setup.py

import os
import sys
import subprocess
import platform
from pathlib import Path

def check_python_version():
    """Check if Python version is compatible"""
    if sys.version_info < (3, 8):
        print("❌ Python 3.8+ required")
        return False
    print(f"✅ Python {sys.version} detected")
    return True

def install_ollama():
    """Install Ollama based on OS"""
    system = platform.system().lower()
    
    print("🔧 Installing Ollama...")
    
    try:
        if system == "linux":
            subprocess.run(["curl", "-fsSL", "https://ollama.com/install.sh", "|", "sh"], shell=True, check=True)
        elif system == "darwin":  # macOS
            print("Please install Ollama from https://ollama.com/download")
            print("Or use: brew install ollama")
        elif system == "windows":
            print("Please download and install Ollama from https://ollama.com/download")
        
        print("✅ Ollama installation initiated")
        return True
    except Exception as e:
        print(f"❌ Ollama installation failed: {e}")
        return False

def setup_virtual_environment():
    """Setup Python virtual environment"""
    venv_path = Path("venv")
    
    if venv_path.exists():
        print("✅ Virtual environment already exists")
        return True
    
    try:
        print("🔧 Creating virtual environment...")
        subprocess.run([sys.executable, "-m", "venv", "venv"], check=True)
        print("✅ Virtual environment created")
        return True
    except Exception as e:
        print(f"❌ Failed to create virtual environment: {e}")
        return False

def install_requirements():
    """Install Python requirements"""
    try:
        print("🔧 Installing Python packages...")
        
        # Determine pip executable based on OS
        if platform.system() == "Windows":
            pip_cmd = ["venv/Scripts/pip"]
        else:
            pip_cmd = ["venv/bin/pip"]
        
        # Upgrade pip first
        subprocess.run(pip_cmd + ["install", "--upgrade", "pip"], check=True)
        
        # Install requirements
        subprocess.run(pip_cmd + ["install", "-r", "requirements.txt"], check=True)
        print("✅ All packages installed successfully")
        return True
    except Exception as e:
        print(f"❌ Package installation failed: {e}")
        return False

def download_ai_models():
    """Download required AI models"""
    models = ["llama3.1:8b", "nomic-embed-text"]
    
    for model in models:
        try:
            print(f"🔧 Downloading {model}...")
            subprocess.run(["ollama", "pull", model], check=True)
            print(f"✅ {model} downloaded successfully")
        except Exception as e:
            print(f"⚠️ Failed to download {model}: {e}")

def create_directories():
    """Create necessary directories"""
    directories = ["data", "logs", "exports", "temp"]
    
    for directory in directories:
        Path(directory).mkdir(exist_ok=True)
        print(f"✅ Created directory: {directory}")

def create_config_files():
    """Create configuration files"""
    
    # .env file
    env_content = """# Local AI System Configuration
DATA_DIRECTORY=./data
LOG_LEVEL=INFO
MODEL_NAME=llama3.1:8b
EMBEDDING_MODEL=nomic-embed-text
CHUNK_SIZE=1000
CHUNK_OVERLAP=200
MAX_FILE_SIZE=100MB
ENABLE_SYSTEM_FILES=false
"""
    
    # VS Code settings
    vscode_settings = """{
    "python.defaultInterpreterPath": "./venv/bin/python",
    "python.terminal.activateEnvironment": true,
    "python.formatting.provider": "black",
    "python.linting.enabled": true,
    "python.linting.flake8Enabled": true,
    "files.associations": {
        "*.py": "python"
    },
    "editor.formatOnSave": true,
    "python.testing.pytestEnabled": true,
    "python.testing.pytestArgs": [
        "tests"
    ]
}"""
    
    # VS Code launch configuration
    launch_config = """{
    "version": "0.2.0",
    "configurations": [
        {
            "name": "Run Local AI System",
            "type": "python",
            "request": "launch",
            "program": "local_ai_system.py",
            "console": "integratedTerminal",
            "cwd": "${workspaceFolder}",
            "env": {
                "PYTHONPATH": "${workspaceFolder}"
            }
        },
        {
            "name": "Streamlit App",
            "type": "python",
            "request": "launch",
            "module": "streamlit",
            "args": ["run", "local_ai_system.py"],
            "console": "integratedTerminal"
        }
    ]
}"""
    
    # Create files
    with open(".env", "w") as f:
        f.write(env_content)
    print("✅ Created .env configuration")
    
    # Create VS Code directory and files
    vscode_dir = Path(".vscode")
    vscode_dir.mkdir(exist_ok=True)
    
    with open(vscode_dir / "settings.json", "w") as f:
        f.write(vscode_settings)
    print("✅ Created VS Code settings")
    
    with open(vscode_dir / "launch.json", "w") as f:
        f.write(launch_config)
    print("✅ Created VS Code launch configuration")

def main():
    """Main setup function"""
    print("🚀 Setting up Local AI File Manager...")
    print("=" * 50)
    
    # Check Python version
    if not check_python_version():
        sys.exit(1)
    
    # Setup virtual environment
    if not setup_virtual_environment():
        sys.exit(1)
    
    # Install requirements
    if not install_requirements():
        sys.exit(1)
    
    # Create directories
    create_directories()
    
    # Create configuration files
    create_config_files()
    
    # Install Ollama
    install_ollama()
    
    # Download models (optional, might take time)
    download_models = input("\\n🤖 Download AI models now? (y/N): ").lower().strip()
    if download_models == 'y':
        download_ai_models()
    
    print("\\n" + "=" * 50)
    print("🎉 Setup complete!")
    print("\\nNext steps:")
    print("1. Activate virtual environment:")
    if platform.system() == "Windows":
        print("   venv\\\\Scripts\\\\activate")
    else:
        print("   source venv/bin/activate")
    print("\\n2. Start the application:")
    print("   streamlit run local_ai_system.py")
    print("\\n3. Or open in VS Code and press F5")

if __name__ == "__main__":
    main()