Update prediction.py

prediction.py

@@ -1,7 +1,74 @@
 import streamlit as st
-import pandas as pd 
+import pandas as pd
 import joblib
-import json
 
 def app():
-    st.title('Make Predictions')
+    st.title("Transaction Data Input")
+    st.write("Choose to upload a CSV file or manually input transaction data.")
+    
+    # Load pre-trained model
+    model = joblib.load('model.pkl')
+
+    # Option to choose upload or manual input
+    option = st.radio("Select input method:", ("Upload CSV", "Manual Input"))
+
+    if option == "Upload CSV":
+        # Option to upload a CSV file
+        file_upload = st.file_uploader("Upload CSV", type=["csv"])
+
+        if file_upload is not None:
+            data = pd.read_csv(file_upload)
+            st.write("Uploaded Data Preview:")
+            st.write(data.head())
+
+            if st.button("Submit CSV"):
+                # Predict using the uploaded CSV data
+                predictions = model.predict(data)
+                data['prediction'] = predictions
+                data['prediction'] = data['prediction'].map({1: 'Fraud Transactions', 0: 'Not Fraud Transactions'})
+
+                st.write("Predictions:")
+                st.write(data[['nameOrig', 'nameDest', 'prediction']])
+            
+    elif option == "Manual Input":
+        st.write("Manually input data:")
+        # Manual input of data
+        step = st.number_input("Step", min_value=0)
+        type = st.selectbox("Type", ["TRANSFER", "PAYMENT", "DEBIT", "CASH_OUT", "CASH_IN"])
+        amount = st.number_input("Amount", min_value=0.0)
+        nameOrig = st.text_input("Origin Account Name")
+        oldbalanceOrg = st.number_input("Old Balance (Origin)", min_value=0.0)
+        newbalanceOrig = st.number_input("New Balance (Origin)", min_value=0.0)
+        nameDest = st.text_input("Destination Account Name")
+        oldbalanceDest = st.number_input("Old Balance (Destination)", min_value=0.0)
+        newbalanceDest = st.number_input("New Balance (Destination)", min_value=0.0)
+        isFlaggedFraud = st.selectbox("Is Flagged Fraud?", [0, 1])
+
+        if st.button("Submit"):
+            # Create a DataFrame from manual input
+            manual_data = pd.DataFrame({
+                "step": [step],
+                "type": [type],
+                "amount": [amount],
+                "nameOrig": [nameOrig],
+                "oldbalanceOrg": [oldbalanceOrg],
+                "newbalanceOrig": [newbalanceOrig],
+                "nameDest": [nameDest],
+                "oldbalanceDest": [oldbalanceDest],
+                "newbalanceDest": [newbalanceDest],
+                "isFlaggedFraud": [isFlaggedFraud]
+            })
+
+            st.write("Manual Input Data:")
+            st.write(manual_data)
+
+            # Predict using the manually input data
+            manual_predictions = model.predict(manual_data)
+            manual_data['prediction'] = manual_predictions
+            manual_data['prediction'] = manual_data['prediction'].map({1: 'Fraud Transactions', 0: 'Not Fraud Transactions'})
+
+            st.write("Predictions:")
+            st.write(manual_data[['nameOrig', 'nameDest', 'prediction']])
+
+if __name__ == "__main__":
+    app()