bin/aipim in aipim-rails-0.0.139 vs bin/aipim in aipim-rails-0.0.157

@@ -20,12 +20,20 @@
 end
 
 if ARGV[0] == 'generate'
 	system('mkdir -p aipim')
 	system('mkdir -p aipim/screenshots')
+	system('touch aipim/screenshots/empty')
+	system('mkdir -p aipim/html')
+	system('mkdir -p aipim/html/costum')
+	system('cp '+path+'lib/assets/bootstrap.min.css aipim/html')
+	system('cp '+path+'lib/assets/jquery-1.9.1.js aipim/html')
+	system('cp '+path+'lib/assets/costum.css aipim/html/costum')
+	system('cp '+path+'lib/assets/costum.js aipim/html/costum')
 
-	width = "1024"
+
+	width = "1366"
 	height = "768"
 
 	unless ARGV[1].nil?
 		width  = ARGV[1].split("x")[0]
 		height = ARGV[1].split("x")[1]
@@ -39,19 +47,17 @@
 	f = File.open("features/support/screenshot.rb", "w")
 	f.puts f_str
 	f.close
 
 elsif ARGV[0] == 'html'
-	%x[rm -rf aipim/html]
-	%x[mkdir -p aipim/html]
+	%x[rm -f aipim/html/*.feature.html]
 	features = []
 	files = get_files
 	files.each do |f|
 		features << Parser.init(f)
 	end
+	ConvertToHtml.index(features)
 	ConvertToHtml.init(features)
-	system('cp '+path+'lib/assets/bootstrap.min.css aipim/html')
-	system('cp '+path+'lib/assets/jquery-1.9.1.js aipim/html')
 
 elsif ARGV[0] == 'markdown' || ARGV[0] == 'md'
 	%x[rm -rf aipim/markdown]
 	%x[mkdir -p aipim/markdown]
 	features = []