FUNNEL_PLOT.py

import sys
from tqdm import tqdm
import pandas as pd
import matplotlib.pyplot as plt
import os
from os import listdir
from os.path import isfile, join
from plot_tm_dockq_es_rms import generate_bayes_components_plot

# List of PDB IDs to process
pdb_ids = ['2jw1', '2lgk', '2lsk', '2law', '2kwv', '2mnu', '2mps', 
           '5tp6', '5urn', '7ovc', '7jyn', '7jq8', '6h8c']

# Generate plots for each PDB ID
for pdb_id in pdb_ids:
    print(f"Processing {pdb_id}...")
    generate_bayes_components_plot(pdb_id)