| import gradio as gr |
| import easyocr |
| from pdf2image import convert_from_path |
| import numpy as np |
| from PIL import Image |
| import os |
|
|
| |
| print("กำลังโหลดโมเดล EasyOCR...") |
| reader = easyocr.Reader(['th', 'en']) |
|
|
| def process_file(file_path): |
| if file_path is None: |
| return "กรุณาอัปโหลดไฟล์" |
| |
| |
| ext = os.path.splitext(file_path)[1].lower() |
| extracted_text = [] |
|
|
| try: |
| if ext == '.pdf': |
| |
| |
| pages = convert_from_path(file_path, dpi=200) |
| |
| for i, page in enumerate(pages): |
| extracted_text.append(f"--- หน้าที่ {i+1} ---") |
| |
| img_np = np.array(page) |
| result = reader.readtext(img_np, detail=0) |
| extracted_text.extend(result) |
| extracted_text.append("\n") |
| |
| elif ext in ['.png', '.jpg', '.jpeg']: |
| |
| result = reader.readtext(file_path, detail=0) |
| extracted_text.extend(result) |
| |
| else: |
| return "รองรับเฉพาะไฟล์ .pdf, .png, .jpg, .jpeg เท่านั้นครับ" |
|
|
| |
| return "\n".join(extracted_text) |
|
|
| except Exception as e: |
| return f"เกิดข้อผิดพลาดในการอ่านไฟล์: {str(e)}" |
|
|
| |
| interface = gr.Interface( |
| fn=process_file, |
| inputs=gr.File(label="อัปโหลดไฟล์รูปภาพ หรือ PDF"), |
| outputs=gr.Textbox(label="ข้อความที่สกัดได้", lines=15), |
| title="Thai OCR Web App (EasyOCR)", |
| description="อัปโหลดไฟล์รูปภาพ หรือไฟล์ PDF เพื่อดึงข้อความภาษาไทยและอังกฤษ", |
| theme="default" |
| ) |
|
|
| |
| if __name__ == "__main__": |
| interface.launch() |
