fix

cas9on.py

@@ -208,28 +208,26 @@ def create_bigwig(df, bigwig_path):
     if not all(column in df.columns for column in ["Chr", "Start Pos", "End Pos", "Prediction"]):
         raise ValueError("DataFrame must contain 'Chr', 'Start Pos', 'End Pos', and 'Prediction' columns.")
 
-    # Convert positions to integers
+    # Convert positions to integers and sort the DataFrame
     df['Start Pos'] = df['Start Pos'].astype(int)
     df['End Pos'] = df['End Pos'].astype(int)
-
-    # Sort the DataFrame by chromosome and start position
     df = df.sort_values(by=['Chr', 'Start Pos'])
 
     # Prepare the BigWig header
     chr_sizes = df.groupby('Chr')['End Pos'].max().to_dict()
-    header = [(chr, int(size)) for chr, size in chr_sizes.items()]
+    header = [(str(chr), size) for chr, size in chr_sizes.items()]
 
-    # Create a BigWig file
+    # Create and write to the BigWig file
     bw = pyBigWig.open(bigwig_path, "w")
     bw.addHeader(header)
 
-    # Add entries to the BigWig file
+    # Group by chromosome and add entries
     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(chr, starts, ends=ends, values=values)
+        bw.addEntries([str(chr)] * len(starts), starts, ends=ends, values=values)
 
-    # Close the BigWig file
     bw.close()
 
+