from subprocess import run
from pathlib import Path

def predict_genes_balrog(contigs_fasta, output_prefix):
    run(('balrog', '-i', contigs_fasta, '-o', f'{output_prefix}.gff3'))

def _predict_genes_balrog(run_accession, contigs_dir, genes_dir):
    predict_genes_balrog(
        Path(contigs_dir) / f'{run_accession}.contigs.fa',
        Path(genes_dir) / run_accession
    )
