Update app.py

app.py

@@ -53,8 +53,9 @@ def retrieve_thoughts(query, n):
 
   # TO-DO: What if user query doesn't match what we provide as documents
 
-    tier_1 = df[df['score'] < 0.7]
-    tier_2 = df[(df['score'] < 0.95) * (df["score"] > 0.7)]
+    tier_1 = df[df['score'] < 0.95]
+    tier_1 = tier_1[:min(len(tier_1),150)]
+    # tier_2 = df[(df['score'] < 0.95) * (df["score"] > 0.7)]
 
 
     chunks_1 = tier_1.groupby(['title', 'url', '_id']).apply(lambda x: "\n...\n".join(x.sort_values('id')['page_content'].values)).values
@@ -62,9 +63,9 @@ def retrieve_thoughts(query, n):
     tier_1_adjusted['ref'] = range(1, len(tier_1_adjusted) + 1 )
     tier_1_adjusted['content'] = chunks_1
 
-    chunks_2 = tier_2.groupby(['title', 'url', '_id']).apply(lambda x: "\n...\n".join(x.sort_values('id')['page_content'].values)).values
-    tier_2_adjusted = tier_2.groupby(['title', 'url', '_id']).first().reset_index()[['_id', 'title', 'url']]
-    tier_2_adjusted['content'] = chunks_2
+    # chunks_2 = tier_2.groupby(['title', 'url', '_id']).apply(lambda x: "\n...\n".join(x.sort_values('id')['page_content'].values)).values
+    # tier_2_adjusted = tier_2.groupby(['title', 'url', '_id']).first().reset_index()[['_id', 'title', 'url']]
+    # tier_2_adjusted['content'] = chunks_2
 
     if n:
       tier_1_adjusted = tier_1_adjusted[:min(len(tier_1_adjusted), n)]