Update app.py

app.py

@@ -77,7 +77,8 @@ def predict_sales_total_batch():
 
     # Create a dictionary of predictions with property IDs as keys
     product_store_ids = input_data[['Product_Id', 'Store_Id']].values.tolist()
-    output_dict = dict(zip(product_store_ids, predicted_prices))  # Use actual prices
+    keys = [f"{pid}_{sid}" for pid, sid in product_store_ids]
+    output_dict = dict(zip(keys, predicted_prices))  # Use actual prices
 
     # Return the predictions dictionary as a JSON response
     return output_dict