#5 결과 검증: 계통 결정 돌연변이와 연관성 #

import pandas as pd
import numpy as np
import os
os.sys.path.append("/data/home/ysh980101/2407/Mutclust") 

from pathlib import Path
from Bin.Utils.utils import *
from Bin.arg_parser import *
from Bin.mlib import *

os.sys.path.append("/data/home/ysh980101/2506/mutclust") 
from Bin.sc import *

os.chdir("/data/home/ysh980101/2506/mutclust")

lineage_info_dir = '/data/home/ysh980101/2411/data/mutation_info'
covid_annotation = "/data/home/ysh980101/2404/Data/covid_annotation.tsv"
sig_hotspots = "result/sig_hotspots.csv"

lineage_info = make_lineage_info(lineage_info_dir)

hotspot_lineage = make_hotspot_lineage(lineage_info, sig_hotspots_path, covid_annotation)
hotspot_lineage

plot_hotspot_lineage(hotspot_lineage)

outdir = "result/"
hotspot_lineage.to_csv(f"{outdir}Supplementary_table_1.csv", index=False)