Update app.py

app.py

@@ -58,8 +58,8 @@ def retrieve_thoughts(query, n):
     # 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
-    tier_1_adjusted = tier_1.groupby(['title', 'url', '_id']).first().reset_index()[['_id', 'title', 'url', 'score']]
+    chunks_1 = tier_1.groupby(['title', 'url']).apply(lambda x: "\n...\n".join(x.sort_values('id')['page_content'].values)).values
+    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['content'] = chunks_1