ML Translation

pom.xml

@@ -60,6 +60,12 @@ limitations under the License.
       <artifactId>google-cloud-datastore</artifactId>
       <version>1.52.0</version>
     </dependency>
+
+    <dependency>
+      <groupId>com.google.cloud</groupId>
+      <artifactId>google-cloud-translate</artifactId>
+      <version>1.55.0</version>
+    </dependency>
 
   <dependency>
 	  <groupId>com.google.cloud</groupId>

src/main/java/com/google/codeu/data/Message.java

@@ -54,6 +54,11 @@ public String getUser() {
     return user;
   }
 
+  public void setText(String words) {
+	  this.text = words;
+	  return;
+  }
+
   public String getText() {
     return text;
   }

src/main/java/com/google/codeu/servlets/MessageServlet.java

@@ -29,6 +29,10 @@
 import javax.servlet.http.HttpServletResponse;
 import org.jsoup.Jsoup;
 import org.jsoup.safety.Whitelist;
+import com.google.cloud.translate.Translate;
+import com.google.cloud.translate.Translate.TranslateOption;
+import com.google.cloud.translate.TranslateOptions;
+import com.google.cloud.translate.Translation;
 
 import com.google.cloud.language.v1.Document;
 import com.google.cloud.language.v1.Document.Type;
@@ -55,6 +59,20 @@ private float getSentimentScore(String text) throws IOException {
 
     return sentiment.getScore();
   }
+
+  private void translateMessages(List<Message> messages, String targetLanguageCode) {
+	  Translate translate = TranslateOptions.getDefaultInstance().getService();
+
+	  for(Message message : messages) {
+	    String originalText = message.getText();
+
+	    Translation translation =
+	        translate.translate(originalText, TranslateOption.targetLanguage(targetLanguageCode));
+	    String translatedText = translation.getTranslatedText();
+
+	    message.setText(translatedText);
+	  }    
+	}
 
 
 
@@ -74,13 +92,19 @@ public void doGet(HttpServletRequest request, HttpServletResponse response) thro
 
     String user = request.getParameter("user");
 
-    if (user == null || user.equals("")) {
+  if (user == null || user.equals("")) {
       // Request is invalid, return empty array
       response.getWriter().println("[]");
       return;
-    }
+    }  
 
     List<Message> messages = datastore.getMessages(user);
+
+    String targetLanguageCode = request.getParameter("language");
+
+    if(targetLanguageCode != null) {
+      translateMessages(messages, targetLanguageCode);
+    }
     Gson gson = new Gson();
     String json = gson.toJson(messages);
 

src/main/webapp/js/user-page-loader.js

@@ -49,6 +49,11 @@ function showMessageFormIfViewingSelf() {
 /** Fetches messages and add them to the page. */
 function fetchMessages() {
   const url = '/messages?user=' + parameterUsername;
+  const parameterLanguage = urlParams.get('language');
+  let url = '/messages?user=' + parameterUsername;
+  if(parameterLanguage) {
+    url += '&language=' + parameterLanguage;
+  }
   fetch(url)
       .then((response) => {
         return response.json();
@@ -60,6 +65,7 @@ function fetchMessages() {
         } else {
           messagesContainer.innerHTML = '';
         }
+
         messages.forEach((message) => {
           const messageDiv = buildMessageDiv(message);
           messagesContainer.appendChild(messageDiv);
@@ -91,10 +97,51 @@ function buildMessageDiv(message) {
 
   return messageDiv;
 }
+/**
+ * Creates an li element.
+ * @param {Element} childElement
+ * @return {Element} li element
+ */
+function createListItem(childElement) {
+  const listItemElement = document.createElement('li');
+  listItemElement.appendChild(childElement);
+  return listItemElement;
+}
+
+/**
+ * Creates an anchor element.
+ * @param {string} url
+ * @param {string} text
+ * @return {Element} Anchor element
+ */
+function createLink(url, text) {
+  const linkElement = document.createElement('a');
+  linkElement.appendChild(document.createTextNode(text));
+  linkElement.href = url;
+  return linkElement;
+}
+
+/** Adds list of languages for translating messages */
+function buildLanguageLinks(){
+  const userPageUrl = '/user-page.html?user=' + parameterUsername;
+  const languagesListElement  = document.getElementById('languages');
+  languagesListElement.appendChild(createListItem(createLink(
+       userPageUrl + '&language=en', 'English')));
+  languagesListElement.appendChild(createListItem(createLink(
+      userPageUrl + '&language=zh', 'Chinese')));
+  languagesListElement.appendChild(createListItem(createLink(
+      userPageUrl + '&language=hi', 'Hindi')));
+  languagesListElement.appendChild(createListItem(createLink(
+      userPageUrl + '&language=es', 'Spanish')));
+  languagesListElement.appendChild(createListItem(createLink(
+      userPageUrl + '&language=ar', 'Arabic')));
+}
 
 /** Fetches data and populates the UI of the page. */
 function buildUI() {
   setPageTitle();
+  buildLanguageLinks();
   showMessageFormIfViewingSelf();
   fetchMessages();
-}
+}
+

src/main/webapp/user-page.html

@@ -30,7 +30,7 @@
       </ul>
     </nav>
     <h1 id="page-title">User Page</h1>
-
+	<ul id="languages"></ul>
     <form id="message-form" action="/messages" method="POST" class="hidden">
       Enter a new message:
       <br/>