diff --git a/README.md b/README.md index dbf7dcf..d4055bf 100644 --- a/README.md +++ b/README.md @@ -17,7 +17,7 @@ ollama library ## Installation Ensure Python and the necessary libraries are installed: ``` -pip install openai speechrecognition torch ollama +pip install openai speechrecognition torch ollama RealtimeSTT python_weather icrawler spotipy ``` Go to https://platform.openai.com/assistants to set up your assistant to get diff --git a/RealtimeSTT.py b/RealtimeSTT.py new file mode 100644 index 0000000..a61e40f --- /dev/null +++ b/RealtimeSTT.py @@ -0,0 +1,25 @@ +import speech_recognition as sr + +class AudioToTextRecorder: + def __init__(self, spinner=False, model="tiny.en", language="en", post_speech_silence_duration=0.1, silero_sensitivity=0.4): + self.recognizer = sr.Recognizer() + self.microphone = sr.Microphone() + self.spinner = spinner + self.model = model + self.language = language + self.post_speech_silence_duration = post_speech_silence_duration + self.silero_sensitivity = silero_sensitivity + + def text(self): + with self.microphone as source: + self.recognizer.adjust_for_ambient_noise(source) + print("Listening...") + audio = self.recognizer.listen(source) + try: + print("Recognizing...") + text = self.recognizer.recognize_google(audio, language=self.language) + return text + except sr.UnknownValueError: + return "Could not understand audio" + except sr.RequestError as e: + return f"Could not request results; {e}" diff --git a/requirements.txt b/requirements.txt index 976837a..bc82f7c 100644 --- a/requirements.txt +++ b/requirements.txt @@ -5,3 +5,8 @@ torch git+https://github.com/openai/whisper.git transformers pygame +RealtimeSTT +openai +python_weather +icrawler +spotipy