Update app.py

app.py

@@ -1,31 +1,36 @@
-import gradio as gr
 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 요청을 위한 함수
-def process_image_to_latex(file_path):
+    # Mathpix API 요청 헤더
     headers = {
-        "app_id": "arxivgpt_2c0986",
-        "app_key": "b5c14c78ea645a6d673195e6360a1cc33ef2bab7a79b90f7cebf6465177171f5",
-        "Content-type": "application/json"
+        "app_id": "arxivgpt_2c0986",  # 여기에 Mathpix ID를 입력
+        "app_key": "b5c14c78ea645a6d673195e6360a1cc33ef2bab7a79b90f7cebf6465177171f5",  # 여기에 Mathpix API 키를 입력
+        "Content-Type": "application/json"
     }
-    with open(file_path, "rb") as file:
-        img_data = file.read()
-    response = requests.post(
-        "https://api.mathpix.com/v3/latex",
-        headers=headers,
-        json={"src": "data:image/jpeg;base64," + base64.b64encode(img_data).decode()}
-    )
-    return response.json()["latex"]
 
-# Gradio 인터페이스 구성
-def create_gradio_interface():
-    with gr.Blocks() as demo:
-        gr.Markdown("## LaTeX 추출기")
-        file_input = gr.File(label="이미지 또는 PDF 파일 업로드")
-        latex_output = gr.Textbox(label="추출된 LaTeX")
-        file_input.change(fn=process_image_to_latex, inputs=[file_input], outputs=[latex_output])
-    demo.launch()
+    # 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
 
-create_gradio_interface()
+# 추출한 LaTeX 출력
+latex_string = get_latex_from_image("path/to/image.png")
+print(latex_string)