|
|
import PIL |
|
|
from PIL import ImageDraw |
|
|
from PIL import Image |
|
|
import streamlit as st |
|
|
import os |
|
|
import easyocr |
|
|
|
|
|
def load_image(image_file): |
|
|
img = PIL.Image.open(image_file) |
|
|
return img |
|
|
|
|
|
def init_session_states(): |
|
|
if 'disp' not in st.session_state: |
|
|
st.session_state['disp'] = st.empty() |
|
|
st.session_state['disp'].text("กำลังตั้งค่า Environment...") |
|
|
if 'init' not in st.session_state: |
|
|
st.session_state['init'] = 1 |
|
|
|
|
|
init_session_states() |
|
|
|
|
|
def draw_bounding_boxes(image, results): |
|
|
"""วาด bounding boxes บนภาพ""" |
|
|
draw = ImageDraw.Draw(image) |
|
|
|
|
|
for (bbox, text, confidence) in results: |
|
|
|
|
|
top_left = tuple(bbox[0]) |
|
|
bottom_right = tuple(bbox[2]) |
|
|
|
|
|
|
|
|
draw.rectangle([top_left, bottom_right], outline="red", width=3) |
|
|
|
|
|
|
|
|
draw.text(top_left, f"{text} ({confidence:.2f})", fill="blue") |
|
|
|
|
|
return image |
|
|
|
|
|
def main(): |
|
|
st.session_state['disp'].text("Environment พร้อมใช้งานแล้ว!") |
|
|
|
|
|
st.title("📝 Image Text Detection") |
|
|
st.write("ใช้ EasyOCR สำหรับตรวจจับและจดจำข้อความจากภาพ") |
|
|
|
|
|
|
|
|
st.sidebar.subheader("🌐 เลือกภาษา") |
|
|
languages = { |
|
|
"English": "en", |
|
|
"Thai": "th", |
|
|
"Chinese Simplified": "ch_sim", |
|
|
"Japanese": "ja", |
|
|
"Korean": "ko", |
|
|
"French": "fr", |
|
|
"German": "de" |
|
|
} |
|
|
|
|
|
selected_langs = st.sidebar.multiselect( |
|
|
"เลือกภาษาที่ต้องการตรวจจับ:", |
|
|
options=list(languages.keys()), |
|
|
default=["English", "Thai"] |
|
|
) |
|
|
|
|
|
|
|
|
lang_codes = [languages[lang] for lang in selected_langs] |
|
|
|
|
|
uploaded_file = st.file_uploader("เลือกไฟล์ภาพ", type=['jpeg','jpg','png']) |
|
|
|
|
|
if uploaded_file is not None: |
|
|
|
|
|
file_details = { |
|
|
"ชื่อไฟล์": uploaded_file.name, |
|
|
"ประเภท": uploaded_file.type, |
|
|
"ขนาด": f"{uploaded_file.size / 1024:.2f} KB", |
|
|
"ภาษาที่ตรวจจับ": ", ".join(selected_langs) |
|
|
} |
|
|
st.write(file_details) |
|
|
|
|
|
|
|
|
image = load_image(uploaded_file) |
|
|
st.image(image, width=500, caption="ภาพต้นฉบับ") |
|
|
|
|
|
|
|
|
with st.spinner(f"กำลังตรวจจับข้อความใน {len(selected_langs)} ภาษา..."): |
|
|
reader = easyocr.Reader(lang_codes, gpu=False) |
|
|
results = reader.readtext(image) |
|
|
|
|
|
st.success("ตรวจจับข้อความสำเร็จ!") |
|
|
|
|
|
|
|
|
st.subheader("🔍 ผลลัพธ์การตรวจจับ") |
|
|
|
|
|
|
|
|
if results: |
|
|
annotated_image = draw_bounding_boxes(image.copy(), results) |
|
|
st.image(annotated_image, width=500, caption="ภาพที่มีการวาด Bounding Boxes") |
|
|
|
|
|
|
|
|
st.subheader("📄 ข้อความที่ตรวจจับได้") |
|
|
|
|
|
for i, (bbox, text, confidence) in enumerate(results): |
|
|
st.write(f"**ข้อความ {i+1}:**") |
|
|
st.write(f"- ข้อความ: `{text}`") |
|
|
st.write(f"- ความมั่นใจ: `{confidence:.2%}`") |
|
|
st.write(f"- ตำแหน่ง: {bbox}") |
|
|
st.write("---") |
|
|
|
|
|
|
|
|
all_text = " ".join([text for (_, text, _) in results]) |
|
|
st.subheader("📝 ข้อความรวม") |
|
|
st.text_area("ข้อความที่ตรวจจับได้ทั้งหมด:", all_text, height=100) |
|
|
|
|
|
else: |
|
|
st.warning("❌ ไม่พบข้อความในภาพนี้") |
|
|
|
|
|
if __name__ == "__main__": |
|
|
main() |
