Update app.py

app.py

@@ -1,87 +1,87 @@
-import gradio as gr
-import numpy as np
-import pandas as pd
-import pickle
-import matplotlib.pyplot as plt
-
-# Load pre-trained models and encoders
-with open('is_fraud.pkl', 'rb') as f:
-    is_fraud_model = pickle.load(f)
-
-with open('fraud_type.pkl', 'rb') as f:
-    fraud_type_model = pickle.load(f)
-
-# Load label encoders for features
-encoders = {}
-for feature in ['card_type', 'location', 'purchase_category', 'time_of_day']:
-    with open(f'{feature}.pkl', 'rb') as f:
-        encoders[feature] = pickle.load(f)
-
-with open('fraud_type_le.pkl', 'rb') as f:
-    fraud_type_encoder = pickle.load(f)
-
-# Define the prediction function
-def predict_fraud(amount, card_type, location, purchase_category, customer_age, time_of_day):
-    # Preprocess input
-    input_data = pd.DataFrame({
-        'amount': [amount],
-        'card_type': [card_type],
-        'location': [location],
-        'purchase_category': [purchase_category],
-        'customer_age': [customer_age],
-        'time_of_day': [time_of_day]
-    })
-
-    # Label encode the inputs
-    for feature in ['card_type', 'location', 'purchase_category', 'time_of_day']:
-        input_data[feature] = encoders[feature].transform(input_data[feature])
-
-    # Fraud Detection (Binary Classification)
-    is_fraud_pred_prob = is_fraud_model.predict_proba(input_data)[0]
-    is_fraud_class = np.argmax(is_fraud_pred_prob)
-
-    # Fraud Type Classification (Multiclass Classification)
-    fraud_type_pred_prob = fraud_type_model.predict_proba(input_data)[0]
-    fraud_type_class = fraud_type_encoder.inverse_transform([np.argmax(fraud_type_pred_prob)])[0]
-
-    # Bar charts for probabilities
-    fig, axes = plt.subplots(1, 2, figsize=(12, 5))
-
-    # Plot Fraud Detection Probabilities
-    axes[0].bar(['Not Fraudulent', 'Fraudulent'], is_fraud_pred_prob)
-    axes[0].set_title('Fraud Detection Probability')
-    axes[0].set_ylabel('Probability')
-
-    # Plot Fraud Type Probabilities
-    fraud_types = fraud_type_encoder.classes_
-    axes[1].bar(fraud_types, fraud_type_pred_prob)
-    axes[1].set_title('Fraud Type Probability')
-    axes[1].set_ylabel('Probability')
-    plt.xticks(rotation=45, ha="right")
-
-    plt.tight_layout()
-
-    return f"Fraud Detection{max(is_fraud_pred_prob)*100}: {'Fraudulent' if is_fraud_class == 1 else 'Not Fraudulent'}", f"Predicted Fraud Type: {fraud_type_class}", fig
-
-# Gradio Interface Setup
-interface = gr.Interface(
-    fn=predict_fraud,
-    inputs=[
-        gr.Number(label="Transaction Amount"),
-        gr.Dropdown(choices=['MasterCard', 'Visa' ,'Rupay'],label="Card Type"),
-        gr.Dropdown(choices=['Surat' ,'Hyderabad' ,'Kolkata' ,'Mumbai' ,'Delhi' ,'Chennai', 'Jaipur',
- 'Ahmedabad', 'Bangalore' ,'Pune'],label="Location"),
-        gr.Dropdown(choices=['POS' ,'Digital'],label="Purchase Category"),
-        gr.Number(label="Customer Age"),
-        gr.Dropdown(choices=['night', 'morning', 'afternoon', 'evening'],label="Time of Day")
-    ],
-    outputs=[
-        gr.Markdown(label="Fraud Detection Result"),
-        gr.Markdown(label="Fraud Type Result"),
-        gr.Plot(label="Probabilities Bar Chart")
-    ],
-    title="Fraud Detection & Fraud Type Prediction",
-    description="Enter transaction details to predict whether it is fraudulent and the type of fraud.",theme=gr.themes.Soft())
-
-# Launch Gradio Interface
-interface.launch()
+import gradio as gr
+import numpy as np
+import pandas as pd
+import pickle
+import matplotlib.pyplot as plt
+
+# Load pre-trained models and encoders
+with open('is_fraud.pkl', 'rb') as f:
+    is_fraud_model = pickle.load(f)
+
+with open('fraud_type.pkl', 'rb') as f:
+    fraud_type_model = pickle.load(f)
+
+# Load label encoders for features
+encoders = {}
+for feature in ['card_type', 'location', 'purchase_category', 'time_of_day']:
+    with open(f'{feature}.pkl', 'rb') as f:
+        encoders[feature] = pickle.load(f)
+
+with open('fraud_type_le.pkl', 'rb') as f:
+    fraud_type_encoder = pickle.load(f)
+
+# Define the prediction function
+def predict_fraud(amount, card_type, location, purchase_category, customer_age, time_of_day):
+    # Preprocess input
+    input_data = pd.DataFrame({
+        'amount': [amount],
+        'card_type': [card_type],
+        'location': [location],
+        'purchase_category': [purchase_category],
+        'customer_age': [customer_age],
+        'time_of_day': [time_of_day]
+    })
+
+    # Label encode the inputs
+    for feature in ['card_type', 'location', 'purchase_category', 'time_of_day']:
+        input_data[feature] = encoders[feature].transform(input_data[feature])
+
+    # Fraud Detection (Binary Classification)
+    is_fraud_pred_prob = is_fraud_model.predict_proba(input_data)[0]
+    is_fraud_class = np.argmax(is_fraud_pred_prob)
+
+    # Fraud Type Classification (Multiclass Classification)
+    fraud_type_pred_prob = fraud_type_model.predict_proba(input_data)[0]
+    fraud_type_class = fraud_type_encoder.inverse_transform([np.argmax(fraud_type_pred_prob)])[0]
+
+    # Bar charts for probabilities
+    fig, axes = plt.subplots(1, 2, figsize=(12, 5))
+
+    # Plot Fraud Detection Probabilities
+    axes[0].bar(['Not Fraudulent', 'Fraudulent'], is_fraud_pred_prob)
+    axes[0].set_title('Fraud Detection Probability')
+    axes[0].set_ylabel('Probability')
+
+    # Plot Fraud Type Probabilities
+    fraud_types = fraud_type_encoder.classes_
+    axes[1].bar(fraud_types, fraud_type_pred_prob)
+    axes[1].set_title('Fraud Type Probability')
+    axes[1].set_ylabel('Probability')
+    plt.xticks(rotation=45, ha="right")
+
+    plt.tight_layout()
+
+    return f"Fraud Detection{max(is_fraud_pred_prob)*100}: {'Fraudulent' if is_fraud_class == 1 else 'Not Fraudulent'}", f"Predicted Fraud Type: {fraud_type_class}", fig
+
+# Gradio Interface Setup
+interface = gr.Interface(
+    fn=predict_fraud,
+    inputs=[
+        gr.Number(label="Transaction Amount"),
+        gr.Dropdown(choices=['MasterCard', 'Visa' ,'Rupay'],label="Card Type"),
+        gr.Dropdown(choices=['Surat' ,'Hyderabad' ,'Kolkata' ,'Mumbai' ,'Delhi' ,'Chennai', 'Jaipur',
+ 'Ahmedabad', 'Bangalore' ,'Pune'],label="Location"),
+        gr.Dropdown(choices=['POS' ,'Digital'],label="Purchase Category"),
+        gr.Number(label="Customer Age"),
+        gr.Dropdown(choices=['night', 'morning', 'afternoon', 'evening'],label="Time of Day")
+    ],
+    outputs=[
+        gr.Markdown(label="Fraud Detection Result"),
+        gr.Markdown(label="Fraud Type Result"),
+        gr.Plot(label="Probabilities Bar Chart")
+    ],
+    title="Fraud Detection & Fraud Type Prediction",
+    description="By detecting fraudulent transactions, this model helps protect users and businesses in the Indian digital landscape, fostering a safer and more trustworthy online environment.",theme=gr.themes.Soft())
+
+# Launch Gradio Interface
+interface.launch()