Merge pull request #2 from slashml/first_release

 second version sdk

Author: JJneid

.DS_Store

@@ -0,0 +1,6 @@
+SDK for SlashML documentation:
+
+- pip install slashml
+- add SLASH_API_KEY to sys path
+- methods: upload_audio, transcribe, status 
+- 

SDK_v1/__pycache__/speechtotext.cpython-310.pyc

@@ -1,22 +1,57 @@
-class SpeechToText:
-    SLASHML_BASE_URL = 'https://api.slashml.com/v1/speech-to-text'
-    SLASHML_UPLOAD_URL = SLASHML_BASE_URL+'/upload'
-    SLASHML_TRANSCRIPT_URL = SLASHML_BASE_URL+'/transcribe'
-    SLASHML_TRANSCRIPT_STATUS_URL = lambda self,id: f"{SpeechToText.SLASHML_TRANSCRIPT_URL}/{id}"
+from django.http import JsonResponse
+import requests
+import os
+from pathlib import Path
 
+class SpeechToText:
+    SLASHML_BASE_URL = 'https://api.slashml.com/speech-to-text/v1'
+    SLASHML_UPLOAD_URL = SLASHML_BASE_URL+'/upload/'
+    SLASHML_TRANSCRIPT_URL = SLASHML_BASE_URL+'/transcribe/'
+    SLASHML_STATUS_URL = SLASHML_BASE_URL+'/transcription'
+    SLASHML_TRANSCRIPT_STATUS_URL = lambda self,id: f"{SpeechToText.SLASHML_STATUS_URL}/{id}/"
+ 
+    HEADERS:dict = {}
+    
+    def __init__(self) -> None:
+        self.HEADERS = {'authorization': os.environ.get('SLASHML_API_KEY')}
 
-    def upload_audio(self, file_location) :
+    def upload_audio(self, file_location:str, API_KEY: str, model_choice: str ):
         # here we can also add the service? assemblyai, aws, gcp?
-        return    self.SLASHML_UPLOAD_URL #response.json()
 
-    def transcribe(self,upload_url, model_params:dict()):
+        token="Token {0}".format(API_KEY)
+        headers = {'authorization': token}
+        payload = { 'model':model_choice }
+        files=[
+        ('audio',('test_audio.mp3',open(file_location,'rb'),'audio/mpeg'))
+        ]
+        #import pdb
+        #pdb.set_trace()
+        response = requests.post(self.SLASHML_UPLOAD_URL,
+                                headers=headers,
+                                data=payload,files=files)
+        return response.text
+
+    def transcribe(self,upload_url:str, API_KEY: str, model_choice: str ):
         # here we can add more model params
         transcript_request = {'audio_url': upload_url}
-        #transcript_response = requests.post( self.SLASHML_TRANSCRIPT_URL, json=transcript_request)
-        job_id=self.SLASHML_TRANSCRIPT_URL
-        return job_id
+        token="Token {0}".format(API_KEY)
+        headers = {'authorization': token}
+        payload = {
+        "uploaded_audio_url": upload_url,
+        "model": model_choice
+        }
+
+        response = requests.post(self.SLASHML_TRANSCRIPT_URL, headers=headers, data=payload)
+        return response.text
 
-    def status(self, job_id:str):
-        #options={"status1"="queue","status2":"completed","status3":"deleted","text":"output_api_result"...}
-        return self.SLASHML_TRANSCRIPT_STATUS_URL(job_id) 
+    def status(self,job_id:str, API_KEY: str, model_choice: str ):
+        token="Token {0}".format(API_KEY)
+        headers = {'authorization': token}
+        payload = {
+        "model": model_choice
+        }
+    #import pdb
+       # pdb.set_trace()        
+        response = requests.get(self.SLASHML_TRANSCRIPT_STATUS_URL(job_id) , headers=headers, data=payload)
+        return response.text
 

SDK_v1/readme.md

@@ -1,41 +1,51 @@
 #### unit testing
 import pytest
 
-
+import json
 def test_upload():
     import speechtotext
     # given: there is a local file
     speect_to_text = speechtotext.SpeechToText()
-    file_location="local_path"
+    file_location="/Users/JJneid/Desktop/SlashMl/s2t_experiments/api_tests/test_audio_1.mp3"
     # when
-    result = speect_to_text.upload_audio(file_location)
-
+    API_KEY="1bd15e1c161ff6d4db2ea1d661d7468b4fa61ca9"
+    model_choice="assembly"
+    result = speect_to_text.upload_audio(file_location,API_KEY, model_choice)
+    print(result)
     # then
-    assert result== 'https://api.slashml.com/v1/speech-to-text/upload'
+  #  assert result== 'https://api.slashml.com/v1/speech-to-text/upload'
 
 def test_transcribe():
     import speechtotext
     # given: there is a local file
     speect_to_text = speechtotext.SpeechToText()
-    upload_url="local_path"
+    upload_url="https://cdn.assemblyai.com/upload/f973cc7a-45f8-47b9-8ebc-20bd2f63a2bd"
     # when
-    result = speect_to_text.transcribe(upload_url,dict())
-
-    # then
-    assert result== 'https://api.slashml.com/v1/speech-to-text/transcribe'
+    API_KEY="1bd15e1c161ff6d4db2ea1d661d7468b4fa61ca9"
+    model_choice="assembly"
+    result = speect_to_text.transcribe(upload_url,API_KEY, model_choice)
 
+    print(result)
 
 def test_status():
     import speechtotext
     # given: there is a local file
     speect_to_text = speechtotext.SpeechToText()
+    upload_url="https://cdn.assemblyai.com/upload/28313e17-9a51-43cd-a29f-68dedd772f83"
     # when
-    job_id = '123'
-    result = speect_to_text.status(job_id)
-
-    # then
-    assert result== 'https://api.slashml.com/v1/speech-to-text/transcribe/{}'.format(job_id)
-
-
-# def test_status():
-#     assert 
+    job_id= "r4ff9gd76c-530e-4e27-bed4-ce12b63b26b0"
+    API_KEY="1bd15e1c161ff6d4db2ea1d661d7468b4fa61ca9"
+    model_choice="assembly"
+    result = speect_to_text.status(job_id,API_KEY, model_choice=model_choice)
+
+    print(json.loads(result)["text"])
+
+#test_transcribe()
+#test_transcribe()
+test_status()
+# API_KEY="1bd15e1c161ff6d4db2ea1d661d7468b4fa61ca9"
+# token="Token {api}".format(api=API_KEY) 
+# print(token)
+
+# headers = {'authorization': token}
+# print(headers)