import requests
import json
import base64

# Mathpix에 이미지 파일을 보내어 LaTeX 문자열을 추출하는 함수
def get_latex_from_image(image_path):
    # 이미지 파일을 base64 인코딩
    with open(image_path, "rb") as f:
        image_base64 = base64.b64encode(f.read()).decode()

    # Mathpix API 요청 헤더
    headers = {
        "app_id": "arxivgpt_2c0986",  # 여기에 Mathpix ID를 입력
        "app_key": "b5c14c78ea645a6d673195e6360a1cc33ef2bab7a79b90f7cebf6465177171f5",  # 여기에 Mathpix API 키를 입력
        "Content-Type": "application/json"
    }

    # Mathpix API 요청 바디
    data = {
        "src": "data:image/png;base64," + image_base64,
        "formats": ["latex_normal"]
    }

    # Mathpix API 요청 보내기
    response = requests.post("https://api.mathpix.com/v3/latex", headers=headers, data=json.dumps(data))
    response.raise_for_status()  # 요청 실패 시 예외 발생

    # 응답에서 LaTeX 추출
    result = response.json()
    latex = result['latex_normal']

    return latex

# 추출한 LaTeX 출력
latex_string = get_latex_from_image("path/to/image.png")
print(latex_string)