File size: 1,234 Bytes
3c36fb5
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45

import os
import time
from fastapi import APIRouter, Depends, HTTPException, status


from libs.convert_to_audio import convert_to_audio
from libs.header_api_auth import get_api_key
from libs.transformer.get_transcript_gradio_api import api_gradio_transcribe



router = APIRouter(prefix="/get-transcript-gradio", tags=["transcript"])

@router.get("/")
def get_transcript(audio_path: str, model_size: str = "distil-whisper/distil-small.en", api_key: str = Depends(get_api_key)):
    st = time.time()

    output_audio_folder = f"./cached/audio"

    if not os.path.exists(output_audio_folder):
        os.makedirs(output_audio_folder)


    output_file = f"{output_audio_folder}/{audio_path.split('/')[-1].split(".")[0]}.mp3"
    convert_to_audio(audio_path.strip(), output_file)

    try:
        text = api_gradio_transcribe(output_file)

    except Exception as error:
        raise HTTPException(status_code=status.HTTP_403_FORBIDDEN, detail=f"error>>>: {error}")
    finally:
         if os.path.exists(output_file):
            os.remove(output_file)

    et = time.time()

    elapsed_time = et - st

    return {"text": text,
            'list_sentence':  [],
            'elapsed_time': round(elapsed_time, 2)
            }