Update app.py

app.py

@@ -47,9 +47,10 @@ def retrieve_thoughts(query, ):
     df = pd.concat((df, pd.DataFrame([doc[1] for doc in docs_with_score], columns = ['score'])), axis = 1)
 
   # TO-DO: What if user query doesn't match what we provide as documents
-
-    tier_1 = df[df['score'] < 0.85]
-    tier_2 = df[((df['score'] < 1) * (df["score"] > 0.7))]
+    df.sort_values("score", inplace = True)
+    
+    tier_1 = df[df['score'] < 0.85].iloc[:30]
+    tier_2 = df[((df['score'] < 1) * (df["score"] > 0.8))]
 
     tier_1
     chunks_1 = tier_1.groupby(['title', 'url', ]).apply(lambda x: "\n...\n".join(x.sort_values('id')['page_content'].values[:min(4, len(x.sort_values('id')['page_content'].values))])).values
@@ -64,7 +65,7 @@ def retrieve_thoughts(query, ):
   # tier_1 = [doc[0]  for doc in docs if ((doc[1] < 1))][:5]
   # tier_2 = [doc[0]  for doc in docs if ((doc[1] > 0.7)*(doc[1] < 1.5))][10:15]
 
-    return {'tier 1':tier_1_adjusted.loc[:20], 'tier 2': tier_2_adjusted.loc[:5]}
+    return {'tier 1':tier_1_adjusted, 'tier 2': tier_2_adjusted.loc[:5]}
 
 def get_references(query):
   # TO-DO FINSIH UPP.