--- a/heatmap_viz.R	Thu Dec 13 04:14:21 2018 -0500
+++ b/heatmap_viz.R	Thu Jun 27 04:18:07 2019 -0400
@@ -1,4 +1,5 @@
 #!/usr/bin/Rscript
+options(warn=-1)  #TURN OFF WARNINGS !!!!!!
 
 suppressMessages(library('plotly',quietly = T))
 suppressMessages(library('heatmaply',quietly = T))
@@ -80,7 +81,7 @@
 }
 
 get_cols <-function(input_cols) {
-  input_cols <- gsub("c","",input_cols)
+  input_cols <- gsub("c","",gsub("C","",gsub(" ","",input_cols)))
   if (grepl(":",input_cols)) {
     first_col=unlist(strsplit(input_cols,":"))[1]
     last_col=unlist(strsplit(input_cols,":"))[2]
@@ -102,7 +103,7 @@
 output <- paste(output[1:length(output)-1],collapse=".")
 output <- paste(output,args$type,sep=".")
 cols = get_cols(args$cols)
-rownames_col = as.integer(gsub("c","",args$row_names))
+rownames_col = as.integer(gsub("c","",gsub("C","",gsub(" ","",args$row_names))))
 if (length(cols) <=1 ){
   stop("You need several colums to build a heatmap")
 }
@@ -116,7 +117,7 @@
 uto <- uto[rowSums(is.na(uto)) != ncol(uto), ]  #remove emptylines
 
 if (header) {
-  col_names = names(data)
+  col_names = names(uto)
 } else {
   col_names = cols
 }