fix: add Tesseract configuration and startup script

README.md

@@ -1,4 +1,3 @@
-
 ---
 title: Doc2Md
 emoji: 📄
@@ -8,6 +7,7 @@ sdk: gradio
 sdk_version: 5.14.0
 app_file: app.py
 build_script: build.sh
+startup_script: setup.sh
 pinned: false
 ---
 

app.py

@@ -1,6 +1,13 @@
 import sys
 import os
 
+# Set TESSDATA_PREFIX if not already set
+if not os.environ.get('TESSDATA_PREFIX'):
+    tessdata_dir = "/usr/share/tesseract-ocr/4.00/tessdata"
+    if os.path.exists(tessdata_dir):
+        os.environ['TESSDATA_PREFIX'] = tessdata_dir
+        print(f"Set TESSDATA_PREFIX to {tessdata_dir}")
+
 # Get the current directory
 current_dir = os.path.dirname(os.path.abspath(__file__))
 

setup.sh

@@ -0,0 +1,35 @@
+#!/bin/bash
+
+# Exit on error
+set -e
+
+# Create tessdata directory if it doesn't exist
+TESSDATA_DIR="/usr/share/tesseract-ocr/4.00/tessdata"
+mkdir -p "$TESSDATA_DIR"
+
+# Download traineddata files if they don't exist
+if [ ! -f "$TESSDATA_DIR/eng.traineddata" ]; then
+    echo "Downloading eng.traineddata..."
+    wget -O "$TESSDATA_DIR/eng.traineddata" "https://github.com/tesseract-ocr/tessdata/raw/main/eng.traineddata"
+fi
+
+if [ ! -f "$TESSDATA_DIR/osd.traineddata" ]; then
+    echo "Downloading osd.traineddata..."
+    wget -O "$TESSDATA_DIR/osd.traineddata" "https://github.com/tesseract-ocr/tessdata/raw/main/osd.traineddata"
+fi
+
+# Set TESSDATA_PREFIX
+export TESSDATA_PREFIX="$TESSDATA_DIR"
+echo "TESSDATA_PREFIX=${TESSDATA_PREFIX}" >> /etc/environment
+
+# Test Tesseract functionality
+echo "Testing Tesseract..."
+echo "Hello World" > test.png
+tesseract test.png stdout
+rm test.png
+
+# Print Tesseract version and available languages
+echo "Tesseract version:"
+tesseract --version
+echo "Available languages:"
+tesseract --list-langs