Update app.py

app.py

@@ -51,7 +51,8 @@ def retrieve_thoughts(query, ):
     tier_1 = df[df['score'] < 0.85]
     tier_2 = df[((df['score'] < 1) * (df["score"] > 0.7))]
 
-    chunks_1 = tier_1.groupby(['title', 'url', ]).apply(lambda x: "\n...\n".join(x.sort_values('id')['page_content'].values)).values
+    tier_1
+    chunks_1 = tier_1.groupby(['title', 'url', ]).apply(lambda x: "\n...\n".join(x.sort_values('id')['page_content'].values[:min(5, len(x.sort_values('id')['page_content'].values))])).values
     tier_1_adjusted = tier_1.groupby(['title', 'url', ]).first().reset_index()[[ 'title', 'url']]
     tier_1_adjusted['ref'] = range(1, len(tier_1_adjusted) + 1 )
     tier_1_adjusted['content'] = chunks_1
@@ -63,7 +64,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[:25], 'tier 2': tier_2.loc[:5]}
+    return {'tier 1':tier_1_adjusted.loc[:25], 'tier 2': tier_2_adjusted.loc[:5]}
 
 def get_references(query):
   # TO-DO FINSIH UPP.