#!/bin/bash

# Script para actualizar yt-dlp en los contenedores sin rebuild completo

echo "═══════════════════════════════════════════════════════════"
echo "  🔄 Actualización Rápida de yt-dlp en Contenedores"
echo "═══════════════════════════════════════════════════════════"
echo ""

# Colores
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
RED='\033[0;31m'
NC='\033[0m'

print_success() {
    echo -e "${GREEN}✅ $1${NC}"
}

print_warning() {
    echo -e "${YELLOW}⚠️  $1${NC}"
}

print_error() {
    echo -e "${RED}❌ $1${NC}"
}

# Verificar que los contenedores estén corriendo
echo "🔍 Verificando contenedores..."

if ! docker ps | grep -q streamlit_panel; then
    print_error "El contenedor streamlit_panel no está corriendo"
    echo "Inicia los contenedores con: docker-compose up -d"
    exit 1
fi

if ! docker ps | grep -q tubescript_api; then
    print_error "El contenedor tubescript_api no está corriendo"
    echo "Inicia los contenedores con: docker-compose up -d"
    exit 1
fi

print_success "Contenedores encontrados"
echo ""

# Actualizar yt-dlp en streamlit_panel
echo "📦 Actualizando yt-dlp en streamlit_panel..."
docker exec streamlit_panel pip install --upgrade yt-dlp

if [ $? -eq 0 ]; then
    print_success "yt-dlp actualizado en streamlit_panel"

    # Verificar versión
    version=$(docker exec streamlit_panel python3 -c "import yt_dlp; print(yt_dlp.version.__version__)" 2>/dev/null)
    if [ ! -z "$version" ]; then
        echo "   Versión instalada: $version"
    fi
else
    print_error "Error al actualizar yt-dlp en streamlit_panel"
fi

echo ""

# Actualizar yt-dlp en tubescript_api
echo "📦 Actualizando yt-dlp en tubescript_api..."
docker exec tubescript_api pip install --upgrade yt-dlp

if [ $? -eq 0 ]; then
    print_success "yt-dlp actualizado en tubescript_api"

    # Verificar versión
    version=$(docker exec tubescript_api python3 -c "import yt_dlp; print(yt_dlp.version.__version__)" 2>/dev/null)
    if [ ! -z "$version" ]; then
        echo "   Versión instalada: $version"
    fi
else
    print_error "Error al actualizar yt-dlp en tubescript_api"
fi

echo ""
echo "═══════════════════════════════════════════════════════════"
print_success "Actualización completada"
echo "═══════════════════════════════════════════════════════════"
echo ""
echo "💡 Ahora puedes probar con un video en vivo en:"
echo "   http://localhost:8501"
echo ""