added ocr processing

Dockerfile

@@ -1,11 +1,34 @@
+# Use a Python 3.9 base image
 FROM python:3.9
+
+# Create a user and set environment variables
 RUN useradd -m -u 1000 user
 USER user
 ENV HOME=/home/user \
     PATH=/home/user/.local/bin:$PATH
+
+# Set the working directory
 WORKDIR $HOME/app
+
+# Copy your application code
 COPY --chown=user . .
-COPY ./requirements.txt ~/app/requirements.txt
+
+# Copy requirements.txt
+COPY ./requirements.txt $HOME/app/requirements.txt
+
+# Install Tesseract and any necessary dependencies
+USER root
+RUN apt-get update && apt-get install -y tesseract-ocr poppler-utils
+
+
+# Switch back to the user
+USER user
+
+# Install Python dependencies
 RUN pip install -r requirements.txt
-RUN mkdir -p files && chown -R user:user files
+
+# Create a directory for files and set permissions
+RUN mkdir -p $HOME/app/files && chown -R user:user $HOME/app/files
+
+# Specify the command to run your application
 CMD ["chainlit", "run", "app.py", "--port", "7860"]

app.py

@@ -3,6 +3,11 @@ import chainlit as cl
 from io import BytesIO
 import re
 import os
+from pdf2image import convert_from_path
+from PIL import Image
+import pytesseract
+
+
 
 def save_pdf_file(file_name, page):
     pdf_writer = PyPDF2.PdfWriter()
@@ -52,12 +57,28 @@ def process_dhl_text(pdf_text):
     return file_name, file_path
 
 def process_ups_pdf(file_name, page):
-    # page.cropbox.upper_left = (0,0)
-    # page.cropbox.lower_right = (500,400)
     page.rotate(90)
-    pdf_text = page.extract_text()
-    print(pdf_text)
     save_pdf_file(file_name, page)
+    file_path = f"files/{file_name}"
+    
+    images = convert_from_path(file_path)
+    image = images[0]
+    image.save(f"{file_path}.png", "PNG")
+    loaded_image = Image.open(f"{file_path}.png")
+    extracted_text = pytesseract.image_to_string(loaded_image)
+    image.close()
+    loaded_image.close()
+    os.remove(f"{file_path}.png")
+    match = re.search(r'TRACKING #:\s+([A-Z\d\s]+)', extracted_text)
+    if match:
+        extracted_text = match.group(1)
+        extracted_text = extracted_text.replace(" ", "").replace("BILLING", "").replace("\n", "")
+        file_name = extracted_text + ".pdf"
+    else:
+        print("Pattern not found in the text.")
+
+    return file_name, file_path
+
 
 def process_coppel_pdf(file_name, page):
     page.cropbox.upper_left = (0,150)
@@ -111,7 +132,7 @@ def process_pdf_file(file):
         process_coppel_pdf(file_name, page)
     else:
         transport_company = "ups"
-        process_ups_pdf(file_name, page)
+        file_name, file_path = process_ups_pdf(file_name, page)
 
     pdf_stream.close()
     return file_name, file_path, transport_company

requirements.txt

@@ -1,2 +1,78 @@
-chainlit
-PyPDF2
+aiofiles==23.2.1
+aiohttp==3.8.6
+aiosignal==1.3.1
+annotated-types==0.6.0
+anyio==3.7.1
+async-timeout==4.0.3
+asyncer==0.0.2
+attrs==23.1.0
+backoff==2.2.1
+bidict==0.22.1
+certifi==2023.7.22
+chainlit==0.7.301
+charset-normalizer==3.3.1
+click==8.1.7
+dataclasses-json==0.5.14
+Deprecated==1.2.14
+exceptiongroup==1.1.3
+fastapi==0.100.1
+fastapi-socketio==0.0.10
+filetype==1.2.0
+frozenlist==1.4.0
+googleapis-common-protos==1.61.0
+grpcio==1.59.0
+h11==0.14.0
+httpcore==0.18.0
+httpx==0.25.0
+idna==3.4
+importlib-metadata==6.8.0
+Jinja2==3.1.2
+Lazify==0.4.0
+MarkupSafe==2.1.3
+marshmallow==3.20.1
+multidict==6.0.4
+mypy-extensions==1.0.0
+nest-asyncio==1.5.8
+nodeenv==1.8.0
+opentelemetry-api==1.20.0
+opentelemetry-exporter-otlp==1.20.0
+opentelemetry-exporter-otlp-proto-common==1.20.0
+opentelemetry-exporter-otlp-proto-grpc==1.20.0
+opentelemetry-exporter-otlp-proto-http==1.20.0
+opentelemetry-instrumentation==0.41b0
+opentelemetry-proto==1.20.0
+opentelemetry-sdk==1.20.0
+opentelemetry-semantic-conventions==0.41b0
+packaging==23.2
+pdf2image==1.16.3
+Pillow==10.1.0
+prisma==0.10.0
+protobuf==4.24.4
+pydantic==2.4.2
+pydantic_core==2.10.1
+PyJWT==2.8.0
+PyPDF2==3.0.1
+pytesseract==0.3.10
+python-dotenv==1.0.0
+python-engineio==4.8.0
+python-graphql-client==0.4.3
+python-multipart==0.0.6
+python-socketio==5.10.0
+requests==2.31.0
+simple-websocket==1.0.0
+sniffio==1.3.0
+starlette==0.27.0
+syncer==2.0.3
+tomli==2.0.1
+tomlkit==0.12.1
+typing-inspect==0.9.0
+typing_extensions==4.8.0
+uptrace==1.20.2
+urllib3==2.0.7
+uvicorn==0.23.2
+watchfiles==0.20.0
+websockets==12.0
+wrapt==1.15.0
+wsproto==1.2.0
+yarl==1.9.2
+zipp==3.17.0