Update app.py

app.py

@@ -3,23 +3,18 @@ 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에 이미지 파일을 보내어 여러 포맷의 LaTeX 문자열을 추출하는 함수
+def get_latex_from_image_all_formats(image):
+    image_base64 = base64.b64encode(image).decode('utf-8')
 
     # Mathpix API 요청 헤더
     headers = {
-        "app_id": "arxivgpt_2c0986",  # 실제 사용하는 app_id로 대체
-        "app_key": "b5c14c78ea645a6d673195e6360a1cc33ef2bab7a79b90f7cebf6465177171f5",  # 실제 사용하는 app_key로 대체
+        "app_id": "arxivgpt_2c0986",
+        "app_key": "b5c14c78ea645a6d673195e6360a1cc33ef2bab7a79b90f7cebf6465177171f5",
         "Content-Type": "application/json"
     }
 
-    # 모든 포맷을 포함하는 데이터
+    # Mathpix API 요청 바디
     data = {
         "src": f"data:image/jpeg;base64,{image_base64}",
         "formats": ["text", "latex_styled", "latex_normal", "latex_list", "latex_simplified", "asciimath", "mathml"]
@@ -27,29 +22,23 @@ def get_latex_from_image_all_formats(image_path):
 
     # 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}"}
-
-    # 각 포맷에 따른 결과를 저장하는 딕셔너리
+    response.raise_for_status()  # 요청 실패 시 예외 발생
+
+    # 응답에서 각 포맷의 LaTeX 추출
     result = response.json()
-    results_by_format = {format_: result.get(format_, "추출 실패") for format_ in data["formats"]}
+    formats_results = {f: result.get(f, "추출 실패") for f in data["formats"]}
 
-    return results_by_format
+    return formats_results
 
 # Gradio 앱 정의
 def build_gradio_app():
     with gr.Blocks() as app:
         with gr.Row():
-            image_input = gr.Image(type="filepath", label="이미지 업로드")
+            image_input = gr.Image(type="file", 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"]}
+        outputs = {f: gr.Textbox(label=f"{f} 결과") for f 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)
+        submit_button.click(fn=get_latex_from_image_all_formats, inputs=[image_input], outputs=outputs)
 
     return app
 
@@ -57,4 +46,3 @@ def build_gradio_app():
 if __name__ == "__main__":
     app = build_gradio_app()
     app.launch()
-