Spaces:
Sleeping
Sleeping
File size: 2,987 Bytes
05abb4e 6179695 919e7d0 2dc4c21 919e7d0 2dc4c21 05abb4e 6179695 05abb4e 6179695 919e7d0 6179695 05abb4e 919e7d0 05abb4e 057e923 919e7d0 6179695 919e7d0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 |
#!/bin/bash
# Exit on error
set -e
echo "Setting up Tesseract OCR environment..."
# Install required packages if not already installed
if ! command -v tesseract &> /dev/null; then
echo "Tesseract not found, attempting to install..."
apt-get update -y || echo "Failed to update apt, continuing anyway"
apt-get install -y tesseract-ocr tesseract-ocr-eng libtesseract-dev libleptonica-dev || echo "Failed to install tesseract via apt, continuing anyway"
fi
# Install Python dependencies
echo "Installing Python dependencies..."
pip install -q -U pytesseract pillow opencv-python-headless pdf2image
pip install -q -U google-genai
echo "Python dependencies installed successfully"
# Install tesserocr with pip
echo "Installing tesserocr..."
pip install -q -U tesserocr || echo "Failed to install tesserocr with pip, trying with specific compiler flags..."
# If tesserocr installation failed, try with specific compiler flags
if ! python -c "import tesserocr" &> /dev/null; then
echo "Trying alternative tesserocr installation..."
CPPFLAGS="-I/usr/local/include -I/usr/include" LDFLAGS="-L/usr/local/lib -L/usr/lib" pip install -q -U tesserocr || echo "Failed to install tesserocr with compiler flags, continuing anyway"
fi
# Create tessdata directory if it doesn't exist
mkdir -p tessdata
# Set TESSDATA_PREFIX environment variable
export TESSDATA_PREFIX="$(pwd)/tessdata"
echo "TESSDATA_PREFIX set to: $TESSDATA_PREFIX"
# Download eng.traineddata if it doesn't exist
if [ ! -f "tessdata/eng.traineddata" ]; then
echo "Downloading eng.traineddata..."
wget -O tessdata/eng.traineddata https://github.com/tesseract-ocr/tessdata/raw/main/eng.traineddata || \
curl -o tessdata/eng.traineddata https://github.com/tesseract-ocr/tessdata/raw/main/eng.traineddata
echo "Downloaded eng.traineddata"
else
echo "eng.traineddata already exists"
fi
# Try to copy to system locations (may fail in restricted environments)
for tessdata_dir in "/usr/share/tesseract-ocr/4.00/tessdata" "/usr/share/tesseract-ocr/tessdata" "/usr/local/share/tessdata"; do
if [ -d "$tessdata_dir" ]; then
echo "Copying eng.traineddata to $tessdata_dir..."
cp -f tessdata/eng.traineddata "$tessdata_dir/" 2>/dev/null || echo "Failed to copy to $tessdata_dir, continuing anyway"
fi
done
# Verify Tesseract installation
echo "Verifying Tesseract installation..."
tesseract --version || echo "Tesseract not found in PATH, but may still be available to Python"
# Test tesserocr if installed
echo "Testing tesserocr..."
python -c "import tesserocr; print(f'tesserocr version: {tesserocr.tesseract_version()}')" || echo "tesserocr not working, but may still be able to use pytesseract"
# Test pytesseract
echo "Testing pytesseract..."
python -c "import pytesseract; print(f'pytesseract path: {pytesseract.tesseract_cmd}')" || echo "pytesseract not working"
echo "Setup completed"
# Add TESSDATA_PREFIX to .env file for persistence
echo "TESSDATA_PREFIX=$(pwd)/tessdata" >> .env |