feat: token count implement

2026-03-17 12:07:56 -03:00
parent 41279585f6
commit be699ad6ab
44 changed files with 9032 additions and 167 deletions
@@ -0,0 +1,93 @@
+#!/bin/bash
+# Deploy Bark TTS to t-800 server
+# Usage: ./deploy_to_t800.sh
+
+set -e
+
+T800_HOST="t-800"
+T800_USER="juan"
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+
+echo "=========================================="
+echo "  Deploying Bark TTS to t-800"
+echo "=========================================="
+echo ""
+
+# Check if SSH key exists
+if [ ! -f ~/.ssh/id_rsa.pub ]; then
+    echo "SSH key not found. Generating one..."
+    ssh-keygen -t rsa -b 4096 -f ~/.ssh/id_rsa -N "" -C "openccb_bark_deployment"
+    echo ""
+    echo "Now copy your SSH key to t-800:"
+    echo "  ssh-copy-id ${T800_USER}@${T800_HOST}"
+    echo ""
+    read -p "Press Enter after copying the key..."
+fi
+
+# Copy installation script to t-800
+echo "Copying installation script to t-800..."
+scp "${SCRIPT_DIR}/install_bark_tts.sh" ${T800_USER}@${T800_HOST}:/tmp/install_bark_tts.sh
+
+# Execute installation on t-800 with pseudo-terminal
+echo ""
+echo "Connecting to t-800 and installing Bark TTS..."
+echo "This may take 10-15 minutes depending on internet speed..."
+echo "You'll be prompted for your password..."
+echo ""
+
+ssh -t ${T800_USER}@${T800_HOST} << 'ENDSSH'
+    echo "Connected to t-800"
+    echo "Hostname: $(hostname)"
+    echo "Memory: $(free -h | grep Mem | awk '{print $2}')"
+    echo "Disk: $(df -h / | tail -1 | awk '{print $4}') available"
+    echo ""
+    
+    # Install jq if not present
+    if ! command -v jq &> /dev/null; then
+        echo "Installing jq..."
+        sudo apt-get update && sudo apt-get install -y jq
+    fi
+    
+    # Make script executable and run
+    chmod +x /tmp/install_bark_tts.sh
+    echo "Running Bark installation..."
+    sudo /tmp/install_bark_tts.sh
+    
+    # Clean up
+    rm /tmp/install_bark_tts.sh
+    
+    # Wait for service to be ready
+    echo ""
+    echo "Waiting for Bark API to be ready..."
+    sleep 10
+    
+    # Test the API
+    echo "Testing Bark API..."
+    if curl -s http://localhost:8443/health | jq . > /dev/null 2>&1; then
+        echo "✅ Bark API is running!"
+        curl -s http://localhost:8443/health | jq .
+    else
+        echo "⚠️  API may still be starting up..."
+        echo "Check status with: sudo systemctl status bark-tts"
+        echo "View logs with: sudo journalctl -u bark-tts -f"
+    fi
+    
+    echo ""
+    echo "Bark TTS installation complete on t-800!"
+ENDSSH
+
+echo ""
+echo "=========================================="
+echo "  Deployment Complete!"
+echo "=========================================="
+echo ""
+echo "Next steps:"
+echo "1. Add BARK_API_URL to your .env file:"
+echo "   BARK_API_URL=http://t-800:8443"
+echo ""
+echo "2. Test the API:"
+echo "   curl 'http://t-800:8443/api/generate?text=Hello%20World&voice=v2/en_speaker_1' -o test.wav"
+echo ""
+echo "3. Generate audio for questions in OpenCCB:"
+echo "   POST /question-bank/{id}/generate-audio"
+echo ""