Spaces:
Runtime error
Runtime error
File size: 2,604 Bytes
23dfaea 9d2915e 5a045fe 9d2915e d6b7429 9d2915e d6b7429 9d2915e d6b7429 9d2915e d6b7429 5a045fe d6b7429 9d2915e d6b7429 9d2915e d6b7429 9d2915e |
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 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 |
import easyocr as ocr
import streamlit as st
from PIL import Image
import numpy as np
import base64
from st_clickable_images import clickable_images
if "language" not in st.session_state:
st.session_state.language=['ch_tra', 'en']
if "output_text" not in st.session_state:
st.session_state.output_text=""
if "image" not in st.session_state:
st.session_state.image=""
def extract_text():
language=st.session_state.language
st.session_state.image=st.session_state.uploaded_file
if st.session_state.image:
st.session_state.output_text=rjxai_extract(
language,
st.session_state.image
)
else:
st.session_state.image=""
st.session_state.output_text=""
def rjxai_extract(
language,
uploaded_file,
):
input_image=Image.open(uploaded_file)
reader=ocr.Reader(language, model_storage_directory='.')
result=reader.readtext(np.array(input_image))
result_text = []
for text in result:
result_text.append(text[1])
return ", ".join(result_text)
# side
with st.sidebar:
st.header("示例")
images = []
with open("assets/example/1.jpg", "rb") as image:
encoded=base64.b64encode(image.read()).decode()
images.append(f"data:image/jpeg;base64,{encoded}")
clicked=clickable_images(
images,
titles=[f"Image #{str(i)}" for i in range(len(images))],
div_style={"display": "flex", "justify-content": "center", "flex-wrap": "wrap"},
img_style={"margin": "5px", "height": "200px"},
key="clicked"
)
if st.session_state.clicked==0:
st.session_state.image="assets/example/1.jpg"
st.session_state.output_text=rjxai_extract(
st.session_state.language,
"assets/example/1.jpg"
)
# content
with st.container():
st.header("Rjxai v4")
st.subheader("Extract Text from Images")
# 参考：https://github.com/JaidedAI/EasyOCR/issues/989
# st.multiselect(
# 'Select extract language',
# ['en', 'ch_sim', 'ch_tra', 'ja', 'es', 'fr', 'fa'],
# key="language"
# )
uploaded_file=st.file_uploader(
label = "Upload your image here: png, jpg, jpeg",
type=['png','jpg','jpeg'],
on_change=extract_text,
key="uploaded_file",
)
if st.session_state.image:
input_image=Image.open(st.session_state.image)
st.image(input_image)
if st.session_state.output_text:
st.text_area(
"extract text:",
key="output_text",
disabled=True
) |