Merge pull request #47 from slashml/update-exampels

update example

Author: eff-kay

examples/pipeline_podcast_summary.py

@@ -2,7 +2,7 @@
 
 # Replace `API_KEY` with your SlasML API token. This example still runs without
 # the API token but usage will be limited
-API_KEY = "a7011983a0f3d64ee113317b1e36f8e5bf56c14a"
+API_KEY = "YOUR_API_KEY"
 
 service_provider_speech_to_text = SpeechToText.ServiceProvider.WHISPER
 service_provider_summarize = TextSummarization.ServiceProvider.OPENAI

examples/speech_summarized.py

@@ -1,69 +1,30 @@
 from slashml import SpeechToText, TextSummarization
-import time
-
-
-def speech_to_text(uploaded_audio_url, service_provider, api_key):
-    # Initialize model
-    model = SpeechToText(api_key=api_key)
-
-    # Submit transcription request
-    job = model.transcribe(uploaded_audio_url, service_provider=service_provider)
-
-    assert job.status != "ERROR", f"{job}"
-    print(f"Job ID: {job.id}")
-
-    # check job status
-    response = model.status(job.id, service_provider=service_provider)
-
-    while response.status == "IN_PROGRESS":
-        time.sleep(30)
-        response = model.status(job.id, service_provider=service_provider)
-        print(f"Response = {response}. Retrying in 30 seconds")
-
-    return response
-
-
-def summarize(text, service_provider, api_key):
-    # Initialize model
-    model = TextSummarization(api_key=api_key)
-
-    # Submit request
-    job = model.summarize(text=text, service_provider=service_provider)
-
-    assert job.status != "ERROR", f"{job}"
-    print(f"Job ID: {job.id}")
-
-    # Check job status
-    response = model.status(job.id, service_provider=service_provider)
-
-    # Keep checking job status until the task is complete
-    while response.status == "PENDING":
-        print(f"Response = {response}. Retrying in 30 seconds")
-        time.sleep(30)
-        response = model.status(job.id, service_provider=service_provider)
-
-    return response
-
 
 # Replace `API_KEY` with your SlasML API token. This example still runs without
 # the API token but usage will be limited
-API_KEY = "b503d137d229fdd4085f59e9ea06ac95d2706182"
+API_KEY = "YOUR_API_KEY"
 
-service_provider_speech_to_text = SpeechToText.ServiceProvider.AWS
+service_provider_speech_to_text = SpeechToText.ServiceProvider.WHISPER
 service_provider_summarize = TextSummarization.ServiceProvider.OPENAI
 
 # 10 minute audio file already uploaded
 uploaded_url = (
     "https://slashml.s3.ca-central-1.amazonaws.com/fda70f6a-6057-4541-adf1-2cf4f4182929"
 )
 
+transcribe = SpeechToText(api_key=API_KEY)
+summarize = TextSummarization(api_key=API_KEY)
 
-response_speech_to_text = speech_to_text(uploaded_url, service_provider_speech_to_text, API_KEY)
+response = transcribe.execute(
+    upload_url=uploaded_url, service_provider=service_provider_speech_to_text
+)
 
-transcribed_text = response_speech_to_text.transcription_data.transcription
+print('starting pipeline, the first response might take 10 secs')
+transcribed_text = response.transcription_data.transcription
 print (f"Transcribed Text = {transcribed_text}")
 
-response_summarize = summarize(transcribed_text, service_provider_summarize, API_KEY)
+response_summarize = summarize.execute(transcribed_text, service_provider_summarize)
+
 summary = response_summarize.summarization_data
 
 print (f"Summarized Text = {summary}")

examples/speech_to_text_async.py

@@ -32,7 +32,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
+API_KEY = "YOUR_API_KEY"
 service_provider = SpeechToText.ServiceProvider.DEEPGRAM
 audio_filepath = "test.mp3"
 

examples/speech_to_text_sync.py

@@ -2,7 +2,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
+API_KEY = "YOUR_API_KEY"
 service_provider = SpeechToText.ServiceProvider.REV
 audio_filepath = "test.mp3"
 

examples/summarize_async.py

@@ -25,7 +25,7 @@ def summarize(text, 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 = "b503d137d229fdd4085f59e9ea06ac95d2706182"
+API_KEY = "YOUR_API_KEY"
 service_provider = TextSummarization.ServiceProvider.OPENAI
 input_text = """A good writer doesn't just think, and then write down what he thought, as a sort of transcript. A good writer will almost always discover new things in the process of writing. And there is, as far as I know, no substitute for this kind of discovery. Talking about your ideas with other people is a good way to develop them. But even after doing this, you'll find you still discover new things when you sit down to write. There is a kind of thinking that can only be done by writing."""
 

examples/summarize_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 = "a7011983a0f3d64ee113317b1e36f8e5bf56c14a"
+API_KEY = "YOUR_API_KEY"
 service_provider = TextSummarization.ServiceProvider.OPENAI
 
 input_text = """A good writer doesn't just think, and then write down what he thought, as a sort of transcript. A good writer will almost always discover new things in the process of writing. And there is, as far as I know, no substitute for this kind of discovery. Talking about your ideas with other people is a good way to develop them. But even after doing this, you'll find you still discover new things when you sit down to write. There is a kind of thinking that can only be done by writing."""

examples/text_to_speech_async.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
+API_KEY = "YOUR_API_KEY"
 service_provider = TextToSpeech.ServiceProvider.AWS
 input_text = "To be or not to be, that is the question!"
 

examples/text_to_speech_sync.py

@@ -2,7 +2,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
+API_KEY = "YOUR_API_KEY"
 service_provider = TextToSpeech.ServiceProvider.AWS
 input_text = "To be or not to be, that is the question!"