#!/bin/bash

echo "========================================"
echo "eDOCr2 - Quick Start Script"
echo "========================================"
echo ""

# Check if virtual environment exists
if [ ! -d "venv" ]; then
    echo "Creating virtual environment..."
    python3 -m venv venv
    echo ""
fi

# Activate virtual environment
echo "Activating virtual environment..."
source venv/bin/activate
echo ""

# Install dependencies
echo "Installing dependencies..."
pip install -q --upgrade pip
pip install -q -r requirements.txt
echo ""

# Check if models exist
if [ ! -f "edocr2/models/recognizer_gdts.keras" ]; then
    echo ""
    echo "========================================"
    echo "WARNING: Model files not found!"
    echo "========================================"
    echo ""
    echo "Please download the model files from:"
    echo "https://github.com/javvi51/edocr2/releases/tag/v1.0.0"
    echo ""
    echo "Required files:"
    echo "- recognizer_gdts.keras"
    echo "- recognizer_gdts.txt"
    echo "- recognizer_dimensions_2.keras"
    echo "- recognizer_dimensions_2.txt"
    echo ""
    echo "Place them in: edocr2/models/"
    echo ""
    read -p "Press Enter to exit..."
    exit 1
fi

# Run the app
echo "========================================"
echo "Starting eDOCr2 Gradio App..."
echo "========================================"
echo ""
echo "Open your browser and go to:"
echo "http://localhost:7860"
echo ""
echo "Press Ctrl+C to stop the server"
echo ""

python app.py