Spaces:
Sleeping
Sleeping
| import requests | |
| from PIL import Image | |
| import streamlit as st | |
| from st_img_pastebutton import paste | |
| from io import BytesIO | |
| import base64 | |
| def encode_image(file): | |
| _, encoded = file.split(",", 1) | |
| binary_data = base64.b64decode(encoded) | |
| bytes_data = BytesIO(binary_data) | |
| return bytes_data | |
| if __name__ == "__main__": | |
| st.set_page_config(page_title="LaTeX-OCR") | |
| st.title("LaTeX OCR") | |
| st.markdown( | |
| "Convert images of equations to corresponding LaTeX code.\n\nThis is based on the `pix2tex` module. Check it out [](https://github.com/lukas-blecher/LaTeX-OCR)" | |
| ) | |
| source = st.radio( | |
| "Choose the source of the image", | |
| options=["Upload", "Paste"], | |
| ) | |
| image = None | |
| if source == "Upload": | |
| uploaded_file = st.file_uploader( | |
| "Upload an image of an equation", | |
| type=["png", "jpg"], | |
| ) | |
| if uploaded_file is not None: | |
| st.image(Image.open(uploaded_file)) | |
| image = uploaded_file.getvalue() | |
| if source == "Paste": | |
| pasted_file = paste("Paste an image of an equation") | |
| if pasted_file is not None: | |
| image = encode_image(pasted_file) | |
| st.image(image) | |
| if st.button("Convert"): | |
| if image is not None: | |
| with st.spinner("Computing"): | |
| response = requests.post( | |
| "http://127.0.0.1:8502/predict/", files={"file": image} | |
| ) | |
| if response.ok: | |
| latex_code = response.json() | |
| st.code(latex_code, language="latex") | |
| st.markdown(f"$\\displaystyle {latex_code}$") | |
| else: | |
| st.error(response.text) | |
| else: | |
| st.error("No image selected") | |