From 73a0492585e238d32869bfa9c53ddc95481ab1c5 Mon Sep 17 00:00:00 2001
From: DH4
Date: Fri, 2 Jun 2017 00:30:44 -0500
Subject: Cancel button stops pre-processing too.

---
 core.py | 9 +++++++++
 1 file changed, 9 insertions(+)

(limited to 'core.py')

diff --git a/core.py b/core.py
index 5478f93..0bbe001 100644
--- a/core.py
+++ b/core.py
@@ -7,6 +7,7 @@ from PIL import Image
 import tempfile
 from shutil import rmtree
 import atexit
+import time
 
 class Core():
 
@@ -67,6 +68,8 @@ class Core():
     completeAudioArray = numpy.empty(0, dtype="int16")
 
     while True:
+      if self.canceled:
+        break
       # read 2 seconds of audio
       raw_audio = in_pipe.stdout.read(88200*4)
       if len(raw_audio) == 0:
@@ -110,3 +113,9 @@ class Core():
          shell=True
       )
       return sorted([os.path.join(self.tempDir, f) for f in os.listdir(self.tempDir)])
+
+  def cancel(self):
+    self.canceled = True
+
+  def reset(self):
+    self.canceled = False
-- 
cgit v1.2.3