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)