backend done

chrome_extension/background.js

@@ -5,7 +5,7 @@ chrome.runtime.onInstalled.addListener(function() {
 	chrome.declarativeContent.onPageChanged.removeRules(undefined, function() {
 		chrome.declarativeContent.onPageChanged.addRules([{
 		  conditions: [new chrome.declarativeContent.PageStateMatcher({
-			pageUrl: {hostEquals: 'developer.chrome.com'},
+			pageUrl: {schemes: [ 'http', 'https']},
 		  })
 		  ],
 			  actions: [new chrome.declarativeContent.ShowPageAction()]

chrome_extension/popup.js

@@ -10,14 +10,15 @@ sendText.onclick = async function(element) {
 		  {code: "window.getSelection().toString()"}, async function(selection){
 			  // sel = selection[0];
 			  var xmlhttp = new XMLHttpRequest();   // new HttpRequest instance 
-			  var theUrl = "";
+			  var theUrl = "http://localhost:5000/get_prediction";
 			  xmlhttp.open("POST", theUrl);
 			  xmlhttp.setRequestHeader("Content-Type", "application/json;charset=UTF-8");
 			  xmlhttp.send(JSON.stringify({ "text": selection[0] }));
 			  xmlhttp.onreadystatechange = function() {
 				  if (xmlhttp.readyState == 4) {
 					// JSON.parse does not evaluate the attacker's scripts.
 					resp = JSON.parse(xmlhttp.responseText);
+					resp = resp['prediction']
 				  }
 			  }
 		  }

predictor.py

@@ -107,7 +107,7 @@ def create_embedding_matrix(field,embeddings):
 
 @app.route('/form',methods=['POST','GET'])
 def model_form():
-    return render_template('form.html')
+    return render_template('index.html')
 
 
 @app.route('/train',methods=['POST','GET'])