Simplify Docker setup and dependencies for minimal API version

Dockerfile

@@ -3,14 +3,7 @@ FROM python:3.10-slim
 # Install system dependencies
 RUN apt-get update && \
     apt-get install -y --no-install-recommends \
-    build-essential \
     curl \
-    libgl1-mesa-glx \
-    libglib2.0-0 \
-    wget \
-    git \
-    libopenblas-dev \
-    libgomp1 \
     && rm -rf /var/lib/apt/lists/*
 
 # Set working directory
@@ -20,20 +13,11 @@ WORKDIR /app
 ENV PYTHONUNBUFFERED=1 \
     PYTHONDONTWRITEBYTECODE=1 \
     UPLOAD_FOLDER=/tmp/pdf_uploads \
-    OUTPUT_FOLDER=/tmp/pdf_output \
-    HF_HUB_DISABLE_PROGRESS_BARS=1 \
-    HF_HUB_ENABLE_HF_TRANSFER=0
+    OUTPUT_FOLDER=/tmp/pdf_output
 
 # Create necessary directories and set permissions
 RUN mkdir -p /tmp/pdf_uploads /tmp/pdf_output \
-    && mkdir -p /tmp/models/MFD/YOLO \
-    && mkdir -p /tmp/models/MFR/unimernet \
-    && mkdir -p /tmp/models/table/rapid \
-    && mkdir -p /tmp/models/layout/doclayout \
-    && chmod -R 777 /tmp/pdf_uploads /tmp/pdf_output /tmp/models
-
-# Create magic-pdf config directories
-RUN mkdir -p /root/.config/magic_pdf
+    && chmod -R 777 /tmp/pdf_uploads /tmp/pdf_output
 
 # Copy requirements and app files
 COPY requirements.txt /app/
@@ -42,23 +26,8 @@ COPY entrypoint.sh /app/
 RUN chmod +x /app/entrypoint.sh
 
 # Install Python dependencies
-RUN pip install --no-cache-dir --upgrade pip setuptools wheel
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy default magic-pdf config
-RUN echo '{"device-mode":"cpu","layout-config":{"model":"doclayout_yolo","enable":true},"formula-config":{"mfd_model":"yolo_v8_mfd","mfr_model":"unimernet_small","enable":false},"table-config":{"model":"rapid_table","sub_model":"slanet_plus","enable":false}}' > /root/.config/magic_pdf/magic-pdf.json
-
-# Download model files directly
-RUN curl -L https://huggingface.co/marcosremar2/mineru-models/resolve/main/doclayout_yolo.pt -o /tmp/models/layout/doclayout/doclayout_yolo.pt
-RUN curl -L https://huggingface.co/marcosremar2/mineru-models/resolve/main/yolo_v8_ft.pt -o /tmp/models/MFD/YOLO/yolo_v8_ft.pt
-
-# Verify model files were downloaded correctly
-RUN ls -la /tmp/models/layout/doclayout/doclayout_yolo.pt
-RUN ls -la /tmp/models/MFD/YOLO/yolo_v8_ft.pt
-
-# Check if magic-pdf exists but don't require it to work yet
-RUN pip show minerupdf || echo "MinerU PDF will be installed at runtime"
-
 # Expose port
 EXPOSE 7860
 

entrypoint.sh

@@ -5,115 +5,10 @@ set -e
 echo "Starting entrypoint.sh..."
 echo "User: $(whoami)"
 echo "Current directory: $(pwd)"
-echo "Directory listing:"
-ls -la
 
-# Set environment variables to disable HF progress bars (prevents hanging)
-export HF_HUB_DISABLE_PROGRESS_BARS=1
-export HF_HUB_ENABLE_HF_TRANSFER=0
-
-# Verify directories existence and permissions
-echo "Checking directories..."
-for dir in /tmp/pdf_uploads /tmp/pdf_output /tmp/models/MFD/YOLO /tmp/models/MFR/unimernet /tmp/models/table/rapid /tmp/models/layout/doclayout
-do
-    if [ ! -d "$dir" ]; then
-        echo "Creating directory: $dir"
-        mkdir -p "$dir"
-    fi
-    chmod -R 777 "$dir"
-    echo "Directory $dir is ready"
-done
-
-# Copy config file to all possible locations to ensure it's found
-echo "Setting up magic-pdf config..."
-CONFIG_CONTENT='{
-  "device-mode": "cpu",
-  "layout-config": {
-    "model": "doclayout_yolo",
-    "enable": true
-  },
-  "formula-config": {
-    "mfd_model": "yolo_v8_mfd",
-    "mfr_model": "unimernet_small",
-    "enable": false
-  },
-  "table-config": {
-    "model": "rapid_table",
-    "sub_model": "slanet_plus",
-    "enable": false
-  }
-}'
-
-# Create config in all possible locations
-mkdir -p ~/.config/magic_pdf
-echo "$CONFIG_CONTENT" > ~/.config/magic_pdf/magic-pdf.json
-echo "$CONFIG_CONTENT" > ~/magic-pdf.json
-echo "$CONFIG_CONTENT" > /app/magic-pdf.json
-echo "$CONFIG_CONTENT" > /home/user/magic-pdf.json
-echo "$CONFIG_CONTENT" > /root/.config/magic_pdf/magic-pdf.json
-
-# Download model function with validation
-function download_model_with_validation() {
-    local model_path=$1
-    local model_url=$2
-    local max_attempts=3
-    local attempt=1
-    
-    echo "Checking model file: $model_path"
-    
-    if [ ! -f "$model_path" ] || [ ! -s "$model_path" ]; then
-        while [ $attempt -le $max_attempts ]; do
-            echo "Downloading model attempt $attempt/$max_attempts: $model_url"
-            curl -L --retry 5 --retry-delay 2 "$model_url" -o "$model_path"
-            
-            # Verify file exists and has content
-            if [ -f "$model_path" ] && [ -s "$model_path" ]; then
-                size=$(du -h "$model_path" | cut -f1)
-                echo "✅ Model downloaded successfully ($size): $model_path"
-                return 0
-            else
-                echo "❌ Download failed or file is empty. Retrying..."
-                rm -f "$model_path" 2>/dev/null
-                attempt=$((attempt+1))
-                sleep 2
-            fi
-        done
-        
-        echo "❌ Failed to download model after $max_attempts attempts: $model_url"
-        exit 1
-    else
-        size=$(du -h "$model_path" | cut -f1)
-        echo "✅ Model already exists ($size): $model_path"
-    fi
-}
-
-# Download and verify all required models
-echo "Verifying model files..."
-download_model_with_validation "/tmp/models/layout/doclayout/doclayout_yolo.pt" "https://huggingface.co/marcosremar2/mineru-models/resolve/main/doclayout_yolo.pt"
-download_model_with_validation "/tmp/models/MFD/YOLO/yolo_v8_ft.pt" "https://huggingface.co/marcosremar2/mineru-models/resolve/main/yolo_v8_ft.pt"
-
-# List all model files for verification
-echo "Model files verification:"
-find /tmp/models -type f -exec ls -la {} \;
-
-# Verify magic-pdf exists and is executable
-echo "Checking magic-pdf installation..."
-if command -v magic-pdf &> /dev/null; then
-    echo "magic-pdf found. Testing version:"
-    magic-pdf --version || echo "Warning: magic-pdf command exists but might not be functioning properly."
-else
-    echo "magic-pdf command not found. Installing..."
-    pip install --no-cache-dir minerupdf==1.3.6
-    if command -v magic-pdf &> /dev/null; then
-        echo "magic-pdf installed successfully. Testing version:"
-        magic-pdf --version || echo "Warning: magic-pdf command was installed but might not be functioning properly."
-    else
-        echo "Warning: Failed to install magic-pdf."
-        echo "PATH: $PATH"
-        pip list | grep miner
-        # Continue anyway
-    fi
-fi
+# Create necessary directories
+mkdir -p /tmp/pdf_uploads /tmp/pdf_output
+chmod -R 777 /tmp/pdf_uploads /tmp/pdf_output
 
 # Start the Flask application
 echo "Starting Flask application..."

requirements.txt

@@ -1,12 +1,4 @@
 flask==2.3.3
 flask-cors==4.0.0
 werkzeug==2.3.7
-Pillow>=9.0.0
-numpy>=1.20.0
-requests>=2.25.0
-opencv-python-headless>=4.5.0
-torch>=1.8.0
-torchvision>=0.9.0
-transformers>=4.15.0
-huggingface_hub>=0.11.0
-minerupdf==1.3.6 
+requests>=2.25.0