add 2nd gen GCF OCR sample

functions/v2/ocr/ocr-process-image/pom.xml

@@ -0,0 +1,141 @@
+<?xml version="1.0" encoding="UTF-8"?>
+
+<!--
+  Copyright 2022 Google LLC
+
+  Licensed under the Apache License, Version 2.0 (the "License");
+  you may not use this file except in compliance with the License.
+  You may obtain a copy of the License at
+
+  http://www.apache.org/licenses/LICENSE-2.0
+
+  Unless required by applicable law or agreed to in writing, software
+  distributed under the License is distributed on an "AS IS" BASIS,
+  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+  See the License for the specific language governing permissions and
+  limitations under the License.
+-->
+
+<project xmlns="http://maven.apache.org/POM/4.0.0"
+    xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+    xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+  <modelVersion>4.0.0</modelVersion>
+
+  <groupId>com.example.cloud.functions</groupId>
+  <artifactId>functions-ocr-process-image</artifactId>
+
+  <parent>
+    <groupId>com.google.cloud.samples</groupId>
+    <artifactId>shared-configuration</artifactId>
+    <version>1.2.0</version>
+  </parent>
+
+  <properties>
+    <maven.compiler.target>11</maven.compiler.target>
+    <maven.compiler.source>11</maven.compiler.source>
+    <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
+  </properties>
+
+  <dependencyManagement>
+    <dependencies>
+      <dependency>
+        <groupId>com.google.cloud</groupId>
+        <artifactId>libraries-bom</artifactId>
+        <version>25.4.0</version>
+        <type>pom</type>
+        <scope>import</scope>
+      </dependency>
+    </dependencies>
+  </dependencyManagement>
+
+  <dependencies>
+    <dependency>
+      <groupId>com.google.cloud.functions</groupId>
+      <artifactId>functions-framework-api</artifactId>
+      <version>1.0.4</version>
+      <scope>provided</scope>
+    </dependency>
+    <dependency>
+      <groupId>io.cloudevents</groupId>
+      <artifactId>cloudevents-core</artifactId>
+      <version>2.3.0</version>
+    </dependency>
+    <dependency>
+      <groupId>org.projectlombok</groupId>
+      <artifactId>lombok</artifactId>
+      <version>1.18.24</version>
+    </dependency>
+    <dependency>
+      <groupId>com.google.cloud</groupId>
+      <artifactId>google-cloud-vision</artifactId>
+    </dependency>
+    <dependency>
+      <groupId>com.google.cloud</groupId>
+      <artifactId>google-cloud-translate</artifactId>
+    </dependency>
+    <dependency>
+      <groupId>com.google.cloud</groupId>
+      <artifactId>google-cloud-pubsub</artifactId>
+    </dependency>
+    <dependency>
+      <groupId>com.google.code.gson</groupId>
+      <artifactId>gson</artifactId>
+      <version>2.9.1</version>
+   </dependency>
+
+
+    <!-- The following dependencies are only required for testing -->
+    <dependency>
+      <groupId>junit</groupId>
+      <artifactId>junit</artifactId>
+      <version>4.13.2</version>
+      <scope>test</scope>
+    </dependency>
+    <dependency>
+      <groupId>com.google.truth</groupId>
+      <artifactId>truth</artifactId>
+      <version>1.1.3</version>
+      <scope>test</scope>
+    </dependency>
+    <dependency>
+      <groupId>com.google.guava</groupId>
+      <artifactId>guava-testlib</artifactId>
+      <version>31.1-jre</version>
+      <scope>test</scope>
+    </dependency>
+  </dependencies>
+
+  <build>
+    <plugins>
+      <plugin>
+        <!--
+          Google Cloud Functions Framework Maven plugin
+
+          This plugin allows you to run Cloud Functions Java code
+          locally. Use the following terminal command to run a
+          given function locally:
+
+          mvn function:run -Drun.functionTarget=your.package.yourFunction
+        -->
+        <groupId>com.google.cloud.functions</groupId>
+        <artifactId>function-maven-plugin</artifactId>
+        <version>0.10.0</version>
+        <configuration>
+          <functionTarget>
+            functions.OcrProcessImage
+          </functionTarget>
+        </configuration>
+      </plugin>
+      <plugin>
+        <groupId>org.apache.maven.plugins</groupId>
+        <artifactId>maven-surefire-plugin</artifactId>
+        <version>3.0.0-M7</version>
+        <configuration>
+          <skipTests>${skipTests}</skipTests>
+          <reportNameSuffix>sponge_log</reportNameSuffix>
+          <trimStackTrace>false</trimStackTrace>
+        </configuration>
+      </plugin>
+    </plugins>
+  </build>
+</project>

