Spaces:
Running
Running
File size: 1,770 Bytes
d8e0bc8 e457e0e 5dee6f4 d8e0bc8 5dee6f4 d8e0bc8 5dee6f4 d8e0bc8 e457e0e d8e0bc8 e457e0e d8e0bc8 5dee6f4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
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.')
|