Update app.py

app.py

@@ -231,19 +231,32 @@ if file is not None and st.button("Predict Batch"):
         if resp_status == "success":
                         
             ## Get Sales Prediction Value
-            predictionS_from_backend = result.get ("predictions")  # Extract only the value
+            predictions_from_backend = result.get ("predictions")  # Extract only the value
 
-            # convert dict to dataframe for better display
-            result_df = pd.DataFrame(list(predictionS_from_backend.items()))
             st.dataframe (result_df)
+            # Convert list → DataFrame
+            result_df = pd.DataFrame({
+                "Prediction": predictions_from_backend
+            })
             
-            # generate output string
-            if prediction_from_backend == 1:
-                resultstr = "Engine **likely** needs maintenance."
-            else:
-                resultstr = "Engine does not need any maintenance"
+            st.dataframe (result_df)
+
+            input_df = pd.read_csv(file)
+
+            # Ensure lengths match
+            if len(predictions_from_backend) == len(input_df):
     
-            st.success(resultstr)
+                # Add prediction column
+                input_df["Prediction"] = predictions_from_backend
+    
+                st.success("Batch prediction completed successfully")
+    
+                # Show combined dataframe
+                st.dataframe(input_df, use_container_width=True)
+    
+            else:
+                st.error("Prediction count does not match input records")            
+
             
         else: