Changed output

app.py

@@ -149,7 +149,19 @@ def get_recommendations(user_id, model, data, G, user_nodes, post_nodes, node2id
     scores = [(post, torch.dot(user_embed, embeddings[node2idx[post]]).item()) for post in post_nodes if post not in user_interacted]
     scores = sorted(scores, key=lambda x: x[1], reverse=True)
 
-    return [post for post, _ in scores[:top_k]]
+    # Return the top_k post IDs
+    recommended_post_ids = [post for post, _ in scores[:top_k]]
+    return recommended_post_ids
+
+def fetch_full_post_records(post_ids):
+    """Fetch full post records from Supabase for the given post IDs."""
+    supabase = get_supabase_client()
+    if not post_ids:
+        return []
+    
+    # Query Supabase for all columns of the posts table where id is in the list of recommended post_ids
+    response = supabase.table('posts').select('*').in_('id', post_ids).execute()
+    return response.data
 
 @app.post("/rebuild")
 async def rebuild_handler():
@@ -161,8 +173,13 @@ async def get_recommendations_handler(user_id: str = Query(...)):
     if trained_model is None:
         raise HTTPException(status_code=500, detail="Model not initialized, please rebuild first.")
 
-    recs = get_recommendations(user_id, trained_model, pyg_data, G, user_nodes, post_nodes, node2idx)
-    return {"status": "success", "recommendations": recs}
+    # Get recommended post IDs
+    recommended_post_ids = get_recommendations(user_id, trained_model, pyg_data, G, user_nodes, post_nodes, node2idx)
+    
+    # Fetch full post records for the recommended post IDs
+    full_post_records = fetch_full_post_records(recommended_post_ids)
+    
+    return {"status": "success", "recommendations": full_post_records}
 
 @app.get("/")
 async def health_check():