#!/bin/bash
# Run this script on the GCP VM after SSH-ing in

echo "🚀 Setting up ORA on GCP VM with GPU"
echo "======================================"

# Update system
echo "📦 Updating system..."
sudo apt-get update

# Install Docker
echo "🐳 Installing Docker..."
curl -fsSL https://get.docker.com -o get-docker.sh
sudo sh get-docker.sh
sudo usermod -aG docker $USER

# Install NVIDIA Container Toolkit
echo "🎮 Installing NVIDIA Container Toolkit..."
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
sudo apt-get update
sudo apt-get install -y nvidia-container-toolkit
sudo systemctl restart docker

# Install Git LFS
echo "📥 Installing Git LFS..."
sudo apt-get install -y git-lfs
git lfs install

# Clone ORA repository
echo "📂 Cloning ORA repository..."
cd ~
git clone https://huggingface.co/spaces/Abdalkaderdev/ORA
cd ORA

# Build Docker image
echo "🏗️  Building Docker image..."
sudo docker build -t ora-app .

# Run ORA with GPU
echo "🚀 Starting ORA with GPU..."
sudo docker run -d \
  --gpus all \
  -p 80:7860 \
  --name ora-app \
  --restart unless-stopped \
  -v $(pwd)/important:/app/important \
  ora-app

echo ""
echo "✅ ORA is now running!"
echo "======================================"
echo "Access ORA at: http://$(curl -s ifconfig.me)"
echo ""
echo "📊 Check logs: sudo docker logs -f ora-app"
echo "🔄 Restart: sudo docker restart ora-app"
echo "🛑 Stop: sudo docker stop ora-app"
echo ""
echo "💰 Remember: This VM costs ~$0.35/hour"
echo "🛑 Stop VM when not using: gcloud compute instances stop ora-gpu-vm --zone=us-central1-a"