Subindo arquivos3

app.py

@@ -1,22 +1,40 @@
+import gradio as gr
 import pytesseract
 from PIL import Image
-import gradio as gr
+import easyocr
+
+#
+easyocr_reader = easyocr.Reader(['en', 'pt', 'es'], gpu=False)  # 
 
-# Definindo a função de OCR
-def extract_text_from_image(image):
-    text = pytesseract.image_to_string(image)
-    return text
+#Tesseract
+def tesseract_ocr(image):
+    return pytesseract.image_to_string(image)
 
-# Criando a interface do Gradio
+#EasyOCR
+def easyocr_ocr(image):
+    return ' '.join(easyocr_reader.readtext(image, detail=0))
+
+#
+def extract_text_from_image(ocr_engine, image):
+    if ocr_engine == "Tesseract":
+        return tesseract_ocr(image)
+    elif ocr_engine == "EasyOCR":
+        return easyocr_ocr(image)
+    else:
+        return "Invalid OCR selection."
+
+#Gradio
 iface = gr.Interface(
     fn=extract_text_from_image,
-    inputs="image",
+    inputs=[gr.Dropdown(["Tesseract", "EasyOCR"], label="Select the OCR Engine"), "image"],
     outputs="text",
-    title="OCR-img2txt",
-    description="This application uses Optical Character Recognition (OCR) technology to extract text from images. It employs the Tesseract OCR engine to process images containing printed text, converting it into editable and searchable text. Ideal for digitizing documents, extracting text from photos, and converting scanned images into text formats."
+    title="OCR Img2txt",
+    description="This application uses Optical Character Recognition (OCR) technology to extract text from images. Choose between Tesseract OCR and EasyOCR engine to process images containing printed text, converting it into editable and searchable text."
+    examples=[["confusionmatrix.jpg"]]
 )
 
-# Executando a interface
+#
 iface.launch(debug=True)
 
 
+