functions/v2/ocr/ocr-process-image/src/main/java/functions/OcrProcessImage.java

@@ -0,0 +1,189 @@
+/*
+ * Copyright 2022 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package functions;
+
+// [START functions_ocr_process]
+
+import com.google.cloud.functions.CloudEventsFunction;
+import com.google.cloud.pubsub.v1.Publisher;
+import com.google.cloud.translate.v3.DetectLanguageRequest;
+import com.google.cloud.translate.v3.DetectLanguageResponse;
+import com.google.cloud.translate.v3.LocationName;
+import com.google.cloud.translate.v3.TranslationServiceClient;
+import com.google.cloud.vision.v1.AnnotateImageRequest;
+import com.google.cloud.vision.v1.AnnotateImageResponse;
+import com.google.cloud.vision.v1.Feature;
+import com.google.cloud.vision.v1.Image;
+import com.google.cloud.vision.v1.ImageAnnotatorClient;
+import com.google.cloud.vision.v1.ImageSource;
+import com.google.gson.Gson;
+import com.google.gson.GsonBuilder;
+import com.google.gson.JsonDeserializationContext;
+import com.google.gson.JsonDeserializer;
+import com.google.gson.JsonElement;
+import com.google.gson.JsonParseException;
+import com.google.protobuf.ByteString;
+import com.google.pubsub.v1.ProjectTopicName;
+import com.google.pubsub.v1.PubsubMessage;
+import functions.eventpojos.StorageObjectData;
+import io.cloudevents.CloudEvent;
+import java.io.IOException;
+import java.lang.reflect.Type;
+import java.nio.charset.StandardCharsets;
+import java.time.OffsetDateTime;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.concurrent.ExecutionException;
+import java.util.logging.Level;
+import java.util.logging.Logger;
+
+// [END functions_ocr_process]
+
+// [START functions_ocr_setup]
+public class OcrProcessImage implements CloudEventsFunction {
+  // TODO<developer> set these environment variables
+  private static final String PROJECT_ID = System.getenv("GCP_PROJECT");
+  private static final String TRANSLATE_TOPIC_NAME = System.getenv("TRANSLATE_TOPIC");
+  private static final String[] TO_LANGS = System.getenv("TO_LANG").split(",");
+
+  private static final Logger logger = Logger.getLogger(OcrProcessImage.class.getName());
+  private static final String LOCATION_NAME = LocationName.of(PROJECT_ID, "global").toString();
+  private Publisher publisher;
+
+  public OcrProcessImage() throws IOException {
+    publisher = Publisher.newBuilder(ProjectTopicName.of(PROJECT_ID, TRANSLATE_TOPIC_NAME)).build();
+  }
+
+  // Create custom deserializer to handle timestamps in event data
+  class DateDeserializer implements JsonDeserializer<OffsetDateTime> {
+    @Override
+    public OffsetDateTime deserialize(
+        JsonElement json, Type typeOfT, JsonDeserializationContext context)
+        throws JsonParseException {
+      return OffsetDateTime.parse(json.getAsString());
+    }
+  }
+
+  Gson gson =
+      new GsonBuilder().registerTypeAdapter(OffsetDateTime.class, new DateDeserializer()).create();
+  // [END functions_ocr_setup]
+
+  // [START functions_ocr_process]
+  @Override
+  public void accept(CloudEvent event) {
+    // Unmarshal data from CloudEvent
+    StorageObjectData gcsEvent =
+        gson.fromJson(
+            new String(event.getData().toBytes(), StandardCharsets.UTF_8), StorageObjectData.class);
+    String bucket = gcsEvent.getBucket();
+    if (bucket == null) {
+      throw new IllegalArgumentException("Missing bucket parameter");
+    }
+    String filename = gcsEvent.getName();
+    if (filename == null) {
+      throw new IllegalArgumentException("Missing name parameter");
+    }
+
+    detectText(bucket, filename);
+  }
+  // [END functions_ocr_process]
+
+  // [START functions_ocr_detect]
+  private void detectText(String bucket, String filename) {
+    logger.info("Looking for text in image " + filename);
+
+    List<AnnotateImageRequest> visionRequests = new ArrayList<>();
+    String gcsPath = String.format("gs://%s/%s", bucket, filename);
+
+    ImageSource imgSource = ImageSource.newBuilder().setGcsImageUri(gcsPath).build();
+    Image img = Image.newBuilder().setSource(imgSource).build();
+
+    Feature textFeature = Feature.newBuilder().setType(Feature.Type.TEXT_DETECTION).build();
+    AnnotateImageRequest visionRequest =
+        AnnotateImageRequest.newBuilder().addFeatures(textFeature).setImage(img).build();
+    visionRequests.add(visionRequest);
+
+    // Detect text in an image using the Cloud Vision API
+    AnnotateImageResponse visionResponse;
+    try (ImageAnnotatorClient client = ImageAnnotatorClient.create()) {
+      visionResponse = client.batchAnnotateImages(visionRequests).getResponses(0);
+      if (visionResponse == null || !visionResponse.hasFullTextAnnotation()) {
+        logger.info(String.format("Image %s contains no text", filename));
+        return;
+      }
+
+      if (visionResponse.hasError()) {
+        // Log error
+        logger.log(
+            Level.SEVERE, "Error in vision API call: " + visionResponse.getError().getMessage());
+        return;
+      }
+    } catch (IOException e) {
+      // Log error (since IOException cannot be thrown by a Cloud Function)
+      logger.log(Level.SEVERE, "Error detecting text: " + e.getMessage(), e);
+      return;
+    }
+
+    String text = visionResponse.getFullTextAnnotation().getText();
+    logger.info("Extracted text from image: " + text);
+
+    // Detect language using the Cloud Translation API
+    DetectLanguageRequest languageRequest =
+        DetectLanguageRequest.newBuilder()
+            .setParent(LOCATION_NAME)
+            .setMimeType("text/plain")
+            .setContent(text)
+            .build();
+    DetectLanguageResponse languageResponse;
+    try (TranslationServiceClient client = TranslationServiceClient.create()) {
+      languageResponse = client.detectLanguage(languageRequest);
+    } catch (IOException e) {
+      // Log error (since IOException cannot be thrown by a function)
+      logger.log(Level.SEVERE, "Error detecting language: " + e.getMessage(), e);
+      return;
+    }
+
+    if (languageResponse.getLanguagesCount() == 0) {
+      logger.info("No languages were detected for text: " + text);
+      return;
+    }
+
+    String languageCode = languageResponse.getLanguages(0).getLanguageCode();
+    logger.info(String.format("Detected language %s for file %s", languageCode, filename));
+
+    // Send a Pub/Sub translation request for every language we're going to translate to
+    for (String targetLanguage : TO_LANGS) {
+      logger.info("Sending translation request for language " + targetLanguage);
+      OcrTranslateApiMessage message = new OcrTranslateApiMessage(text, filename, targetLanguage);
+      ByteString byteStr = ByteString.copyFrom(message.toPubsubData());
+      PubsubMessage pubsubApiMessage = PubsubMessage.newBuilder().setData(byteStr).build();
+      try {
+        publisher.publish(pubsubApiMessage).get();
+      } catch (InterruptedException | ExecutionException e) {
+        // Log error
+        logger.log(Level.SEVERE, "Error publishing translation request: " + e.getMessage(), e);
+        return;
+      }
+    }
+  }
+  // [END functions_ocr_detect]
+
+  // [START functions_ocr_process]
+  // [START functions_ocr_setup]
+}
+// [END functions_ocr_setup]
+// [END functions_ocr_process]