1 files changed, 4 insertions, 1 deletions
diff --git a/hangup.py b/hangup.py
index 2bfb1a7..cfe6bdf 100644
--- a/hangup.py
+++ b/hangup.py
@@ -5,9 +5,10 @@ import time
 BUTTON_GPIO = 24
 
 class HangUpThread(threading.Thread):
-    def __init__(self, phone_held, dial_manager):
+    def __init__(self, phone_held, phone_hung_up, dial_manager):
         threading.Thread.__init__(self, args=(), kwargs=None)
         self.phone_held = phone_held
+        self.phone_hung_up = phone_hung_up
         self.dial_manager = dial_manager
         self.daemon = True
 
@@ -18,8 +19,10 @@ class HangUpThread(threading.Thread):
         while True:
             if not GPIO.input(BUTTON_GPIO):
                 self.phone_held.clear()
+                self.phone_hung_up.set()
                 self.dial_manager.clear_sequence()
             else:
                 self.phone_held.set()
+                self.phone_hung_up.clear()
             time.sleep(0.1)