Update app.py

app.py

@@ -2,41 +2,57 @@ import gradio as gr
 import requests
 import json
 import base64
+from PIL import Image
+import io
 
+# Mathpix에 이미지 파일을 보내어 여러 포맷의 LaTeX 문자열을 추출하는 함수
 def get_latex_from_image_all_formats(image):
-    image_base64 = base64.b64encode(image).decode('utf-8')
+    # PIL 이미지를 바이트로 변환
+    buffered = io.BytesIO()
+    image.save(buffered, format="JPEG")
+    image_base64 = base64.b64encode(buffered.getvalue()).decode('utf-8')
 
+    # Mathpix API 요청 헤더
     headers = {
         "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"]
     }
 
+    # Mathpix API 요청 보내기
     response = requests.post("https://api.mathpix.com/v3/latex", headers=headers, json=data)
-    response.raise_for_status()
+    response.raise_for_status()  # 요청 실패 시 예외 발생
 
+    # 응답에서 각 포맷의 LaTeX 추출
     result = response.json()
-    formats_results = {f: result.get(f, "추출 실패") for f in data["formats"]}
+    formats_results = {f: result.get(f, f"{f} 추출 실패") for f in data["formats"]}
 
     return formats_results
 
+# Gradio 앱 정의
 def build_gradio_app():
     with gr.Blocks() as app:
         with gr.Row():
-            image_input = gr.Image(type="filepath", label="이미지 업로드")  # 'file' 타입을 'filepath'로 변경
+            image_input = gr.Image(type="pil", label="이미지 업로드")
             submit_button = gr.Button("변환하기")
+        
         outputs = [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)
 
-    
-    
+        def process_and_output(image):
+            results = get_latex_from_image_all_formats(image)
+            return [results.get(f) for f in ["text", "latex_styled", "latex_normal", "latex_list", "latex_simplified", "asciimath", "mathml"]]
+        
+        submit_button.click(fn=process_and_output, inputs=image_input, outputs=outputs)
+
     return app
 
+# Gradio 앱 실행
 if __name__ == "__main__":
     app = build_gradio_app()
     app.launch()