From 5bc25568e78167f24e414cb9ac824d1557475e81 Mon Sep 17 00:00:00 2001
From: dfcarvajal <carvajal.diego@gmail.com>
Date: Mon, 22 Dec 2025 09:46:30 +0100
Subject: [PATCH] ADD: new dataset cells for endpoint

---
 src/app.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/app.py b/src/app.py
index 883a7c5..f7ece6e 100644
--- a/src/app.py
+++ b/src/app.py
@@ -269,6 +269,12 @@ def recommend_gruas():
                 "statistics": result.get('statistics', []),
                 "coverage": result.get('coverage', {})
             }
+
+
+        include_cell_data = request.args.get("include_cell_data", "false").lower() == "true"
+    
+        if include_cell_data:
+            response['prediction_data'] = df.to_dict('records')
         
         return jsonify(response)