From ab8a442b22f6036036ea78e88cd4b9886db44476 Mon Sep 17 00:00:00 2001
From: Amir Khan <amir.khan@salesforce.com>
Date: Tue, 6 Aug 2024 23:11:34 +0200
Subject: [PATCH] redeploying PDFBox (langchain4J)

redeploying PDFBox (langchain4J)
---
 pom.xml                                       |  7 ++-
 .../LangchainImageModelsOperations.java       | 43 ++++++++++---------
 2 files changed, 28 insertions(+), 22 deletions(-)

diff --git a/pom.xml b/pom.xml
index 0fc8bf8..3ad1aec 100644
--- a/pom.xml
+++ b/pom.xml
@@ -5,7 +5,7 @@
 	<modelVersion>4.0.0</modelVersion>
 	<groupId>com.mule.mulechain</groupId>
 	<artifactId>mulechain-ai-connector</artifactId>
-	<version>0.2.12</version>
+	<version>0.2.13</version>
 	<packaging>mule-extension</packaging>
 	<name>MuleChain</name>
 
@@ -169,6 +169,11 @@
 			<version>2.2.21</version>
 		</dependency>
 		<!--<dependency>
+			<groupId>dev.langchain4j</groupId>
+			<artifactId>langchain4j-document-parser-apache-pdfbox</artifactId>
+			<version>0.31.0</version>
+		</dependency>
+		<dependency>
 			<groupId>org.apache.pdfbox</groupId>
 			<artifactId>pdfbox</artifactId>
 			<version>3.0.2</version>
diff --git a/src/main/java/org/mule/extension/mulechain/internal/operation/LangchainImageModelsOperations.java b/src/main/java/org/mule/extension/mulechain/internal/operation/LangchainImageModelsOperations.java
index 77cc94b..e771c53 100644
--- a/src/main/java/org/mule/extension/mulechain/internal/operation/LangchainImageModelsOperations.java
+++ b/src/main/java/org/mule/extension/mulechain/internal/operation/LangchainImageModelsOperations.java
@@ -29,15 +29,13 @@
 
 import java.awt.image.BufferedImage;
 import java.io.ByteArrayOutputStream;
-import java.io.File;
+import java.io.FileInputStream;
 import java.io.IOException;
 import java.util.Base64;
-import java.util.List;
 
 import javax.imageio.ImageIO;
-
+import java.io.InputStream;
 import org.apache.pdfbox.pdmodel.PDDocument;
-import org.apache.pdfbox.pdmodel.PDPage;
 import org.apache.pdfbox.rendering.PDFRenderer;
 
 
@@ -95,54 +93,57 @@ public String drawImage(@Config LangchainLLMConfiguration configuration, String
    * Reads an scanned document.
    */
 
-  /*
+
   @MediaType(value = ANY, strict = false)
   @Alias("IMAGE-read-scanned-documents")
   public String readScannedDocumentPDF(@Config LangchainLLMConfiguration configuration, String data, String filePath) {
-  
+
     ChatLanguageModel model = configuration.getModel();
-  
+
     String sourceDir = filePath;
-  
+
     JSONObject jsonObject = new JSONObject();
     JSONArray docPages = new JSONArray();
-    
-    try (PDDocument document = Loader.loadPDF(new File(sourceDir))) {
-  
+
+    //try (PDDocument document = Loader.loadPDF(new File(sourceDir))) {
+    try (InputStream inputStream = new FileInputStream(sourceDir);
+        PDDocument document = PDDocument.load(inputStream);) {
+
+
       PDFRenderer pdfRenderer = new PDFRenderer(document);
       int totalPages = document.getNumberOfPages();
       LOGGER.info("Total files to be converted -> " + totalPages);
       jsonObject.put(MuleChainConstants.TOTAL_PAGES, totalPages);
-  
+
       JSONObject docPage;
       for (int pageNumber = 0; pageNumber < totalPages; pageNumber++) {
-  
+
         BufferedImage image = pdfRenderer.renderImageWithDPI(pageNumber, 300);
         LOGGER.info("Reading page -> " + pageNumber);
-  
+
         String imageBase64 = convertToBase64String(image);
         UserMessage userMessage = UserMessage.from(
                                                    TextContent.from(data),
                                                    ImageContent.from(imageBase64, "image/png"));
-  
+
         Response<AiMessage> response = model.generate(userMessage);
-  
+
         docPage = new JSONObject();
         docPage.put(MuleChainConstants.PAGE, pageNumber + 1);
         docPage.put(MuleChainConstants.RESPONSE, response.content().text());
         docPage.put(MuleChainConstants.TOKEN_USAGE, JsonUtils.getTokenUsage(response));
         docPages.put(docPage);
       }
-  
+
     } catch (IOException e) {
       LOGGER.info("Error occurred while processing the file: " + e.getMessage());
     }
-  
+
     jsonObject.put(MuleChainConstants.PAGES, docPages);
-  
+
     return jsonObject.toString();
   }
-  
+
   private String convertToBase64String(BufferedImage image) {
     String base64String;
     try (ByteArrayOutputStream outputStream = new ByteArrayOutputStream()) {
@@ -155,5 +156,5 @@ private String convertToBase64String(BufferedImage image) {
       LOGGER.info("Error occurred while processing the file: " + e.getMessage());
       return "Error";
     }
-  } */
+  }
 }