File size: 1,613 Bytes
81603f2 |
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 |
import easyocr
import gradio as gr
import re
# Initialize EasyOCR reader with English and Hindi languages
reader = easyocr.Reader(['en', 'hi'])
# Define the OCR function
def ocr_text(image):
"""
Performs OCR on the given image and returns the extracted text.
Args:
image: The image to perform OCR on.
Returns:
The extracted text and the original image.
"""
result = reader.readtext(image)
full_text = ' '.join([text for _, text, _ in result])
return full_text, image
# Define the text highlighting function
def highlight_text(full_text, search_term):
"""
Highlights the search term in the given text.
Args:
full_text: The text to search in.
search_term: The term to search for.
Returns:
The text with the search term highlighted.
"""
if search_term:
highlighted_text = re.sub(f"({search_term})", r"<mark>\1</mark>", full_text, flags=re.IGNORECASE)
return highlighted_text
return full_text
# Define the Gradio interface
with gr.Blocks() as demo:
image_input = gr.Image(type="filepath")
extracted_text = gr.Textbox(lines=5, placeholder="Extracted text will appear here...")
ocr_button = gr.Button("Extract Text")
ocr_button.click(fn=ocr_text, inputs=image_input, outputs=[extracted_text, image_input])
search_term = gr.Textbox(lines=1, placeholder="Enter search term")
highlighted_text = gr.HTML()
search_button = gr.Button("Search")
search_button.click(fn=highlight_text, inputs=[extracted_text, search_term], outputs=highlighted_text)
demo.launch(share=True)
|