app.py

import gradio as gr
import requests
import json
import base64


def get_latex_from_image_all_formats(image_path):
    # 파일 경로에서 이미지 파일을 바이트 형식으로 읽음
    with open(image_path, "rb") as image_file:
        image_bytes = image_file.read()
    
    # 이미지 파일을 base64 인코딩
    image_base64 = base64.b64encode(image_bytes).decode('utf-8')

    # Mathpix API 요청 헤더
    headers = {
        "app_id": "arxivgpt_2c0986",  # 실제 사용하는 app_id로 대체
        "app_key": "b5c14c78ea645a6d673195e6360a1cc33ef2bab7a79b90f7cebf6465177171f5",  # 실제 사용하는 app_key로 대체
        "Content-Type": "application/json"
    }

    # 모든 포맷을 포함하는 데이터
    data = {
        "src": f"data:image/jpeg;base64,{image_base64}",
        "formats": ["text", "latex_styled", "latex_normal", "latex_list", "latex_simplified", "asciimath", "mathml"]
    }

    # Mathpix API 요청 보내기
    response = requests.post("https://api.mathpix.com/v3/latex", headers=headers, json=data)
    
    # 요청에 실패한 경우 예외 처리
    try:
        response.raise_for_status()
    except requests.HTTPError as e:
        return {"error": f"Mathpix API 요청 실패: {e}"}

    # 각 포맷에 따른 결과를 저장하는 딕셔너리
    result = response.json()
    results_by_format = {format_: result.get(format_, "추출 실패") for format_ in data["formats"]}

    return results_by_format

# Gradio 앱 정의
def build_gradio_app():
    with gr.Blocks() as app:
        with gr.Row():
            image_input = gr.Image(type="filepath", label="이미지 업로드")
            submit_button = gr.Button("변환하기")
        outputs = {format_: gr.Textbox(label=f"{format_} 결과") for format_ in ["text", "latex_styled", "latex_normal", "latex_list", "latex_simplified", "asciimath", "mathml"]}

        # 변환 버튼을 눌렀을 때 처리
        submit_button.click(fn=get_latex_from_image_all_formats, inputs=image_input, outputs=outputs)

    return app

# Gradio 앱 실행
if __name__ == "__main__":
    app = build_gradio_app()
    app.launch()