Update app.py

app.py

@@ -47,13 +47,15 @@ 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
-    df.sort_values("score", inplace = True)
+  # df.sort_values("score", inplace = True)
     
     tier_1 = df
     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(3, len(x.sort_values('id')['page_content'].values))])).values
+    chunks_1 = tier_1.groupby(['title', 'url', ]).apply(lambda x: "\n...\n".join(x.sort_values('score').iloc[:3].sort_values('id')['page_content'].values)).values
+    score = tier_1.groupby(['title', 'url', ]).apply(lambda x: x.sort_values('score').iloc[:3]['score'].mean()).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
@@ -64,6 +66,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]
+    tier_1_adjusted.sort_values("score", inplace = True)
 
     return {'tier 1':tier_1_adjusted[:18], 'tier 2': tier_2_adjusted.loc[:5]}