Update app.py

app.py

@@ -41,23 +41,22 @@ def sale_pred_batch():
     predicted_sale = model.predict(input_data).tolist()
     print("Predicted Sales Length:", len(predicted_sale))
     
-    predicted_sales = [round(float(i)) for i in predicted_sale]
-    sale_outlets = input_data['Store_Id'].tolist()
-    print("Sale Outlets Length:", len(sale_outlets))
+    # Add predictions to input data
+    input_data['Predicted_Sale'] = predicted_sale
+    print("Input Data with Predictions:", input_data.head())
     
-    # Create response
-    response = dict(zip(sale_outlets, predicted_sales))
-    
-    print("Response:", response)
-    repshape = {'input_shape': input_data.shape, 'predicted_count': len(predicted_sale)}
+    # Group by Store_Id and sum the predicted sales
+    grouped_sales = input_data.groupby('Store_Id')['Predicted_Sale'].sum().to_dict()
+    print("Grouped Sales:", grouped_sales)
     
-    print("Response Shape:", repshape)
+    # Create response
+    response = {
+        'store_sales': {store_id: round(float(sale), 2) for store_id, sale in grouped_sales.items()}
+    }
+    print("Final Response:", response)
     
-    # Return combined response
-    return jsonify({
-        'predictions': response,
-        'metadata': repshape
-    })
+    return jsonify(response)
+