#!/bin/bash
#===============================================================================
# Upload Apollo to Hugging Face
#===============================================================================
# Uploads the Apollo codebase to the zirobtc/oracle repository
#
# Usage:
#   huggingface-cli login
#   ./scripts/upload_to_hf.sh
#===============================================================================

set -e

REPO_ID="zirobtc/oracle"
APOLLO_DIR="$(cd "$(dirname "$0")/.." && pwd)"

echo "📤 Uploading Apollo to ${REPO_ID}..."
echo "   Source: ${APOLLO_DIR}"
echo ""

cd "$APOLLO_DIR"

# Files/folders to exclude from upload
EXCLUDE_PATTERNS=(
    "venv/"
    "__pycache__/"
    "*.pyc"
    ".git/"
    "data/cache/"
    "data/pump_fun/"
    "checkpoints/"
    "runs/"
    "*.pt"
    "*.bin"
    ".env"
)

# Build exclude args
EXCLUDE_ARGS=""
for pattern in "${EXCLUDE_PATTERNS[@]}"; do
    EXCLUDE_ARGS="$EXCLUDE_ARGS --exclude $pattern"
done

# Upload using huggingface_hub
python3 << EOF
from huggingface_hub import HfApi, upload_folder
import os

api = HfApi()

print("Uploading to ${REPO_ID}...")

upload_folder(
    folder_path="${APOLLO_DIR}",
    repo_id="${REPO_ID}",
    repo_type="model",
    ignore_patterns=[
        "venv/*",
        "__pycache__/*",
        "*.pyc",
        ".git/*",
        "data/cache/*",
        "data/pump_fun/*",
        "checkpoints/*",
        "runs/*",
        "*.pt",
        "*.bin",
        ".env",
        "*.log",
    ],
)

print("✓ Upload complete!")
EOF

echo ""
echo "✅ Upload complete!"
echo "   View at: https://huggingface.co/${REPO_ID}"