changed model transcription endpoint

app.py

@@ -17,7 +17,7 @@ from audio_recorder_streamlit import audio_recorder
 
 from logger import logger
 from utils import fs
-from enums import SAVE_PATH, ELO_JSON_PATH, ELO_CSV_PATH, EMAIL_PATH, TEMP_DIR, CREATE_TASK_URL
+from enums import SAVE_PATH, ELO_JSON_PATH, ELO_CSV_PATH, EMAIL_PATH, TEMP_DIR, CREATE_TASK_URL,TRANSCRIBE_URL
 
 result_queue = Queue()
 
@@ -136,7 +136,10 @@ def send_task(payload):
     header = {
         "Authorization": f"Bearer {os.getenv('CREATE_TASK_API_KEY')}"
     }
-    response = requests.post(CREATE_TASK_URL,json=payload,headers=header,timeout=300)
+    if payload["task"] in ["fetch_audio","write_result"]:
+        response = requests.post(CREATE_TASK_URL,json=payload,headers=header,timeout=300)
+    else:
+        response = requests.post(TRANSCRIBE_URL,json=payload,headers=header,timeout=300,stream=True)
     try:
         response = response.json()
     except Exception as e:
@@ -178,12 +181,17 @@ def call_function(model_name):
                     "audio_b64":True
                 }}
     else:
+        sr = st.session_state.audio['sample_rate']
+        array = st.session_state.audio['data']
+        if sr != 22050:
+            array = librosa.resample(array,sr,22050)
+        encoded_array = encode_audio_array(array)
         payload = {
                 "task":"transcribe_with_fastapi",
                 "payload":{
-                    "file_path":st.session_state.audio_path,
+                    "file_path":encoded_array,
                     "model_name":model_name,
-                    "audio_b64":False
+                    "audio_b64":True
                 }}
 
     transcript = send_task(payload)

enums.py

@@ -5,4 +5,5 @@ ELO_JSON_PATH = f"s3://{os.getenv('AWS_BUCKET_NAME')}/{os.getenv('ELO_JSON_PATH'
 ELO_CSV_PATH = f"s3://{os.getenv('AWS_BUCKET_NAME')}/{os.getenv('ELO_CSV_KEY')}"
 EMAIL_PATH = f"s3://{os.getenv('AWS_BUCKET_NAME')}/{os.getenv('EMAILS_KEY')}"
 TEMP_DIR = f"s3://{os.getenv('AWS_BUCKET_NAME')}/{os.getenv('AUDIOS_KEY')}"
-CREATE_TASK_URL = os.getenv("CREATE_TASK_URL")
+CREATE_TASK_URL = os.getenv("CREATE_TASK_URL")
+TRANSCRIBE_URL = os.getenv("TRANSCRIPTION_TASK_URL")