def index(): form = MushroomFeaturesForm() result = None if form.validate_on_submit(): tree_data = grab_tree(current_app.config['TREE_DATA_FILE_NAME']) data = [ form.cap_shape.data, form.cap_surface.data, form.cap_color.data, form.bruises.data, form.odor.data, form.gill_attachment.data, form.gill_spacing.data, form.gill_size.data, form.gill_color.data, form.stalk_shape.data, form.stalk_root.data, form.stalk_surface_above_ring.data, form.stalk_surface_below_ring.data, form.stalk_color_above_ring.data, form.stalk_color_below_ring.data, form.veil_type.data, form.veil_color.data, form.ring_number.data, form.ring_type.data, form.spore_print_color.data, form.population.data, form.habitat.data ] result = classify(tree_data, DATASET_LABELS, data) return render_template('main/index.html', form=form, result=result)
def setUp(self): self.tree_data = grab_tree(current_app.config['TREE_DATA_FILE_NAME'])