Testing for transcribe

stream_test.py

@@ -0,0 +1,62 @@
+# Start by making sure the `assemblyai` package is installed.
+# If not, you can install it by running the following command:
+# pip install -U assemblyai
+#
+# Then, make sure you have PyAudio installed: https://pypi.org/project/PyAudio/
+#
+# Note: Some macOS users might need to use `pip3` instead of `pip`.
+
+import pyaudio
+import assemblyai as aai
+
+aai.settings.api_key = "aa9962f0088a449a9c4ab2361e96cc08"
+
+def on_open(session_opened: aai.RealtimeSessionOpened):
+  "This function is called when the connection has been established."
+
+  print("Session ID:", session_opened.session_id)
+
+def on_data(transcript: aai.RealtimeTranscript):
+  "This function is called when a new transcript has been received."
+
+  if not transcript.text:
+    return
+
+  if isinstance(transcript, aai.RealtimeFinalTranscript):
+    print(transcript.text, end="\r\n")
+  else:
+    print(transcript.text, end="\r")
+
+def on_error(error: aai.RealtimeError):
+  "This function is called when the connection has been closed."
+
+  print("An error occured:", error)
+
+def on_close():
+  "This function is called when the connection has been closed."
+
+  print("Closing Session")
+
+transcriber = aai.RealtimeTranscriber(
+  on_data=on_data,
+  on_error=on_error,
+  sample_rate=44_100,
+  on_open=on_open, # optional
+  on_close=on_close, # optional
+)
+
+
+pa = pyaudio.PyAudio()
+for i in range(pa.get_device_count()):
+  print (pa.get_device_info_by_index(i))
+
+# Start the connection
+#transcriber.connect()
+
+# Open a microphone stream
+#microphone_stream = aai.extras.MicrophoneStream()
+
+# Press CTRL+C to abort
+#transcriber.stream(microphone_stream)
+
+#transcriber.close()