Spaces:
Running
Running
| from PIL import Image | |
| import streamlit | |
| from simple_latex_ocr.models import Latex_OCR | |
| # import execjs | |
| # js_code = """ | |
| # function encodeToBase64(latex) { | |
| # var str = encodeURI(latex); // URL | |
| # str = btoa(str); // Base64 | |
| # return str; | |
| # } | |
| # """ | |
| # ctx = execjs.compile(js_code) | |
| model = Latex_OCR() | |
| if __name__ == '__main__': | |
| streamlit.set_page_config(page_title='Simple-LaTeX-OCR') | |
| streamlit.title('Simple-LaTeX-OCR') | |
| streamlit.markdown( | |
| 'Convert images of equations to corresponding LaTeX code.\n\nThis is based on the `Simple-LaTeX-OCR` module. Check it out (https://github.com/chaodreaming/Simple-LaTeX-OCR)') | |
| uploaded_file = streamlit.file_uploader( | |
| 'Upload an image an equation', | |
| type=['png', 'jpg'], | |
| ) | |
| if uploaded_file is not None: | |
| image = Image.open(uploaded_file) | |
| streamlit.image(image) | |
| else: | |
| streamlit.text('\n') | |
| if streamlit.button('Convert'): | |
| if uploaded_file is not None and image is not None: | |
| with streamlit.spinner('Computing'): | |
| response = model.predict(uploaded_file.getvalue()) | |
| if "formula" in response: | |
| latex_code = response["formula"] | |
| # latex_encode = ctx.call("encodeToBase64", latex_code) | |
| streamlit.code(latex_code, language='latex') | |
| streamlit.markdown(f'$\\displaystyle {latex_code}$') | |
| # link_html = f'<a href="https://www.latexlive.com/#{latex_encode}" target="_blank"><button style="color: black;">Online Editing</button></a>' | |
| # streamlit.markdown(link_html, unsafe_allow_html=True) | |
| else: | |
| streamlit.error(response) | |
| else: | |
| streamlit.error('Please upload an image.') | |