1 files changed, 4 insertions, 0 deletions
diff --git a/application.rb b/application.rb
index 4b36e7b..1e9cf35 100644
--- a/application.rb
+++ b/application.rb
@@ -202,6 +202,10 @@ post '/predict/?' do
 
   # process batch prediction
   if !params[:fileselect].blank?
+    if params[:fileselect][:filename] !~ /\.csv$/
+      @error_report = "Please submit a csv file."
+      return haml :error
+    end
     File.open('tmp/' + params[:fileselect][:filename], "w") do |f|
       f.write(params[:fileselect][:tempfile].read)
     end