Update app.py

app.py

@@ -3,8 +3,8 @@ import requests
 import json
 import base64
 
-# Mathpix에 이미지 파일을 보내어 LaTeX 문자열을 추출하는 함수
-def get_latex_from_image(image_path):
+
+def get_latex_from_image_all_formats(image_path):
     # 파일 경로에서 이미지 파일을 바이트 형식으로 읽음
     with open(image_path, "rb") as image_file:
         image_bytes = image_file.read()
@@ -14,26 +14,31 @@ def get_latex_from_image(image_path):
 
     # Mathpix API 요청 헤더
     headers = {
-        "app_id": "arxivgpt_2c0986",
-        "app_key": "b5c14c78ea645a6d673195e6360a1cc33ef2bab7a79b90f7cebf6465177171f5",
+        "app_id": "arxivgpt_2c0986",  # 실제 사용하는 app_id로 대체
+        "app_key": "b5c14c78ea645a6d673195e6360a1cc33ef2bab7a79b90f7cebf6465177171f5",  # 실제 사용하는 app_key로 대체
         "Content-Type": "application/json"
     }
 
-    # Mathpix API 요청 바디
+    # 모든 포맷을 포함하는 데이터
     data = {
         "src": f"data:image/jpeg;base64,{image_base64}",
-        "formats": ["latex_normal"]
+        "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)
-    response.raise_for_status()  # 요청 실패 시 예외 발생
+    
+    # 요청에 실패한 경우 예외 처리
+    try:
+        response.raise_for_status()
+    except requests.HTTPError as e:
+        return {"error": f"Mathpix API 요청 실패: {e}"}
 
-    # 응답에서 LaTeX 추출
+    # 각 포맷에 따른 결과를 저장하는 딕셔너리
     result = response.json()
-    latex = result.get('latex_normal', "LaTeX 추출 실패")
+    results_by_format = {format_: result.get(format_, "추출 실패") for format_ in data["formats"]}
 
-    return latex
+    return results_by_format
 
 # Gradio 앱 정의
 def build_gradio_app():
@@ -41,9 +46,10 @@ def build_gradio_app():
         with gr.Row():
             image_input = gr.Image(type="filepath", label="이미지 업로드")
             submit_button = gr.Button("변환하기")
-        latex_output = gr.Textbox(label="추출된 LaTeX")
+        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, inputs=image_input, outputs=latex_output)
+        # 변환 버튼을 눌렀을 때 처리
+        submit_button.click(fn=get_latex_from_image_all_formats, inputs=image_input, outputs=outputs)
 
     return app
 
@@ -51,3 +57,4 @@ def build_gradio_app():
 if __name__ == "__main__":
     app = build_gradio_app()
     app.launch()
+