app.py

import gradio as gr
import requests

# Mathpix 요청을 위한 함수
def process_image_to_latex(file_path):
    headers = {
        "app_id": "arxivgpt_2c0986",
        "app_key": "b5c14c78ea645a6d673195e6360a1cc33ef2bab7a79b90f7cebf6465177171f5",
        "Content-type": "application/json"
    }
    with open(file_path, "rb") as file:
        img_data = file.read()
    response = requests.post(
        "https://api.mathpix.com/v3/latex",
        headers=headers,
        json={"src": "data:image/jpeg;base64," + base64.b64encode(img_data).decode()}
    )
    return response.json()["latex"]

# Gradio 인터페이스 구성
def create_gradio_interface():
    with gr.Blocks() as demo:
        gr.Markdown("## LaTeX 추출기")
        file_input = gr.File(label="이미지 또는 PDF 파일 업로드")
        latex_output = gr.Textbox(label="추출된 LaTeX")
        file_input.change(fn=process_image_to_latex, inputs=[file_input], outputs=[latex_output])
    demo.launch()

create_gradio_interface()