Update app.py

app.py

@@ -1,25 +1,29 @@
 import gradio as gr
 import requests
-import urllib.parse
 
-def latex_to_image_via_api(latex_code):
-    encoded_latex = urllib.parse.quote(latex_code)  # LaTeX 코드를 URL 인코딩
-    api_url = f"https://latex.codecogs.com/png.download?{encoded_latex}"  # URL 생성
-    
-    response = requests.get(api_url)  # API 요청
+# Mathpix 요청을 위한 함수
+def process_image_to_latex(file_path):
+    headers = {
+        "app_id": "arxivgpt_2c0986",
+        "app_key": "b5c14c78ea645a6d673195e6360a1cc33ef2bab7a79b90f7cebf6465177171f5",
+        "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"]
 
-    if response.status_code == 200:
-        # 응답으로부터 바이너리 데이터를 받아 임시 파일에 저장
-        with open("latex_image.png", "wb") as f:
-            f.write(response.content)  # 이미지 내용을 파일에 쓴다
-        return "latex_image.png"  # 이미지 파일 경로 반환
-    else:
-        return "LaTeX 변환 실패: API 호출에 실패했습니다."
+# 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()
 
-iface = gr.Interface(fn=latex_to_image_via_api,
-                     inputs=gr.TextArea(placeholder="LaTeX 코드를 입력하세요..."),                     
-                     outputs=gr.Image(label="변환된 이미지"),
-                     title="LaTeX 코드를 이미지로 변환",
-                     description="Codecogs LaTeX Rendering API를 사용하여 LaTeX 코드를 이미지로 변환합니다.")
-
-iface.launch()
+create_gradio_interface()