Linted

Author: makermelissa

Magic_AI_Storybook/listener.py

@@ -14,7 +14,9 @@ def __init__(self, api_key, energy_threshold=300, record_timeout=30):
         self.recognizer = sr.Recognizer()
         self.recognizer.energy_threshold = energy_threshold
         with self.microphone as source:
-            self.recognizer.adjust_for_ambient_noise(source)  # we only need to calibrate once, before we start listening
+            self.recognizer.adjust_for_ambient_noise(
+                source
+            )  # we only need to calibrate once, before we start listening
         self.record_timeout = record_timeout
         self.listener_handle = None
         self.audio = None
@@ -24,18 +26,17 @@ def listen(self, ready_callback=None):
         self._start_listening()
         if ready_callback:
             ready_callback()
-        while (
-            self.listener_handle
-            and self.audio is None
-        ):
+        while self.listener_handle and self.audio is None:
             time.sleep(0.1)
         self.stop_listening()
 
     def _save_audio_callback(self, _recognizer, audio):
         self.audio = audio
 
     def _start_listening(self):
-        self.listener_handle = self.recognizer.listen_in_background(self.microphone, self._save_audio_callback)
+        self.listener_handle = self.recognizer.listen_in_background(
+            self.microphone, self._save_audio_callback
+        )
 
     def stop_listening(self, wait_for_stop=False):
         if self.listener_handle:
@@ -60,9 +61,9 @@ def recognize(self):
                     )
 
                     return result.strip()
-                except sr.RequestError as e:
+                except sr.RequestError:
                     time.sleep(3)
                 attempts += 1
                 print("I wasn't able to understand you. Please repeat that.")
             return None
-        return None
+        return None

Magic_AI_Storybook/make_shortcut.py

@@ -3,22 +3,26 @@
 # SPDX-FileCopyrightText: 2023 Melissa LeBlanc-Williams for Adafruit Industries
 #
 # SPDX-License-Identifier: MIT
-# Desktop Icon from <a href="https://www.flaticon.com/free-icons/book" title="book icons">Book icons created by Freepik - Flaticon</a>
+# Desktop Icon from <a href="https://www.flaticon.com/free-icons/book"
+# title="book icons">Book icons created by Freepik - Flaticon</a>
 
 import os
 
+
 def create_folders(file_path):
     path = os.path.dirname(file_path)
     if not os.path.exists(path):
         os.makedirs(path)
 
+
 def write_file(path, contents):
     create_folders(path)
     with open(path, "w") as f:
         f.write(contents)
 
     print(f"Shortcut created at {path}")
 
+
 def main():
     APP_TITLE = "Magic Storybook"
     RUN_IN_TERMINAL = True
@@ -40,17 +44,18 @@ def main():
     APP_ICON = APP_ICON.replace("~", user_homedir)
 
     shortcut_template = f"""[Desktop Entry]
-    Comment=Run {APP_TITLE}
-    Terminal={"true" if RUN_IN_TERMINAL else "false"}
-    Name={APP_TITLE}
-    Exec=sudo python {APP_PATH}
-    Type=Application
-    Icon={APP_ICON}
-    """
+Comment=Run {APP_TITLE}
+Terminal={"true" if RUN_IN_TERMINAL else "false"}
+Name={APP_TITLE}
+Exec=sudo python {APP_PATH}
+Type=Application
+Icon={APP_ICON}
+"""
 
     write_file(f"{user_homedir}/Desktop/{FILENAME}", shortcut_template)
     if AUTO_START:
         write_file(f"{user_homedir}/.config/autostart/{FILENAME}", shortcut_template)
 
+
 if __name__ == "__main__":
-    main()
+    main()

Magic_AI_Storybook/story.py

@@ -31,16 +31,16 @@
 
 # Quit Settings (Close book QUIT_CLOSES within QUIT_TIME_PERIOD to quit)
 QUIT_CLOSES = 3
-QUIT_TIME_PERIOD = 5 # Time period in Seconds
+QUIT_TIME_PERIOD = 5  # Time period in Seconds
 
 # Neopixel Settings
 NEOPIXEL_COUNT = 10
 NEOPIXEL_BRIGHTNESS = 0.2
 NEOPIXEL_ORDER = neopixel.GRBW
-NEOPIXEL_LOADING_COLOR = (0, 255, 0, 0)     # Loading/Dreaming (Green)
-NEOPIXEL_SLEEP_COLOR = (0, 0, 0, 0)         # Sleeping (Off)
-NEOPIXEL_WAITING_COLOR = (255, 255, 0, 0)   # Waiting for Input (Yellow)
-NEOPIXEL_READING_COLOR = (0, 0, 255, 0)     # Reading (Blue)
+NEOPIXEL_LOADING_COLOR = (0, 255, 0, 0)  # Loading/Dreaming (Green)
+NEOPIXEL_SLEEP_COLOR = (0, 0, 0, 0)  # Sleeping (Off)
+NEOPIXEL_WAITING_COLOR = (255, 255, 0, 0)  # Waiting for Input (Yellow)
+NEOPIXEL_READING_COLOR = (0, 0, 255, 0)  # Reading (Blue)
 NEOPIXEL_PULSE_SPEED = 0.1
 
 # Image Names
@@ -116,11 +116,13 @@
     print("Please make sure PROMPT_FILE points to a valid file.")
     sys.exit(1)
 
+
 def strip_fancy_quotes(text):
     text = re.sub(r"[\u2018\u2019]", "'", text)
     text = re.sub(r"[\u201C\u201D]", '"', text)
     return text
 
+
 class Position(Enum):
     TOP = 0
     CENTER = 1
@@ -237,7 +239,7 @@ def start(self):
         with open(PROMPT_FILE, "r") as f:
             self._prompt = f.read()
 
-        #Initialize the Listener
+        # Initialize the Listener
         self.listener = Listener(openai.api_key, ENERGY_THRESHOLD, RECORD_TIMEOUT)
 
         # Preload remaining images
@@ -348,7 +350,12 @@ def _handle_loading_status(self):
         self.pixels.show()
 
     def _set_status_color(self, status_color):
-        if status_color not in [NEOPIXEL_READING_COLOR, NEOPIXEL_WAITING_COLOR, NEOPIXEL_SLEEP_COLOR, NEOPIXEL_LOADING_COLOR]:
+        if status_color not in [
+            NEOPIXEL_READING_COLOR,
+            NEOPIXEL_WAITING_COLOR,
+            NEOPIXEL_SLEEP_COLOR,
+            NEOPIXEL_LOADING_COLOR,
+        ]:
             raise ValueError(f"Invalid status color {status_color}.")
 
         # Handle loading color by setting the loading flag
@@ -713,6 +720,7 @@ def _sendchat(self, prompt):
     def running(self):
         return self._running
 
+
 def parse_args():
     parser = argparse.ArgumentParser()
     # Book will only be rendered vertically for the sake of simplicity