Satyam0077's picture
Initial commit with app.py and requirements.txt
948ea45
raw
history blame contribute delete
901 Bytes
import easyocr
import gradio as gr
# Initialize EasyOCR reader
reader = easyocr.Reader(['en', 'hi'])
# Function for OCR and search functionality
def process_image(image, keyword):
result = reader.readtext(image, detail=0)
extracted_text = " ".join(result)
# Check if the keyword is in the text
if keyword.lower() in extracted_text.lower():
return f"Keyword '{keyword}' found in the text.", extracted_text
else:
return f"Keyword '{keyword}' not found.", extracted_text
# Gradio interface
interface = gr.Interface(fn=process_image,
inputs=["image", "text"],
outputs=["text", "text"],
title="OCR and Document Search",
description="Upload an image, extract text, and search for keywords")
# Launch the app
if __name__ == "__main__":
interface.launch()