From d7ba4efa8ec64c7b4b94037cc3cc2a6dcf1b0f59 Mon Sep 17 00:00:00 2001
From: Silver Kuusik <silver.kuusik@gmail.com>
Date: Thu, 27 Dec 2018 08:56:16 +0100
Subject: [PATCH] Update main.py

---
 main.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/main.py b/main.py
index 44d7270..c84fa85 100755
--- a/main.py
+++ b/main.py
@@ -88,9 +88,12 @@ def ws_handler():
             sumorobot.move(STOP)
             # for terminating delays in code
             sumorobot.terminate = True
-        elif b'calibrate_line' in data:
-            sumorobot.calibrate_line()
-            #print('calibrate')
+        elif b'calibrate_line_value' in data:
+            sumorobot.calibrate_line_value()
+            #print('calibrate_line')
+        elif b'calibrate_line_threshold' in data:
+            sumorobot.calibrate_line_threshold()
+            #print('calibrate_threshold')
         elif b'Gone' in data:
             print("server said 410 Gone, attempting to reconnect...")
             #conn = uwebsockets.connect(url)