image2text.py

import os
import sys
from PyQt5.QtWidgets import QApplication, QMainWindow, QLabel, QTextEdit, QPushButton, QFileDialog, QVBoxLayout, QHBoxLayout, QWidget
from PyQt5.QtGui import QPixmap
from PyQt5.QtCore import Qt
from PIL import Image
import pytesseract

# MainWindow class that sets up the GUI


class MainWindow(QMainWindow):
    def __init__(self):
        super().__init__()

        # Set the window title and geometry
        self.setWindowTitle("Image2Text")
        self.setGeometry(100, 100, 600, 500)
        self.setMinimumSize(300, 250)

        # Create header label and buttons for opening and converting
        self.label_header = QLabel("Image2Text", self)
        self.label_header.setAlignment(Qt.AlignCenter)
        font = self.label_header.font()
        font.setPointSize(20)
        self.label_header.setFont(font)
        self.open_button = QPushButton("Open Image", self)
        self.open_button.clicked.connect(self.open_image)
        self.convert_button = QPushButton("Convert", self)
        self.convert_button.clicked.connect(self.convert)

        # Create image label and text box
        self.label_image = QLabel(self)
        self.label_image.setAlignment(Qt.AlignCenter)
        self.text_box = QTextEdit(self)

        # Create horizontal layout for header
        layout_header = QHBoxLayout()
        layout_header.addWidget(self.label_header)

        # Create horizontal layout for open button
        layout_open = QHBoxLayout()
        layout_open.addWidget(self.open_button)

        # Create vertical layout for image label
        layout_image = QVBoxLayout()
        layout_image.addWidget(self.label_image)

        # Create horizontal layout for convert button
        layout_convert = QHBoxLayout()
        layout_convert.addWidget(self.convert_button)

        # Create main vertical layout and add other layouts and widgets
        layout_main = QVBoxLayout()
        layout_main.addLayout(layout_header)
        layout_main.addLayout(layout_open)
        layout_main.addLayout(layout_image)
        layout_main.addLayout(layout_convert)
        layout_main.addWidget(self.text_box)

        # Create widget and set main layout
        widget = QWidget(self)
        widget.setLayout(layout_main)
        self.setCentralWidget(widget)

        # Set the path to Tesseract
        pytesseract.pytesseract.tesseract_cmd = os.path.join(
            os.getcwd(), 'Tesseract-OCR/tesseract')

    # Method to open image file
    def open_image(self):
        # Open a file dialog to select an image
        self.file = QFileDialog.getOpenFileName(
            self, "Open Image", "", "Images (*.png *.xpm *.jpg *.bmp *.gif);;All Files (*)")[0]

        # If a file is selected
        if self.file:
            # Set the file path as the previous file
            self.previous = self.file

            # If the file path is longer than 30 characters
            if len(self.file) >= 30:
                # Truncate the file path and set it as the text on the open button
                self.open_button.setText(
                    self.file[:30] + "..." + self.file[-3:])
            else:
                # Set the file path as the text on the open button
                self.open_button.setText(self.file)

            # Open the image file using PIL
            self.image = Image.open(self.file)
            # Create a QPixmap from the image file
            pixmap = QPixmap(self.file)
            # Set the pixmap on the image label
            self.label_image.setPixmap(pixmap)
        else:
            # If a file was previously selected, set it as the current file
            if self.previous != "":
                self.file = self.previous

    def convert(self):
        # If no file has been selected
        if self.file == "":
            return
        try:
            # Use pytesseract to extract text from the image
            result = pytesseract.image_to_string(self.image)
        except Exception as e:
            # Print the error message if there is an exception
            print("Error:", e)
            return

        # Set the extracted text on the text box
        self.text_box.setText(result)


# Create a QApplication
app = QApplication(sys.argv)
# Create the MainWindow instance
window = MainWindow()
# Show the window
window.show()
# Exit the application when the window is closed
sys.exit(app.exec_())