import matplotlib.pyplot as plt
import gradio as gr
import tempfile
import shutil

def render_latex(latex_code):
    # 임시 파일 생성
    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix='.png')
    temp_file_name = temp_file.name
    temp_file.close()
    
    # LaTeX 코드를 matplotlib을 사용하여 이미지로 렌더링
    plt.rc('text', usetex=True)
    plt.rc('font', family='serif')
    plt.axis('off')
    plt.text(0.5, 0.5, f'${latex_code}$', ha='center', va='center', fontsize=20)
    plt.savefig(temp_file_name, bbox_inches='tight', pad_inches=0.1)
    plt.close()
    
    # 임시 이미지 파일 경로 반환
    return temp_file_name

# Gradio 인터페이스 정의
iface = gr.Interface(fn=render_latex, 
                     inputs=gr.Textbox(placeholder="Enter LaTeX code here..."), 
                     outputs=gr.Image(type='filepath'),  # 수정된 부분
                     title="LaTeX Renderer",
                     description="Enter a LaTeX code to render it into an image.")

# Gradio 인터페이스 실행
iface.launch()