Merge pull request #40 from slashml/dev

Dev

Author: eff-kay

CHANGELOG.md

@@ -2,6 +2,15 @@
 All notable changes to `slashml-python-client` aka `slashml` will be documented in this file.
 This project adheres to [Semantic Versioning](https://semver.org/).
 
+## 0.1.0 - 2023-05-13
+
+### Added
+- Added deepgram model for SpeechToText service
+- Added gcp model for SpeechToText service
+
+### Fixed
+- Fixed a bug in examples by replacing verbs with  `submit_job`
+
 ## 0.0.6 - 2023-05-11
 
 ### Changed

README.md

@@ -109,9 +109,11 @@ If the user preferes using the API calls directly, the documentation is availabl
 ### Speech-to-text
 For transcription, SlashML supports the following service providers:
 
-* [AssemblyAI](https://github.com/AssemblyAI)
+* [AssemblyAI](https://www.assemblyai.com/)
 * [AWS](https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/transcribe.html) 
 * Whisper ([OpenAI](https://openai.com/blog/whisper/))
+* [GOOGLE](https://cloud.google.com/speech-to-text)
+* [DEEPGRAM](https://deepgram.com/) 
 
 ### Summarization
 For text summarization, SlashML supports the following service providers:

examples/speech_to_text_async.py

@@ -12,7 +12,7 @@ def speech_to_text(audio_filepath, service_provider, api_key):
     print(f"file uploaded: {uploaded_file}")
 
     # Submit transcription request
-    job = model.transcribe(
+    job = model.submit_job(
         uploaded_file["upload_url"], service_provider=service_provider
     )
 
@@ -33,7 +33,7 @@ def speech_to_text(audio_filepath, service_provider, api_key):
 # Replace `API_KEY` with your SlasML API token. This example still runs without
 # the API token but usage will be limited
 API_KEY = None
-service_provider = SpeechToText.ServiceProvider.ASSEMBLY
+service_provider = SpeechToText.ServiceProvider.DEEPGRAM
 audio_filepath = "test.mp3"
 
 # Find all the service providers that we support by running the choices() method

examples/speech_to_text_sync.py

@@ -3,7 +3,7 @@
 # Replace `API_KEY` with your SlasML API token. This example still runs without
 # the API token but usage will be limited
 API_KEY = None
-service_provider = SpeechToText.ServiceProvider.AWS
+service_provider = SpeechToText.ServiceProvider.GOOGLE
 audio_filepath = "test.mp3"
 
 # Find all the service providers that we support by running the choices() method

examples/summarize_async.py

@@ -6,7 +6,7 @@ def summarize(text, service_provider, api_key):
     model = TextSummarization(api_key=api_key)
 
     # Submit request
-    job = model.summarize(text=text, service_provider=service_provider)
+    job = model.submit_job(text=text, service_provider=service_provider)
 
     assert job.status != "ERROR", f"{job}"
     print(f"Job ID: {job.id}")

examples/text_to_speech_async copy.py examples/text_to_speech_async.pyexamples/text_to_speech_async copy.py renamed to examples/text_to_speech_async.py

@@ -1,4 +1,5 @@
 from slashml import TextToSpeech
+import time
 
 # Replace `API_KEY` with your SlasML API token. This example still runs without
 # the API token but usage will be limited
@@ -13,7 +14,15 @@
 model = TextToSpeech(api_key=API_KEY)
 
 # Submit request
-job = model.speechify(text=input_text, service_provider=service_provider)
+job = model.submit_job(text=input_text, service_provider=service_provider)
 print(job)
 
-print (f"\nYou can access the audio file here: {job.audio_url}")
+# check job status
+response = model.status(job.id, service_provider=service_provider)
+
+while response.status == "IN_PROGRESS":
+    print(f"Response = {response}. Retrying in 30 seconds")
+    time.sleep(30)
+    response = model.status(job.id, service_provider=service_provider)
+
+print (f"\nYou can access the audio file here: {response.audio_url}")

setup.py

@@ -14,7 +14,7 @@ def read_req_file(req_type):
 
 setup(
     name="slashml",
-    version="0.0.6",
+    version="0.1.0",
     url="https://slashml.com/",
     author="eff-kay",
     author_email="faiizan14@gmail.com",

slashml/speech_to_text.py

@@ -9,6 +9,8 @@ class ServiceProvider(Enum):
         ASSEMBLY = "assembly"
         AWS = "aws"
         WHISPER = "whisper"
+        DEEPGRAM = 'deepgram'
+        GOOGLE = 'google'
 
         @classmethod
         def choices(cls):
@@ -55,6 +57,7 @@ def execute(self, upload_url: str, service_provider: ServiceProvider):
         # check job status
         response = getTaskStatus(self._base_url, self._headers, job.id, service_provider)
 
+        # while response.status == "IN_PROGRESS":
         while response.status == "IN_PROGRESS":
             time.sleep(5)
             response = getTaskStatus(self._base_url, self._headers, job.id, service_provider)