Update app.py

app.py

@@ -63,6 +63,9 @@ def retrieve_thoughts(query, n):
     tier_1_adjusted = tier_1.groupby(['title', 'url']).first().reset_index()[['title', 'url', 'score']]
     tier_1_adjusted['ref'] = range(1, len(tier_1_adjusted) + 1 )
     tier_1_adjusted['chunks'] = chunks_1
+    score = tier_1.groupby(['title', 'url', ]).apply(lambda x: x['score'].mean()).values
+    tier_1_adjusted['score'] = score
+    tier_1_adjusted.sort_values("score", inplace = True)
 
     # 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']]