post_endpoint.patch

��diff --git a/detector/server.py b/detector/server.py
index 34a0c85..8c190e4 100644
--- a/detector/server.py
+++ b/detector/server.py
@@ -20,6 +20,22 @@ def log(*args):
 
 class RequestHandler(SimpleHTTPRequestHandler):
 
+    def do_POST(self):
+        content_length = int(self.headers['Content-Length']) # <--- Gets the size of data
+        post_data = self.rfile.read(content_length) # <--- Gets the data itself
+        post_data = json.loads(post_data.decode('utf-8'))
+
+        self.begin_content('application/json;charset=UTF-8')
+
+        all_tokens, used_tokens, real, fake = self.infer(post_data['text'])
+
+        self.wfile.write(json.dumps(dict(
+            all_tokens=all_tokens,
+            used_tokens=used_tokens,
+            real_probability=real,
+            fake_probability=fake
+        )).encode()) 
+
     def do_GET(self):
         query = unquote(urlparse(self.path).query)
 
@@ -32,6 +48,16 @@ class RequestHandler(SimpleHTTPRequestHandler):
 
         self.begin_content('application/json;charset=UTF-8')
 
+        all_tokens, used_tokens, real, fake = self.infer(query)
+
+        self.wfile.write(json.dumps(dict(
+            all_tokens=all_tokens,
+            used_tokens=used_tokens,
+            real_probability=real,
+            fake_probability=fake
+        )).encode())
+
+    def infer(query):
         tokens = tokenizer.encode(query)
         all_tokens = len(tokens)
         tokens = tokens[:tokenizer.max_len - 2]
@@ -45,12 +71,7 @@ class RequestHandler(SimpleHTTPRequestHandler):
 
         fake, real = probs.detach().cpu().flatten().numpy().tolist()
 
-        self.wfile.write(json.dumps(dict(
-            all_tokens=all_tokens,
-            used_tokens=used_tokens,
-            real_probability=real,
-            fake_probability=fake
-        )).encode())
+        return all_tokens, used_tokens, real, fake
 
     def begin_content(self, content_type):
         self.send_response(200)