utils/transcriber.py

import os
from dotenv import load_dotenv
from gradio_client import Client, handle_file

def transcriber(invideo_file:str, srt_file:str,
        max_words_per_line:int, task:str, model_version:str):
        load_dotenv()
        HF_TOKEN = os.getenv("hf_token")
        HF_SPACE = os.getenv("hf_space")
        client = Client(HF_SPACE, hf_token=HF_TOKEN)
        result = client.predict(
                file_input=handle_file(invideo_file),
                max_words_per_line=max_words_per_line,
                task=task,
                model_version=model_version,
                api_name="/predict"
        )
        with open(srt_file, "w", encoding='utf-8') as file:
                file.write(result[0])
        return srt_file