lint

aider/voice.py

@@ -1,45 +1,41 @@
-import sounddevice as sd
-import numpy as np
-import keyboard
-import openai
-import io
-import tempfile
-import queue
-import soundfile as sf
 import os
+import queue
+import tempfile
+
+import openai
+import sounddevice as sd
+import soundfile as sf
+
+from .dump import dump  # noqa: F401
 
-from .dump import dump
 
 def record_and_transcribe():
-
     q = queue.Queue()
 
     def callback(indata, frames, time, status):
         """This is called (from a separate thread) for each audio block."""
-        if status:
-            print(status, file=sys.stderr)
         q.put(indata.copy())
 
-
-    filename = tempfile.mktemp(prefix='delme_rec_unlimited_', suffix='.wav', dir='')
+    filename = tempfile.mktemp(prefix="delme_rec_unlimited_", suffix=".wav", dir="")
 
     sample_rate = 16000  # 16kHz
 
-    with sf.SoundFile(filename, mode='x', samplerate=sample_rate, channels=1) as file:
+    with sf.SoundFile(filename, mode="x", samplerate=sample_rate, channels=1) as file:
         with sd.InputStream(samplerate=sample_rate, channels=1, callback=callback):
-            input('Press ENTER when done speaking...')
+            input("Press ENTER when done speaking...")
 
         while not q.empty():
             file.write(q.get())
 
-    with open(filename, 'rb') as fh:
+    with open(filename, "rb") as fh:
         transcript = openai.Audio.transcribe("whisper-1", fh)
 
-    text = transcript['text']
+    text = transcript["text"]
     return text
 
+
 if __name__ == "__main__":
-    api_key = os.getenv('OPENAI_API_KEY')
+    api_key = os.getenv("OPENAI_API_KEY")
     if not api_key:
         raise ValueError("Please set the OPENAI_API_KEY environment variable.")
     print(record_and_transcribe())