Update app.py

app.py

@@ -59,7 +59,6 @@ def read_root():
 def test_post(item: Item):
     return {"message": f"Received item '{item.name}' with quantity {item.quantity}"}
 
-
 @app.post("/predict")
 def predict(input_data: UserInput):
     try:
@@ -81,7 +80,8 @@ def predict(input_data: UserInput):
             'child': user_dict['family']['child'],
             'consumption': v['predicted_consumption'],
             'finish_error': v['predicted_finish_error'],
-            'finish_days': v['predicted_finish_days']
+            'finish_days': v['predicted_finish_days'],
+            'stock_quantity': user_dict['stock'][k]   # <--- Added stock quantity here
         } for k, v in predictions.items()])
 
         model.insert_feedback(user_id, feedback)