import easyocr
import gradio as gr
from PIL import Image

reader = easyocr.Reader(['en'])

def inference(img_path, width_ths):
    output = reader.readtext(img_path, detail=0, slope_ths=0.7, ycenter_ths=0.9,
                          height_ths=0.8, width_ths=width_ths, add_margin=0.2)

    output = "\n".join(output)

title = "Receipt RAG"
description = "A simple Gradio interface to query receipts using RAG"
examples = [["data/receipt_00000.JPG", 7.7],
            ["data/receipr_00001.jpg", 7.7]]

demo = gr.Interface(inference, 
                    inputs = [gr.Image(width=320, height=320, label="Input Receipt"), 
                              gr.Slider(0, 10, 7.7, 0.1, label="Width Threshold to merge bounding boxes")],
                    outputs= [gr.Textbox(label="OCR Output", type="text")],
                    title=title,
                    description=description,
                    examples=examples)

demo.launch()