import gradio as gr
from PIL import Image
from pix2tex.cli import LatexOCR

# Receives an image and returns the latex code and the markdown code
def img_latex(image):
    img = Image.open(image)
    model = LatexOCR()
    latex = model(img)
    markdown = latex
    markdown = r'$$' + latex + r'$$'
    return latex, markdown


gr.Interface(
    img_latex,
    inputs=gr.inputs.Image(label="Upload", type="filepath"),
    outputs=[gr.outputs.Textbox(label="Latex"),gr.outputs.Textbox(label="Markdown")],
    title="Latex OCR",
).launch()