search!

1 files changed, 9 insertions, 4 deletions

diff --git a/app.py b/app.py
index 5c4fdde..3d87995 100644
--- a/app.py
+++ b/app.py
@@ -11,13 +11,18 @@ app = Flask(__name__)
 
 with open("./data/beer_info_small.json") as beers_json:
     beer = json.loads(beers_json.read())
+
+valid_names = set()
 # beer_ids = {}
-# for long_id, val in beer.items():
-#     beer_ids[val[0]] = long_id.split('-')[1]
+for long_id, val in beer.items():
+    valid_names.add(val[0])
+
+print(valid_names)
+
 with open('./data/name_list.json') as f:
     name_list = json.load(f)
 
-orig_terms = list(name_list.keys())
+orig_terms = list([x for x in name_list.keys() if x.split(' — ')[1] in valid_names])
 
 search_words = [''.join(f for f in x.lower() if f not in string.punctuation) for x in orig_terms]
 
@@ -25,7 +30,7 @@ orig_term_map = dict(zip(search_words, orig_terms))
 
 
 beers_for_random = [b for b in beer if beer[b][3] > 75]
-print(len(beers_for_random))
+
 with open('./data/brewery_names.json') as f:
     brewery_names = json.load(f)
     # brewery_ids = {}