styles

1 files changed, 6 insertions, 3 deletions

diff --git a/app.py b/app.py
index d04a6ba..cab272b 100644
--- a/app.py
+++ b/app.py
@@ -3,7 +3,7 @@ from flask import render_template
 from flask import redirect
 from flask import request
 from flask import url_for
-from utils import get_closest, translate_to_attr
+from utils import get_closest, translate_to_attr, get_style_preds
 import json
 import random
 import re
@@ -45,6 +45,9 @@ def get_beer_details(brewery_id, beer_id):
              'Juicy', 'Light', 'Thin', 'Spicy', 'Dry', 'Fruity', 'Herbal',
              'Salty']
     similar_beers = get_closest(full_id)
+
+    style_map = get_style_preds(full_id)
+    
     similar_beer_objects = []
     for item in similar_beers:
         item_result, item_brewer = beer_lookup(item)
@@ -60,10 +63,10 @@ def get_beer_details(brewery_id, beer_id):
             attr_beer_objects.append(BeerInfo(item[0], item_brewer, item_result))
         return render_template('beerpage.jinja2', beer_data=beerinfo,
                 attrs=attrs, similar_beers=similar_beer_objects, attr=attr,
-                direction=direction, attr_beers=attr_beer_objects)
+                direction=direction, attr_beers=attr_beer_objects, style_preds=similar_beers)
 
     return render_template('beerpage.jinja2', beer_data=beerinfo, attrs=attrs,
-                           similar_beers=similar_beer_objects)
+                           similar_beers=similar_beer_objects, style_preds=similar_beers)
 
 
 def beer_lookup(brewery_id, beer_id=None):