Update app.py

app.py

@@ -47,8 +47,17 @@ def sale_pred_batch():
     
     # Create response
     response = dict(zip(sale_outlets, predicted_sales))
-    repshape=dict(zip(input_data.shape,len(predicted_sale)))
+    
     print("Response:", response)
+    repshape = {'input_shape': input_data.shape, 'predicted_count': len(predicted_sale)}
+    
+    print("Response Shape:", repshape)
+    
+    # Return combined response
+    return jsonify({
+        'predictions': response,
+        'metadata': repshape
+    })
     
     return jsonify(response) jsonify(repshape)