Update app.py

app.py

@@ -43,7 +43,7 @@ def retrieve_thoughts(query, n):
     
 
     chunks_1 = tier_1.groupby(['_id' ]).apply(lambda x: {f"chunk_{i}": row for i, row  in enumerate(x.sort_values('id')[['id', 'score','page_content']].to_dict('records'))}).values
-    tier_1_adjusted = tier_1.groupby(['_id']).first().reset_index()[['_id', 'title', 'url', 'score']]
+    tier_1_adjusted = tier_1.groupby(['_id']).first().reset_index()[['_id', 'title', 'author','url', 'score']]
     tier_1_adjusted['ref'] = range(1, len(tier_1_adjusted) + 1 )
     tier_1_adjusted['chunks'] = list(chunks_1)
     score = tier_1.groupby(['_id' ]).apply(lambda x: x['score'].mean()).values
@@ -73,7 +73,7 @@ def qa_retrieve(query,):
 
     tier_1 = thoughts['tier 1']
     
-    reference = tier_1[['_id', 'url', 'title', 'chunks', 'score']].to_dict('records')
+    reference = tier_1[['_id', 'url', 'author', 'title', 'chunks', 'score']].to_dict('records')
 
     return {'Reference': reference}