How to use it in python script?

[PiotrEsse]

Hi,
Can You help me with this - how to invoke whisper-diarization from a python code? Ive already prepared another script which call diarize.py using subprocess but In my opinion its not optimal.
Below is my working code. If you want to use parameters - you need to modify diarize.py itself as I found issues passing parametrs (like ie: --language pl ) in my script to a diarize.py

import os
import subprocess

# specify the directory you want to use
directory = '/home/Audio'

# use os.listdir to get all files in the directory
files = os.listdir(directory)

# loop over the files
for file in files:
    # ensure the file is an audio file
     if file.endswith('.wav') or file.endswith('.mp3') or file.endswith('.MP3') or file.endswith('.ogg') or file.endswith('.amr'):
        # create the command you want to run
        cmd = ['python', 'diarize.py', '-a', os.path.join(directory, file)]

        # use subprocess to run the command
        subprocess.run(cmd)

[PiotrEsse]

Anyone, please help.