fix

cas9on.py

@@ -202,10 +202,6 @@ def process_gene(gene_symbol, model_path):
 # def create_csv_from_df(df, output_path):
 #     df.to_csv(output_path, index=False)
 
-
-import pyBigWig
-
-
 def create_bigwig(df, bigwig_path):
     # Check for required columns in the DataFrame
     required_columns = ["Chr", "Start Pos", "End Pos", "Prediction"]
@@ -229,15 +225,11 @@ def create_bigwig(df, bigwig_path):
     bw.addHeader(header)
 
     # Add entries for each chromosome
-    for chr in sorted(chr_sizes.keys()):
-        chrom_df = df[df['Chr'] == chr]
-        starts = chrom_df['Start Pos'].tolist()
-        ends = chrom_df['End Pos'].tolist()
-        values = chrom_df['Prediction'].tolist()
-
-        # Ensure data is in the correct order and format
-        if starts and ends and values:
-            bw.addEntries(chr, starts, ends=ends, values=values)
+    for chr, group in df.groupby('Chr'):
+        starts = group['Start Pos'].tolist()
+        ends = group['End Pos'].tolist()
+        values = group['Prediction'].astype(float).tolist()
+        bw.addEntries(str(chr), starts, ends=ends, values=values)
 
     # Close the BigWig file
     bw.close()