Upload app.py

app.py

@@ -14,14 +14,14 @@ from pyngrok import conf, ngrok
 import requests
 
 # Manually set ngrok path
-# ngrok_path = "/Users/ninadmandavkar/Desktop/ngrok/ngrok 3"
+ngrok_path = "/Users/ninadmandavkar/Desktop/ngrok/ngrok 3"
 
 # Set the ngrok path in the configuration
-# config = conf.get_default()
-# config.ngrok_path = ngrok_path
+config = conf.get_default()
+config.ngrok_path = ngrok_path
 
 # Set your ngrok auth token
-# ngrok.set_auth_token('2jQ6LRMBFqe25TP7o2ugti9EH6Y_3mBzBYbsjZmhh69eGq2Q1')
+ngrok.set_auth_token('2jQ6LRMBFqe25TP7o2ugti9EH6Y_3mBzBYbsjZmhh69eGq2Q1')
 
 # logging.basicConfig(level=logging.INFO)
 
@@ -35,6 +35,8 @@ global_user_data = []  # For appending user_data
 global_data_rows = [] # For appending data rows
 
 
+
+
 # Function to initialize session state
 def initialize_session_state():
     if 'data' not in st.session_state:
@@ -111,6 +113,9 @@ def generate_plan_id(user_data):
 
     company_id = user_data.get('Company ID', '')  # Get company_id from user_data
 
+    if company_id is None or not company_id.strip():
+        company_id = ''
+
     # Increment serial counter and reset if it reaches 999999
     serial_counter += 1
     if serial_counter > 999999:
@@ -292,39 +297,48 @@ html_content_1 = """
 
 st.sidebar.markdown(html_content_1, unsafe_allow_html=True)
 
-
+if 'company_name' not in st.session_state:
+    st.session_state['company_name'] = "Company name not available"
 
 v1 = ["Commerce India", "Reliance", "Commerce Global", "Government Projects", "Individual BH"]
 business_head = st.sidebar.selectbox("Business Head", [""] + v1, help="Business Head is used for cost centre allocation")
+
+# Initialize company_name in session state
+if 'company_name' not in st.session_state:
+    st.session_state['company_name'] = ""
+
+# User inputs Company ID
 company_id = st.sidebar.text_input("Company ID", help="Enter the company ID")
-if company_id:
+company_id_int = None
+
+if company_id.strip():
     try:
-        # Construct the API URL with the provided cmpid
-        api_url = f"https://api.boltic.io/service/platform/bolt/share/v1/data?api_token=eyJpZCI6IjU4OTdmZjZlLTMwMTYtNDZjMC1hMzdlLTJmNjBkMjhjYjc5YyIsInRlbmFudCI6ImdvZnluZGNvbSJ9&batch_size=10000&display_meta=true&cmpid={company_id}"
+        company_id_int = int(company_id.strip())
+    except ValueError:
+        st.error("Invalid Company ID. Please enter a valid integer.")
+        company_id_int = None
 
-        # Fetch data from the API
+# Construct the API URL and fetch data only if company_id_int is valid
+if company_id_int is not None:
+    try:
+        api_url = f"https://api.boltic.io/service/platform/bolt/share/v1/data?api_token=eyJpZCI6IjU4OTdmZjZlLTMwMTYtNDZjMC1hMzdlLTJmNjBkMjhjYjc5YyIsInRlbmFudCI6ImdvZnluZGNvbSJ9&batch_size=10000&display_meta=true&cmpid={company_id_int}"
         response = requests.get(api_url)
 
-        # Check if the request was successful
         if response.status_code == 200:
             data = response.json()
-
-            # Extract company name from the data
             if "results" in data and len(data["results"]) > 0:
-                company_name = data["results"][0].get("company_name", "Company name not found")
+                st.session_state['company_name'] = data["results"][0].get("company_name", "Company name not found")
             else:
-                company_name = "Company name not found"
+                st.session_state['company_name'] = "Company name not found"
         else:
-            company_name = f"Error: Unable to fetch data (Status code: {response.status_code})"
+            st.session_state['company_name'] = f"Error: Unable to fetch data (Status code: {response.status_code})"
     except Exception as e:
-        company_name = f"An error occurred: {e}"
-
-
-    # Display the company name
-    st.sidebar.text_input("Company Name", value=company_name)
+        st.session_state['company_name'] = f"An error occurred: {e}"
 else:
-    st.sidebar.text_input("Company Name", value="")
+    st.session_state['company_name'] = ""  # Set to an empty string if Company ID is not provided or invalid
 
+# Display the company name
+st.sidebar.text_input("Company Name", value=st.session_state['company_name'], disabled=False)
 
 currency = st.sidebar.radio("Currency", options= ["INR", "USD"], help= "Select the type of currency", index =None)
 bundle_by = st.sidebar.radio(options=["Single value", "Feature specific"], label="Bundle by", help = "Select the bundle", index=None)
@@ -348,313 +362,529 @@ if bundle_by == "Feature specific":
     chargeable_on_options = ["Active", "Build", "Cancelled", "Delivered", "DTO", "Invoiced", "Packed", "Picked", "Placed", "Return Window", "RTO", "Setup", "SiteOps-hr", "Subscribed", "TechOps-hr"]
     plan_validity_options = ["One time", "Monthly", "Quarterly", "Bi-Annually", "Annually"]
     payment_method_options = ["Prepaid", "Postpaid"]
+ 
+
+    # Initialize session state variables if they don't exist
+    if 'new_mapping' not in st.session_state:
+        st.session_state['new_mapping'] = {}
+    if 'file_data' not in st.session_state:
+        st.session_state['file_data'] = None
+    if 'mapping_mode' not in st.session_state:
+        st.session_state['mapping_mode'] = None  # None, 'default', 'custom'
+    if 'button_clicked' not in st.session_state:
+        st.session_state['button_clicked'] = None
+
+    # Function to handle custom button click
+    def custom_button_click():
+        st.session_state.button_clicked = 'custom'
+
+    # Function to handle default button click
+    def default_button_click():
+        st.session_state.button_clicked = 'default'
 
-    # Given new_mapping dictionary
-    new_mapping = {
-        "GoFynd": {
-            "Marketing": {
-                "Order": ["Placed"]
-            },
-            "Transaction": {
-                "Bag": ["Return Window"],
-                "Shipment": ["Cancelled", "RTO", "DTO"]
-            },
-            "Logistics": {
-                "Shipment": ["Picked", "RTO", "DTO"]
-            },
-            "Packaging": {
-                "Shipment": ["Packed"]
-            }
-        },
-        "Uniket": {
-            "Marketing": {
-                "Order": ["Placed"]
-            },
-            "Transaction": {
-                "Bag": ["Return Window"],
-                "Shipment": ["Cancelled", "RTO", "DTO"]
-            },
-            "Logistics": {
-                "Shipment": ["Picked", "RTO", "DTO"]
-            },
-            "Packaging": {
-                "Shipment": ["Packed"]
-            }
-        },
-        "ONDC": {
-            "Transaction": {
-                "Bag": ["Return Window"],
-                "Shipment": ["Cancelled", "RTO", "DTO"]
-            },
-            "Logistics": {
-                "Shipment": ["Picked", "RTO", "DTO"]
-            },
-            "Packaging": {
-                "Shipment": ["Packed"]
-            }
-        },
-        "StoreOS": {
-            "Licensing": {
-                "Application": ["Setup"]
-            },
-            "Subscription": {
-                "Application": ["Subscribed"],
-                "User": ["Active"],
-                "Resource": ["TechOps-hr", "SiteOps-hr"]
-            },
-            "Marketing": {
-                "Order": ["Placed"]
-            },
-            "Transaction": {
-                "Order": ["Placed"],
-                "Bag": ["Invoiced", "Delivered"]
-            },
-            "Logistics": {
-                "Shipment": ["Picked", "RTO", "DTO"]
-            },
-            "Packaging": {
-                "Shipment": ["Packed"]
-            }
-        },
-        "Website": {
-            "Licensing": {
-                "Application": ["Setup"]
-            },
-            "Subscription": {
-                "Application": ["Subscribed"],
-                "User": ["Active"],
-                "Resource": ["TechOps-hr", "SiteOps-hr"]
-            },
-            "Transaction": {
-                "Order": ["Placed"],
-                "Bag": ["Invoiced", "Delivered"]
-            },
-            "Logistics": {
-                "Shipment": ["Picked", "RTO", "DTO"]
-            },
-            "Packaging": {
-                "Shipment": ["Packed"]
-            },
-            "Development": {
-                "Application": ["Build"]
-            }
-        },
-        "OMS": {
-            "Development": {
-                "Integration": ["Build"]
-            },
-            "Licensing": {
-                "Integration": ["Setup"]
-            },
-            "Subscription": {
-                "Integration": ["Subscribed"],
-                "User": ["Active"],
-                "Resource": ["TechOps-hr", "SiteOps-hr"]
-            },
-            "Transaction": {
-                "Order": ["Placed"],
-                "Bag": ["Invoiced", "Delivered"]
-            }
-        },
-        "WMS": {
-            "Development": {
-                "Integration": ["Build"]
-            },
-            "Licensing": {
-                "Integration": ["Setup"]
-            },
-            "Subscription": {
-                "Integration": ["Subscribed"],
-                "User": ["Active"],
-                "Resource": ["TechOps-hr", "SiteOps-hr"]
-            },
-            "Transaction": {
-                "Order": ["Placed"],
-                "Bag": ["Invoiced", "Delivered"],
-                "B2B": ["Invoiced"],
-                "B2C": ["Invoiced"]
-            }
-        },
-        "TMS": {
-            "Development": {
-                "Integration": ["Build"]
-            },
-            "Licensing": {
-                "Integration": ["Setup"]
-            },
-            "Subscription": {
-                "Integration": ["Subscribed"],
-                "User": ["Active"],
-                "Resource": ["TechOps-hr", "SiteOps-hr"]
-            },
-            "Logistics": {
-                "Shipment": ["Picked", "RTO", "DTO"]
-            }
-        },
-        "GMC": {
-            "Development": {
-                "Extension": ["Build"],
-                "Integration": ["Build"]
-            },
-            "Licensing": {
-                "Extension": ["Setup"]
-            },
-            "Subscription": {
-                "Extension": ["Subscribed"],
-                "User": ["Active"],
-                "Resource": ["TechOps-hr", "SiteOps-hr"]
-            }
-        },
-        "Partner": {
-            "Development": {
-                "Extension": ["Build"]
-            },
-            "Licensing": {
-                "Extension": ["Setup"]
-            },
-            "Subscription": {
-                "Extension": ["Subscribed"],
-                "User": ["Active"],
-                "Resource": ["TechOps-hr", "SiteOps-hr"]
-            }
-        },
-        "Catalog Cloud": {
-            "Development": {
-                "Extension": ["Build"]
-            },
-            "Licensing": {
-                "Extension": ["Setup"]
-            },
-            "Subscription": {
-                "Extension": ["Subscribed"],
-                "User": ["Active"],
-                "Resource": ["TechOps-hr", "SiteOps-hr"]
-            }
-        },
-        "FCP": {
-            "Subscription": {
-                "Platform": ["Subscribed"],
-                "User": ["Active"],
-                "Resource": ["TechOps-hr", "SiteOps-hr"]
-            }
-        },
-        "PixelBin": {
-            "Subscription": {
-                "Platform": ["Subscribed"],
-                "User": ["Active"],
-                "Resource": ["TechOps-hr", "SiteOps-hr"]
+    
+    # Display buttons without immediate execution on click
+    col1, col2 = st.sidebar.columns([0.25, 0.45])
+    with col1:
+        default_clicked = st.button("Default mapping")
+    with col2:
+        custom_clicked = st.button("Custom mapping")
+
+    # Determine action based on button clicked
+    if default_clicked:
+        default_button_click()
+    elif custom_clicked:
+        custom_button_click()
+
+     # Define CSS styles for the button
+    button_styles = """
+    <style>
+    div.stButton > button {
+        color: #ffffff; /* Text color */
+        font-size: 50px;
+        background-image: linear-gradient(0deg, #a689f6 3%, #413bb9 61%);
+        border: none;
+        padding: 10px 20px;
+        cursor: pointer;
+        border-radius: 15px;
+        display: inline-block;
+    }
+    div.stButton > button:hover {
+        background-color: #00ff00; /* Hover background color */
+        color: #ff0000; /* Hover text color */
+    }
+    </style>
+    """
+
+# Render the button with the specified styles
+    st.markdown(button_styles, unsafe_allow_html=True)
+
+
+    # Handle Default Mapping
+    if st.session_state.button_clicked == 'default':
+        # Default mapping data
+        new_mapping = {
+            "GoFynd": {
+                "Marketing": {
+                    "Order": ["Placed"]
+                },
+                "Transaction": {
+                    "Bag": ["Return Window"],
+                    "Shipment": ["Cancelled", "RTO", "DTO"]
+                },
+                "Logistics": {
+                    "Shipment": ["Picked", "RTO", "DTO"]
+                },
+                "Packaging": {
+                    "Shipment": ["Packed"]
+                }
+            },
+            "Uniket": {
+                "Marketing": {
+                    "Order": ["Placed"]
+                },
+                "Transaction": {
+                    "Bag": ["Return Window"],
+                    "Shipment": ["Cancelled", "RTO", "DTO"]
+                },
+                "Logistics": {
+                    "Shipment": ["Picked", "RTO", "DTO"]
+                },
+                "Packaging": {
+                    "Shipment": ["Packed"]
+                }
+            },
+            "ONDC": {
+                "Transaction": {
+                    "Bag": ["Return Window"],
+                    "Shipment": ["Cancelled", "RTO", "DTO"]
+                },
+                "Logistics": {
+                    "Shipment": ["Picked", "RTO", "DTO"]
+                },
+                "Packaging": {
+                    "Shipment": ["Packed"]
+                }
+            },
+            "StoreOS": {
+                "Licensing": {
+                    "Application": ["Setup"]
+                },
+                "Subscription": {
+                    "Application": ["Subscribed"],
+                    "User": ["Active"],
+                    "Resource": ["TechOps-hr", "SiteOps-hr"]
+                },
+                "Marketing": {
+                    "Order": ["Placed"]
+                },
+                "Transaction": {
+                    "Order": ["Placed"],
+                    "Bag": ["Invoiced", "Delivered"]
+                },
+                "Logistics": {
+                    "Shipment": ["Picked", "RTO", "DTO"]
+                },
+                "Packaging": {
+                    "Shipment": ["Packed"]
+                }
+            },
+            "Website": {
+                "Licensing": {
+                    "Application": ["Setup"]
+                },
+                "Subscription": {
+                    "Application": ["Subscribed"],
+                    "User": ["Active"],
+                    "Resource": ["TechOps-hr", "SiteOps-hr"]
+                },
+                "Transaction": {
+                    "Order": ["Placed"],
+                    "Bag": ["Invoiced", "Delivered"]
+                },
+                "Logistics": {
+                    "Shipment": ["Picked", "RTO", "DTO"]
+                },
+                "Packaging": {
+                    "Shipment": ["Packed"]
+                },
+                "Development": {
+                    "Application": ["Build"]
+                }
+            },
+            "OMS": {
+                "Development": {
+                    "Integration": ["Build"]
+                },
+                "Licensing": {
+                    "Integration": ["Setup"]
+                },
+                "Subscription": {
+                    "Integration": ["Subscribed"],
+                    "User": ["Active"],
+                    "Resource": ["TechOps-hr", "SiteOps-hr"]
+                },
+                "Transaction": {
+                    "Order": ["Placed"],
+                    "Bag": ["Invoiced", "Delivered"]
+                }
+            },
+            "WMS": {
+                "Development": {
+                    "Integration": ["Build"]
+                },
+                "Licensing": {
+                    "Integration": ["Setup"]
+                },
+                "Subscription": {
+                    "Integration": ["Subscribed"],
+                    "User": ["Active"],
+                    "Resource": ["TechOps-hr", "SiteOps-hr"]
+                },
+                "Transaction": {
+                    "Order": ["Placed"],
+                    "Bag": ["Invoiced", "Delivered"],
+                    "B2B": ["Invoiced"],
+                    "B2C": ["Invoiced"]
+                }
+            },
+            "TMS": {
+                "Development": {
+                    "Integration": ["Build"]
+                },
+                "Licensing": {
+                    "Integration": ["Setup"]
+                },
+                "Subscription": {
+                    "Integration": ["Subscribed"],
+                    "User": ["Active"],
+                    "Resource": ["TechOps-hr", "SiteOps-hr"]
+                },
+                "Logistics": {
+                    "Shipment": ["Picked", "RTO", "DTO"]
+                }
+            },
+            "GMC": {
+                "Development": {
+                    "Extension": ["Build"],
+                    "Integration": ["Build"]
+                },
+                "Licensing": {
+                    "Extension": ["Setup"]
+                },
+                "Subscription": {
+                    "Extension": ["Subscribed"],
+                    "User": ["Active"],
+                    "Resource": ["TechOps-hr", "SiteOps-hr"]
+                }
+            },
+            "Partner": {
+                "Development": {
+                    "Extension": ["Build"]
+                },
+                "Licensing": {
+                    "Extension": ["Setup"]
+                },
+                "Subscription": {
+                    "Extension": ["Subscribed"],
+                    "User": ["Active"],
+                    "Resource": ["TechOps-hr", "SiteOps-hr"]
+                }
+            },
+            "Catalog Cloud": {
+                "Development": {
+                    "Extension": ["Build"]
+                },
+                "Licensing": {
+                    "Extension": ["Setup"]
+                },
+                "Subscription": {
+                    "Extension": ["Subscribed"],
+                    "User": ["Active"],
+                    "Resource": ["TechOps-hr", "SiteOps-hr"]
+                }
+            },
+            "FCP": {
+                "Subscription": {
+                    "Platform": ["Subscribed"],
+                    "User": ["Active"],
+                    "Resource": ["TechOps-hr", "SiteOps-hr"]
+                }
+            },
+            "PixelBin": {
+                "Subscription": {
+                    "Platform": ["Subscribed"],
+                    "User": ["Active"],
+                    "Resource": ["TechOps-hr", "SiteOps-hr"]
+                }
+            },
+            "Boltic": {
+                "Subscription": {
+                    "Platform": ["Subscribed"],
+                    "User": ["Active"],
+                    "Resource": ["TechOps-hr", "SiteOps-hr"]
+                }
+            },
+            "CoPilot": {
+                "Subscription": {
+                    "Platform": ["Subscribed"],
+                    "User": ["Active"],
+                    "Resource": ["TechOps-hr", "SiteOps-hr"]
+                }
             }
-        },
-        "Boltic": {
-            "Subscription": {
-                "Platform": ["Subscribed"],
-                "User": ["Active"],
-                "Resource": ["TechOps-hr", "SiteOps-hr"]
+        }
+        
+        st.session_state['new_mapping'] = new_mapping
+        st.session_state['mapping_mode'] = 'default'
+
+        html_content_1 = """
+                            <h1 style='
+                                color: #a689f6;
+                                font-size: 22px;
+                                background-image: -webkit-linear-gradient(0deg, #a689f6 3%, #272191 33%, #413bb9 61%);
+                                background-clip: text;
+                                -webkit-background-clip: text;
+                                text-fill-color: transparent;
+                                -webkit-text-fill-color: transparent;
+                            '>Section 2: Rule Info</h1>
+                        """
+
+        st.sidebar.markdown(html_content_1, unsafe_allow_html=True)
+
+    # Handle Custom Mapping
+    elif st.session_state.button_clicked == 'custom':
+
+        # Define CSS styles for the button
+        button_styles = """
+            <style>
+            div.stButton > button {
+                color: #ffffff; /* Text color */
+                font-size: 50px;
+                background-image: linear-gradient(0deg, #a689f6 3%, #413bb9 61%);
+                border: none;
+                padding: 10px 20px;
+                cursor: pointer;
+                border-radius: 15px;
+                display: inline-block;
             }
-        },
-        "CoPilot": {
-            "Subscription": {
-                "Platform": ["Subscribed"],
-                "User": ["Active"],
-                "Resource": ["TechOps-hr", "SiteOps-hr"]
+            div.stButton > button:hover {
+                background-color: #00ff00; /* Hover background color */
+                color: #ff0000; /* Hover text color */
             }
-        }
-    }
+            </style>
+            """
 
-    # Initialize mappings
-    fee_type_mapping = {}
-    variable_type_mapping = {}
-    chargeable_on_mapping = {}
-
-    # Populate mappings from new_mapping
-    for ordering_channel, details in new_mapping.items():
-        for fee_type, variables in details.items():
-            if fee_type not in fee_type_mapping:
-                fee_type_mapping[fee_type] = []
-            if fee_type not in variable_type_mapping:
-                variable_type_mapping[fee_type] = []
-            variable_types = list(variables.keys())
-            variable_type_mapping[fee_type].extend(variable_types)
-            for variable_type, chargeable_on_list in variables.items():
-                if variable_type not in chargeable_on_mapping:
-                    chargeable_on_mapping[variable_type] = []
-                chargeable_on_mapping[variable_type].extend(chargeable_on_list)
-
-    # Remove duplicates in mappings
-    fee_type_mapping = {k: list(set(v)) for k, v in fee_type_mapping.items()}
-    variable_type_mapping = {k: list(set(v)) for k, v in variable_type_mapping.items()}
-    chargeable_on_mapping = {k: list(set(v)) for k, v in chargeable_on_mapping.items()}
-
-
-     # Section 2: Company Info
-    
-    html_content_3 = """
-        <h1 style='
-            color: #a689f6;
-            font-size: 22px;
-            background-image: -webkit-linear-gradient(0deg, #a689f6 3%, #272191 33%, #413bb9 61%);
-            background-clip: text;
-            -webkit-background-clip: text;
-            text-fill-color: transparent;
-            -webkit-text-fill-color: transparent;
-        '>Section 2: Rule Info</h1>
-    """
+# Render the button with the specified styles
+        st.markdown(button_styles, unsafe_allow_html=True)
+
+        # File upload
+        uploaded_file = st.sidebar.file_uploader("Upload a file:", type=['csv', 'xlsx'], key='file_uploader')
+
+        html_content_1 = """
+                            <h1 style='
+                                color: #a689f6;
+                                font-size: 22px;
+                                background-image: -webkit-linear-gradient(0deg, #a689f6 3%, #272191 33%, #413bb9 61%);
+                                background-clip: text;
+                                -webkit-background-clip: text;
+                                text-fill-color: transparent;
+                                -webkit-text-fill-color: transparent;
+                            '>Section 2: Rule Info</h1>
+                        """
+
+        st.sidebar.markdown(html_content_1, unsafe_allow_html=True)
+
+        if uploaded_file is not None:
+            try:
+                # Read the file, treating the first row as the header
+                if uploaded_file.name.endswith('.csv'):
+                    data = pd.read_csv(uploaded_file, header=0)
+                elif uploaded_file.name.endswith('.xlsx'):
+                    data = pd.read_excel(uploaded_file, header=0)
 
-    st.sidebar.markdown(html_content_3, unsafe_allow_html=True)
+                if st.sidebar.button("Overwrite headers"):
+                    # Rename the headers
+                    new_headers = ['ordering_channel', 'fee_type', 'variable_type', 'chargeable_on']
+                    data.columns = new_headers
 
-    # 1st layer: Select ordering channel
-    selected_ordering_channel = st.sidebar.selectbox(
-        "Product Lines",
-        [""] + list(new_mapping.keys()),
-        help="Select a product line"
-    )
+                    # Save the data to session state
+                    st.session_state['file_data'] = data
 
-    # Layer 1: Mapping of Ordering channels with Fulfilling location
-    if "TMS" in selected_ordering_channel or "GMC" in selected_ordering_channel or "Catalog Cloud" in selected_ordering_channel:
-        fulfilling_location = None
-        application_id = None
-    else:
-        abc = ["Store", "Warehouse"]
-        fulfilling_location = st.sidebar.selectbox("Fulfilling Location", [""] + abc, help="Select the fulfilling location")
-        application_id = st.sidebar.text_input("Application ID", key="application_id_input", help="Enter the application ID")
-
-    # 2nd layer: Select fee type based on the selected ordering channel
-    fee_types_for_channel = []
-    if selected_ordering_channel:
-        details = new_mapping.get(selected_ordering_channel, {})
-        fee_types_for_channel = list(details.keys())
-
-    selected_fee_type = st.selectbox(
-        "Fee Type",
-        [""] + fee_types_for_channel,
-        help="Fee type is a revenue head for grouping the revenue"
-    )
+                    # Initialize the mapping
+                    new_mapping = {}
 
-    # 3rd layer: Select variable type based on the selected fee type
-    variable_type_options = []
-    if selected_fee_type and selected_ordering_channel:
-        details = new_mapping.get(selected_ordering_channel, {})
-        variables = details.get(selected_fee_type, {})
-        variable_type_options = list(variables.keys())
-
-    selected_variable_type = st.selectbox(
-        "Variable Type",
-        [""] + variable_type_options,
-        help="Variable type is an attribute on which revenue calculation will be done"
-    )
+                    
+
+                    # Build the mapping
+                    for _, row in data.iterrows():
+
+                
+                        ordering_channel = row['ordering_channel']
+                        fee_type = row['fee_type']
+                        variable_type = row['variable_type']
+                        chargeable_on = row['chargeable_on']
+
+                        if ordering_channel not in new_mapping:
+                            new_mapping[ordering_channel] = {}
+
+                        if fee_type not in new_mapping[ordering_channel]:
+                            new_mapping[ordering_channel][fee_type] = {}
+
+                        if variable_type not in new_mapping[ordering_channel][fee_type]:
+                            new_mapping[ordering_channel][fee_type][variable_type] = []
+
+                        if chargeable_on not in new_mapping[ordering_channel][fee_type][variable_type]:
+                            new_mapping[ordering_channel][fee_type][variable_type].append(chargeable_on)
+
+                    st.session_state['new_mapping'] = new_mapping
+                    st.session_state['mapping_mode'] = 'custom'
+
+            except Exception as e:
+                st.error(f"Error reading the file: {e}")
+
+    # Handle mapping display
+   # Initialize session state variables if they don't exist
+    # Initialize session state variables with default values if not already set
+    if 'business_head' not in st.session_state:
+        st.session_state['business_head'] = None
+    if 'company_id' not in st.session_state:
+        st.session_state['company_id'] = None
+    if 'company_name' not in st.session_state:
+        st.session_state['company_name'] = None
+    if 'currency' not in st.session_state:
+        st.session_state['currency'] = None
+    if 'bundle_by' not in st.session_state:
+        st.session_state['bundle_by'] = None
+    if 'plan_name' not in st.session_state:
+        st.session_state['plan_name'] = None
+    if 'plan_description' not in st.session_state:
+        st.session_state['plan_description'] = None
+    if 'selected_ordering_channel' not in st.session_state:
+        st.session_state['selected_ordering_channel'] = None
+    if 'fulfilling_location' not in st.session_state:
+        st.session_state['fulfilling_location'] = None
+    if 'application_id' not in st.session_state:
+        st.session_state['application_id'] = None
+    if 'selected_fee_type' not in st.session_state:
+        st.session_state['selected_fee_type'] = None
+    if 'selected_variable_type' not in st.session_state:
+        st.session_state['selected_variable_type'] = None
+    if 'selected_chargeable_on' not in st.session_state:
+        st.session_state['selected_chargeable_on'] = None
+    if 'selected_fee_nature' not in st.session_state:
+        st.session_state['selected_fee_nature'] = None
+    if 'user_input_3' not in st.session_state:
+        st.session_state['user_input_3'] = None
+    if 'usage_3' not in st.session_state:
+        st.session_state['usage_3'] = None
+    if 'Capping_or_Minimum_Guarantee_3' not in st.session_state:
+        st.session_state['Capping_or_Minimum_Guarantee_3'] = None
+    if 'threshold' not in st.session_state:
+        st.session_state['threshold'] = None
+    if 'expected_billing' not in st.session_state:
+        st.session_state['expected_billing'] = None
+    if 'selected_payment_method' not in st.session_state:
+        st.session_state['selected_payment_method'] = None
+    if 'selected_plan_validity' not in st.session_state:
+        st.session_state['selected_plan_validity'] = None
+    if 'final_billing' not in st.session_state:
+        st.session_state['final_billing'] = None
+    if 'fee_reversal' not in st.session_state:
+        st.session_state['fee_reversal'] = None
+    if 'reversal_per' not in st.session_state:
+        st.session_state['reversal_per'] = None
+
+    # Check if 'mapping_mode' exists in session state before using it
+    if 'mapping_mode' in st.session_state and st.session_state['mapping_mode']:
+        # 1st layer: Select ordering channel
+        selected_ordering_channel = st.sidebar.selectbox(
+            "Product lines",
+            [""] + list(st.session_state['new_mapping'].keys()),
+            help="Select a Product line",
+            key='ordering_channel'
+        )
+
+        if selected_ordering_channel:
+            selected_ordering_channel = str(selected_ordering_channel) or int(selected_ordering_channel)
+
+            # Determine fulfilling_location and application_id based on selected_ordering_channel
+            if any(channel in selected_ordering_channel for channel in ["TMS", "GMC", "Catalog Cloud"]):
+                st.session_state['fulfilling_location'] = None
+                st.session_state['application_id'] = None
+            else:
+                abc = ["Store", "Warehouse"]
+                st.session_state['fulfilling_location'] = st.sidebar.selectbox(
+                    "Fulfilling Location", [""] + abc, help="Select the fulfilling location"
+                )
+                st.session_state['application_id'] = st.sidebar.text_input(
+                    "Application ID", key="application_id_input", help="Enter the application ID"
+                )
+
+            # Use the session state variables in your app
+            fulfilling_location = st.session_state['fulfilling_location']
+            application_id = st.session_state['application_id']
+
+
+
+            # 2nd layer: Select fee type based on the selected ordering channel
+
+            if 'selected_fee_type' not in st.session_state:
+                st.session_state['selected_fee_type'] = None
+
+            fee_types_for_channel = st.session_state['new_mapping'].get(selected_ordering_channel, {}).keys()
+            fee_types_for_channel = list(fee_types_for_channel)
+
+            if fee_types_for_channel:
+
+
+                # Define selected_fee_type based on user input
+
+                st.session_state['selected_fee_type'] = st.selectbox(
+                    "Fee Type",
+                    [""] + fee_types_for_channel,
+                    help="Fee type is a revenue head for grouping the revenue",
+                    key='fee_type'
+                )
+
+                # Access selected_fee_type for any session state operations
+                selected_fee_type = st.session_state['selected_fee_type']
+
+                # 3rd layer: Select variable type based on the selected fee type
+                variable_type_options = []
+                if selected_fee_type:
+                    details = st.session_state['new_mapping'].get(selected_ordering_channel, {})
+                    variables = details.get(selected_fee_type, {})
+                    variable_type_options = list(variables.keys())
+
+                st.session_state['selected_variable_type'] = st.selectbox(
+                    "Variable Type",
+                    [""] + variable_type_options,
+                    help="Variable type is an attribute on which revenue calculation will be done",
+                    key='variable_type'
+                )
+
+                selected_variable_type = st.session_state['selected_variable_type']
+
+            
+                # 4th layer: Select chargeable on based on the selected variable type
+                chargeable_on_options = []
+                if selected_variable_type:
+                    chargeable_on_options = variables.get(selected_variable_type, [])
+
+                st.session_state['selected_chargeable_on'] = st.selectbox(
+                    "Chargeable on",
+                    [""] + chargeable_on_options,
+                    help="Chargeable on is a service status on which the above fee will be applicable",
+                    key='chargeable_on'
+                )
+
+                selected_chargeable_on = st.session_state['selected_chargeable_on']
+
+            else:
+                st.write("No fee types available for the selected ordering channel.")
+        else:
+            st.write("")
 
-    # 4th layer: Select chargeable on based on the selected variable type
-    chargeable_on_options = []
-    if selected_variable_type and selected_fee_type and selected_ordering_channel:
-        details = new_mapping.get(selected_ordering_channel, {})
-        variables = details.get(selected_fee_type, {})
-        chargeable_on_options = variables.get(selected_variable_type, [])
-
-    selected_chargeable_on = st.selectbox(
-    "Chargeable on",
-    [""] + chargeable_on_options,
-    help="Chargeable on is a service status on which the above fee will be applicable"
-    )
 
 # Create an empty DataFrame with the desired column names
     slabs_df = pd.DataFrame(columns=["Slab", "Max_value", "Commercial_value", "Usage", "Capping/Min_Guarantee_value", "Threshold"])
@@ -743,6 +973,10 @@ if bundle_by == "Feature specific":
     
 
     # 6th layer: Expected Billing
+
+        selected_fee_type = st.session_state['selected_fee_type']
+        selected_variable_type = st.session_state['selected_variable_type']
+
         fee_reversal_options = ["RTO", "DTO", "Cancel"]
         fee_reversal = st.multiselect("Fee Reversal", fee_reversal_options, help = "Fee reversal is an option of reversing to be given to any fee on a particular service status") if selected_fee_type == "Transaction" and selected_variable_type == "Bag" else None
         reversal_per = st.number_input("Reversal %", min_value= 0.0, help = "Enter the reversal percentage") if selected_fee_type == "Transaction" and selected_variable_type == "Bag" else None
@@ -777,6 +1011,7 @@ if bundle_by == "Feature specific":
         else:
             expected_billing = initial_expected_billing # Default to initial Product value if no threshold selected
 
+        plan_validity_options = ["One time", "Monthly", "Quarterly", "Bi-Annually", "Annually"]
         selected_plan_validity = st.selectbox("Plan Validity", [""] + plan_validity_options, help="Plan validity defines the periodicity of calculation for the above billing", key="plan_vd_2")
 
         html_content_7 = f"""
@@ -825,7 +1060,7 @@ if bundle_by == "Feature specific":
         st.markdown(html_content_8, unsafe_allow_html=True)
 
 
-   
+    payment_method_options = ["Prepaid", "Postpaid"]
     selected_payment_method = st.selectbox("Payment Method", [""] + payment_method_options, help="Payment method defines the mode of payment of the plan", key="py_vd_2")
 
     
@@ -857,12 +1092,35 @@ if bundle_by == "Feature specific":
 
     rule_id = None
     plan_id = None 
+    business_head = st.session_state['business_head']
+    company_name = st.session_state['company_name']
+    company_id = st.session_state['company_id']
+    currency = st.session_state['currency']
+    # bundle_by = st.session_state['bundle_by']
+    plan_name = st.session_state['plan_name']
+    plan_description = st.session_state['plan_description']
+    selected_ordering_channel = st.session_state['selected_ordering_channel']
+    fulfilling_location = st.session_state['fulfilling_location']
+    application_id = st.session_state['application_id']
+    selected_fee_type = st.session_state['selected_fee_type']
+    selected_variable_type = st.session_state['selected_variable_type']
+    selected_chargeable_on = st.session_state['selected_chargeable_on']
+    user_input_3 = st.session_state['user_input_3']
+    usage_3 = st.session_state['usage_3']
+    Capping_or_Minimum_Guarantee_3 = st.session_state['Capping_or_Minimum_Guarantee_3']
+    fee_reversal = st.session_state['fee_reversal']
+    reversal_per = st.session_state['reversal_per']
+    threshold = st.session_state['threshold']
+    expected_billing = st.session_state['expected_billing']
+    selected_payment_method = st.session_state['selected_payment_method']
+    selected_plan_validity = st.session_state['selected_plan_validity']
+    final_billing = int(st.session_state['final_billing']) if st.session_state['final_billing'] is not None else 0
 
     if st.button("Submit", key ='submit_button'):
     # Save user data
         user_data = {
         "Business_Head": business_head,
-        "Company_ID": company_id,
+        "Company_ID": company_id.strip() if company_id else None,
         "Company_Name": company_name,
         "Currency": currency,
         "Bundle_by": bundle_by,
@@ -935,6 +1193,12 @@ if bundle_by == "Feature specific":
         project_id = "fynd-db"
         dataset_id = "finance_dwh"
         table_id = "plan_v2"
+        fulfilling_location = st.session_state['fulfilling_location']
+        application_id = st.session_state['application_id']
+        # bundle_by = st.session_state['bundle_by']
+
+        created_at = datetime.now().isoformat()
+
         user_data = {
             "Business_Head": business_head,
             "Company_ID": company_id,
@@ -949,7 +1213,7 @@ if bundle_by == "Feature specific":
             "Fee_Type": selected_fee_type,
             "Variable_Type": selected_variable_type,
             "Chargeable_on": selected_chargeable_on,
-            "Fee_Nature": "Flat currency",
+            "Fee_Nature": selected_fee_nature,
             "Commercial_value": user_input_3,
             "Fee_reversal": 0,
             "Reversal_%": 0,
@@ -961,9 +1225,22 @@ if bundle_by == "Feature specific":
             "Plan_Validity": selected_plan_validity,
             "rule_id": rule_id,
             "plan_id": plan_id,
+            "created_at": created_at,
             "Net_total_billing":int(final_billing)
         }
         
+        global_user_data.append((user_name, user_data))
+
+    # Generate PM_id and add it to user_data
+        rule_id = generate_rule_id(user_data)
+        user_data['rule_id'] = rule_id
+
+    # Generate Plan ID and add it to user_data
+        plan_id = generate_plan_id(user_data)
+        user_data['plan_id'] = plan_id
+
+        save_user_data(user_data)  # Save user data after submission
+
         service_account_info = {
             "type": "service_account",
             "project_id": "fynd-db",
@@ -1007,311 +1284,533 @@ elif bundle_by == "Single value":
     payment_method_options = ["Prepaid", "Postpaid"]
 
 
+    # Initialize session state variables if they don't exist
+    if 'new_mapping' not in st.session_state:
+        st.session_state['new_mapping'] = {}
+    if 'file_data' not in st.session_state:
+        st.session_state['file_data'] = None
+    if 'mapping_mode' not in st.session_state:
+        st.session_state['mapping_mode'] = None  # None, 'default', 'custom'
+    if 'button_clicked' not in st.session_state:
+        st.session_state['button_clicked'] = None
+
+    # Function to handle custom button click
+    def custom_button_click():
+        st.session_state.button_clicked = 'custom'
+
+    # Function to handle default button click
+    def default_button_click():
+        st.session_state.button_clicked = 'default'
+
+    
+    # Display buttons without immediate execution on click
+    col1, col2 = st.sidebar.columns([0.25, 0.45])
+    with col1:
+        default_clicked = st.button("Default mapping")
+    with col2:
+        custom_clicked = st.button("Custom mapping")
+
+    # Determine action based on button clicked
+    if default_clicked:
+        default_button_click()
+    elif custom_clicked:
+        custom_button_click()
+
+     # Define CSS styles for the button
+    button_styles = """
+    <style>
+    div.stButton > button {
+        color: #ffffff; /* Text color */
+        font-size: 50px;
+        background-image: linear-gradient(0deg, #a689f6 3%, #413bb9 61%);
+        border: none;
+        padding: 10px 20px;
+        cursor: pointer;
+        border-radius: 15px;
+        display: inline-block;
+    }
+    div.stButton > button:hover {
+        background-color: #00ff00; /* Hover background color */
+        color: #ff0000; /* Hover text color */
+    }
+    </style>
+    """
+
+# Render the button with the specified styles
+    st.markdown(button_styles, unsafe_allow_html=True)
+
 # Given new_mapping dictionary
-    new_mapping = {
-        "GoFynd": {
-            "Marketing": {
-                "Order": ["Placed"]
-            },
-            "Transaction": {
-                "Bag": ["Return Window"],
-                "Shipment": ["Cancelled", "RTO", "DTO"]
-            },
-            "Logistics": {
-                "Shipment": ["Picked", "RTO", "DTO"]
-            },
-            "Packaging": {
-                "Shipment": ["Packed"]
+ # Handle Default Mapping
+    if st.session_state.button_clicked == 'default':
+        new_mapping = {
+            "GoFynd": {
+                "Marketing": {
+                    "Order": ["Placed"]
+                },
+                "Transaction": {
+                    "Bag": ["Return Window"],
+                    "Shipment": ["Cancelled", "RTO", "DTO"]
+                },
+                "Logistics": {
+                    "Shipment": ["Picked", "RTO", "DTO"]
+                },
+                "Packaging": {
+                    "Shipment": ["Packed"]
+                }
+            },
+            "Uniket": {
+                "Marketing": {
+                    "Order": ["Placed"]
+                },
+                "Transaction": {
+                    "Bag": ["Return Window"],
+                    "Shipment": ["Cancelled", "RTO", "DTO"]
+                },
+                "Logistics": {
+                    "Shipment": ["Picked", "RTO", "DTO"]
+                },
+                "Packaging": {
+                    "Shipment": ["Packed"]
+                }
+            },
+            "ONDC": {
+                "Transaction": {
+                    "Bag": ["Return Window"],
+                    "Shipment": ["Cancelled", "RTO", "DTO"]
+                },
+                "Logistics": {
+                    "Shipment": ["Picked", "RTO", "DTO"]
+                },
+                "Packaging": {
+                    "Shipment": ["Packed"]
+                }
+            },
+            "StoreOS": {
+                "Licensing": {
+                    "Application": ["Setup"]
+                },
+                "Subscription": {
+                    "Application": ["Subscribed"],
+                    "User": ["Active"],
+                    "Resource": ["TechOps-hr", "SiteOps-hr"]
+                },
+                "Marketing": {
+                    "Order": ["Placed"]
+                },
+                "Transaction": {
+                    "Order": ["Placed"],
+                    "Bag": ["Invoiced", "Delivered"]
+                },
+                "Logistics": {
+                    "Shipment": ["Picked", "RTO", "DTO"]
+                },
+                "Packaging": {
+                    "Shipment": ["Packed"]
+                }
+            },
+            "Website": {
+                "Licensing": {
+                    "Application": ["Setup"]
+                },
+                "Subscription": {
+                    "Application": ["Subscribed"],
+                    "User": ["Active"],
+                    "Resource": ["TechOps-hr", "SiteOps-hr"]
+                },
+                "Transaction": {
+                    "Order": ["Placed"],
+                    "Bag": ["Invoiced", "Delivered"]
+                },
+                "Logistics": {
+                    "Shipment": ["Picked", "RTO", "DTO"]
+                },
+                "Packaging": {
+                    "Shipment": ["Packed"]
+                },
+                "Development": {
+                    "Application": ["Build"]
+                }
+            },
+            "OMS": {
+                "Development": {
+                    "Integration": ["Build"]
+                },
+                "Licensing": {
+                    "Integration": ["Setup"]
+                },
+                "Subscription": {
+                    "Integration": ["Subscribed"],
+                    "User": ["Active"],
+                    "Resource": ["TechOps-hr", "SiteOps-hr"]
+                },
+                "Transaction": {
+                    "Order": ["Placed"],
+                    "Bag": ["Invoiced", "Delivered"]
+                }
+            },
+            "WMS": {
+                "Development": {
+                    "Integration": ["Build"]
+                },
+                "Licensing": {
+                    "Integration": ["Setup"]
+                },
+                "Subscription": {
+                    "Integration": ["Subscribed"],
+                    "User": ["Active"],
+                    "Resource": ["TechOps-hr", "SiteOps-hr"]
+                },
+                "Transaction": {
+                    "Order": ["Placed"],
+                    "Bag": ["Invoiced", "Delivered"],
+                    "B2B": ["Invoiced"],
+                    "B2C": ["Invoiced"]
+                }
+            },
+            "TMS": {
+                "Development": {
+                    "Integration": ["Build"]
+                },
+                "Licensing": {
+                    "Integration": ["Setup"]
+                },
+                "Subscription": {
+                    "Integration": ["Subscribed"],
+                    "User": ["Active"],
+                    "Resource": ["TechOps-hr", "SiteOps-hr"]
+                },
+                "Logistics": {
+                    "Shipment": ["Picked", "RTO", "DTO"]
+                }
+            },
+            "GMC": {
+                "Development": {
+                    "Extension": ["Build"],
+                    "Integration": ["Build"]
+                },
+                "Licensing": {
+                    "Extension": ["Setup"]
+                },
+                "Subscription": {
+                    "Extension": ["Subscribed"],
+                    "User": ["Active"],
+                    "Resource": ["TechOps-hr", "SiteOps-hr"]
+                }
+            },
+            "Partner": {
+                "Development": {
+                    "Extension": ["Build"]
+                },
+                "Licensing": {
+                    "Extension": ["Setup"]
+                },
+                "Subscription": {
+                    "Extension": ["Subscribed"],
+                    "User": ["Active"],
+                    "Resource": ["TechOps-hr", "SiteOps-hr"]
+                }
+            },
+            "Catalog Cloud": {
+                "Development": {
+                    "Extension": ["Build"]
+                },
+                "Licensing": {
+                    "Extension": ["Setup"]
+                },
+                "Subscription": {
+                    "Extension": ["Subscribed"],
+                    "User": ["Active"],
+                    "Resource": ["TechOps-hr", "SiteOps-hr"]
+                }
+            },
+            "FCP": {
+                "Subscription": {
+                    "Platform": ["Subscribed"],
+                    "User": ["Active"],
+                    "Resource": ["TechOps-hr", "SiteOps-hr"]
+                }
+            },
+            "PixelBin": {
+                "Subscription": {
+                    "Platform": ["Subscribed"],
+                    "User": ["Active"],
+                    "Resource": ["TechOps-hr", "SiteOps-hr"]
+                }
+            },
+            "Boltic": {
+                "Subscription": {
+                    "Platform": ["Subscribed"],
+                    "User": ["Active"],
+                    "Resource": ["TechOps-hr", "SiteOps-hr"]
+                }
+            },
+            "CoPilot": {
+                "Subscription": {
+                    "Platform": ["Subscribed"],
+                    "User": ["Active"],
+                    "Resource": ["TechOps-hr", "SiteOps-hr"]
+                }
             }
-        },
-        "Uniket": {
-            "Marketing": {
-                "Order": ["Placed"]
-            },
-            "Transaction": {
-                "Bag": ["Return Window"],
-                "Shipment": ["Cancelled", "RTO", "DTO"]
-            },
-            "Logistics": {
-                "Shipment": ["Picked", "RTO", "DTO"]
-            },
-            "Packaging": {
-                "Shipment": ["Packed"]
-            }
-        },
-        "ONDC": {
-            "Transaction": {
-                "Bag": ["Return Window"],
-                "Shipment": ["Cancelled", "RTO", "DTO"]
-            },
-            "Logistics": {
-                "Shipment": ["Picked", "RTO", "DTO"]
-            },
-            "Packaging": {
-                "Shipment": ["Packed"]
-            }
-        },
-        "StoreOS": {
-            "Licensing": {
-                "Application": ["Setup"]
-            },
-            "Subscription": {
-                "Application": ["Subscribed"],
-                "User": ["Active"],
-                "Resource": ["TechOps-hr", "SiteOps-hr"]
-            },
-            "Marketing": {
-                "Order": ["Placed"]
-            },
-            "Transaction": {
-                "Order": ["Placed"],
-                "Bag": ["Invoiced", "Delivered"]
-            },
-            "Logistics": {
-                "Shipment": ["Picked", "RTO", "DTO"]
-            },
-            "Packaging": {
-                "Shipment": ["Packed"]
-            }
-        },
-        "Website": {
-            "Licensing": {
-                "Application": ["Setup"]
-            },
-            "Subscription": {
-                "Application": ["Subscribed"],
-                "User": ["Active"],
-                "Resource": ["TechOps-hr", "SiteOps-hr"]
-            },
-            "Transaction": {
-                "Order": ["Placed"],
-                "Bag": ["Invoiced", "Delivered"]
-            },
-            "Logistics": {
-                "Shipment": ["Picked", "RTO", "DTO"]
-            },
-            "Packaging": {
-                "Shipment": ["Packed"]
-            },
-            "Development": {
-                "Application": ["Build"]
-            }
-        },
-        "OMS": {
-            "Development": {
-                "Integration": ["Build"]
-            },
-            "Licensing": {
-                "Integration": ["Setup"]
-            },
-            "Subscription": {
-                "Integration": ["Subscribed"],
-                "User": ["Active"],
-                "Resource": ["TechOps-hr", "SiteOps-hr"]
-            },
-            "Transaction": {
-                "Order": ["Placed"],
-                "Bag": ["Invoiced", "Delivered"]
-            }
-        },
-        "WMS": {
-            "Development": {
-                "Integration": ["Build"]
-            },
-            "Licensing": {
-                "Integration": ["Setup"]
-            },
-            "Subscription": {
-                "Integration": ["Subscribed"],
-                "User": ["Active"],
-                "Resource": ["TechOps-hr", "SiteOps-hr"]
-            },
-            "Transaction": {
-                "Order": ["Placed"],
-                "Bag": ["Invoiced", "Delivered"],
-                "B2B": ["Invoiced"],
-                "B2C": ["Invoiced"]
-            }
-        },
-        "TMS": {
-            "Development": {
-                "Integration": ["Build"]
-            },
-            "Licensing": {
-                "Integration": ["Setup"]
-            },
-            "Subscription": {
-                "Integration": ["Subscribed"],
-                "User": ["Active"],
-                "Resource": ["TechOps-hr", "SiteOps-hr"]
-            },
-            "Logistics": {
-                "Shipment": ["Picked", "RTO", "DTO"]
-            }
-        },
-        "GMC": {
-            "Development": {
-                "Extension": ["Build"],
-                "Integration": ["Build"]
-            },
-            "Licensing": {
-                "Extension": ["Setup"]
-            },
-            "Subscription": {
-                "Extension": ["Subscribed"],
-                "User": ["Active"],
-                "Resource": ["TechOps-hr", "SiteOps-hr"]
-            }
-        },
-        "Partner": {
-            "Development": {
-                "Extension": ["Build"]
-            },
-            "Licensing": {
-                "Extension": ["Setup"]
-            },
-            "Subscription": {
-                "Extension": ["Subscribed"],
-                "User": ["Active"],
-                "Resource": ["TechOps-hr", "SiteOps-hr"]
-            }
-        },
-        "Catalog Cloud": {
-            "Development": {
-                "Extension": ["Build"]
-            },
-            "Licensing": {
-                "Extension": ["Setup"]
-            },
-            "Subscription": {
-                "Extension": ["Subscribed"],
-                "User": ["Active"],
-                "Resource": ["TechOps-hr", "SiteOps-hr"]
-            }
-        },
-        "FCP": {
-            "Subscription": {
-                "Platform": ["Subscribed"],
-                "User": ["Active"],
-                "Resource": ["TechOps-hr", "SiteOps-hr"]
-            }
-        },
-        "PixelBin": {
-            "Subscription": {
-                "Platform": ["Subscribed"],
-                "User": ["Active"],
-                "Resource": ["TechOps-hr", "SiteOps-hr"]
-            }
-        },
-        "Boltic": {
-            "Subscription": {
-                "Platform": ["Subscribed"],
-                "User": ["Active"],
-                "Resource": ["TechOps-hr", "SiteOps-hr"]
+        }
+
+        st.session_state['new_mapping'] = new_mapping
+        st.session_state['mapping_mode'] = 'default'
+
+        html_content_1 = """
+                            <h1 style='
+                                color: #a689f6;
+                                font-size: 22px;
+                                background-image: -webkit-linear-gradient(0deg, #a689f6 3%, #272191 33%, #413bb9 61%);
+                                background-clip: text;
+                                -webkit-background-clip: text;
+                                text-fill-color: transparent;
+                                -webkit-text-fill-color: transparent;
+                            '>Section 2: Rule Info</h1>
+                        """
+
+        st.sidebar.markdown(html_content_1, unsafe_allow_html=True)
+
+    # Handle Custom Mapping
+    elif st.session_state.button_clicked == 'custom':
+
+        # Define CSS styles for the button
+        button_styles = """
+            <style>
+            div.stButton > button {
+                color: #ffffff; /* Text color */
+                font-size: 50px;
+                background-image: linear-gradient(0deg, #a689f6 3%, #413bb9 61%);
+                border: none;
+                padding: 10px 20px;
+                cursor: pointer;
+                border-radius: 15px;
+                display: inline-block;
             }
-        },
-        "CoPilot": {
-            "Subscription": {
-                "Platform": ["Subscribed"],
-                "User": ["Active"],
-                "Resource": ["TechOps-hr", "SiteOps-hr"]
+            div.stButton > button:hover {
+                background-color: #00ff00; /* Hover background color */
+                color: #ff0000; /* Hover text color */
             }
-        }
-    }
+            </style>
+            """
+
+# Render the button with the specified styles
+        st.markdown(button_styles, unsafe_allow_html=True)
+
+        # File upload
+        uploaded_file = st.sidebar.file_uploader("Upload a file:", type=['csv', 'xlsx'], key='file_uploader')
+
+        html_content_1 = """
+                            <h1 style='
+                                color: #a689f6;
+                                font-size: 22px;
+                                background-image: -webkit-linear-gradient(0deg, #a689f6 3%, #272191 33%, #413bb9 61%);
+                                background-clip: text;
+                                -webkit-background-clip: text;
+                                text-fill-color: transparent;
+                                -webkit-text-fill-color: transparent;
+                            '>Section 2: Rule Info</h1>
+                        """
+
+        st.sidebar.markdown(html_content_1, unsafe_allow_html=True)
+
+        if uploaded_file is not None:
+            try:
+                # Read the file, treating the first row as the header
+                if uploaded_file.name.endswith('.csv'):
+                    data = pd.read_csv(uploaded_file, header=0)
+                elif uploaded_file.name.endswith('.xlsx'):
+                    data = pd.read_excel(uploaded_file, header=0)
+
+                if st.sidebar.button("Overwrite headers"):
+                    # Rename the headers
+                    new_headers = ['ordering_channel', 'fee_type', 'variable_type', 'chargeable_on']
+                    data.columns = new_headers
+
+                    # Save the data to session state
+                    st.session_state['file_data'] = data
+
+                    # Initialize the mapping
+                    new_mapping = {}
+
+                    
+
+                    # Build the mapping
+                    for _, row in data.iterrows():
+
+                
+                        ordering_channel = row['ordering_channel']
+                        fee_type = row['fee_type']
+                        variable_type = row['variable_type']
+                        chargeable_on = row['chargeable_on']
+
+                        if ordering_channel not in new_mapping:
+                            new_mapping[ordering_channel] = {}
+
+                        if fee_type not in new_mapping[ordering_channel]:
+                            new_mapping[ordering_channel][fee_type] = {}
+
+                        if variable_type not in new_mapping[ordering_channel][fee_type]:
+                            new_mapping[ordering_channel][fee_type][variable_type] = []
 
-    # Initialize mappings
-    fee_type_mapping = {}
-    variable_type_mapping = {}
-    chargeable_on_mapping = {}
-
-    # Populate mappings from new_mapping
-    for ordering_channel, details in new_mapping.items():
-        for fee_type, variables in details.items():
-            if fee_type not in fee_type_mapping:
-                fee_type_mapping[fee_type] = []
-            if fee_type not in variable_type_mapping:
-                variable_type_mapping[fee_type] = []
-            variable_types = list(variables.keys())
-            variable_type_mapping[fee_type].extend(variable_types)
-            for variable_type, chargeable_on_list in variables.items():
-                if variable_type not in chargeable_on_mapping:
-                    chargeable_on_mapping[variable_type] = []
-                chargeable_on_mapping[variable_type].extend(chargeable_on_list)
-
-    # Remove duplicates in mappings
-    fee_type_mapping = {k: list(set(v)) for k, v in fee_type_mapping.items()}
-    variable_type_mapping = {k: list(set(v)) for k, v in variable_type_mapping.items()}
-    chargeable_on_mapping = {k: list(set(v)) for k, v in chargeable_on_mapping.items()}
+                        if chargeable_on not in new_mapping[ordering_channel][fee_type][variable_type]:
+                            new_mapping[ordering_channel][fee_type][variable_type].append(chargeable_on)
 
+                    st.session_state['new_mapping'] = new_mapping
+                    st.session_state['mapping_mode'] = 'custom'
+
+            except Exception as e:
+                st.error(f"Error reading the file: {e}")
     # Section 2: Company Info
     
-    html_content_3 = """
-        <h1 style='
-            color: #a689f6;
-            font-size: 22px;
-            background-image: -webkit-linear-gradient(0deg, #a689f6 3%, #272191 33%, #413bb9 61%);
-            background-clip: text;
-            -webkit-background-clip: text;
-            text-fill-color: transparent;
-            -webkit-text-fill-color: transparent;
-        '>Section 2: Rule Info</h1>
-    """
+    
 
-    st.sidebar.markdown(html_content_3, unsafe_allow_html=True)
 
-    # 1st layer: Select ordering channel
-    selected_ordering_channel = st.sidebar.selectbox(
-        "Product lines",
-        [""] + list(new_mapping.keys()),
-        help="Select a Product line"
-    )
+     # Handle mapping display
+   # Initialize session state variables if they don't exist
+    # Initialize session state variables with default values if not already set
+    if 'business_head' not in st.session_state:
+        st.session_state['business_head'] = None
+    if 'company_id' not in st.session_state:
+        st.session_state['company_id'] = None
+    if 'company_name' not in st.session_state:
+        st.session_state['company_name'] = None
+    if 'currency' not in st.session_state:
+        st.session_state['currency'] = None
+    if 'bundle_by' not in st.session_state:
+        st.session_state['bundle_by'] = None
+    if 'plan_name' not in st.session_state:
+        st.session_state['plan_name'] = None
+    if 'plan_description' not in st.session_state:
+        st.session_state['plan_description'] = None
+    if 'selected_ordering_channel' not in st.session_state:
+        st.session_state['selected_ordering_channel'] = None
+    if 'fulfilling_location' not in st.session_state:
+        st.session_state['fulfilling_location'] = None
+    if 'application_id' not in st.session_state:
+        st.session_state['application_id'] = None
+    if 'selected_fee_type' not in st.session_state:
+        st.session_state['selected_fee_type'] = None
+    if 'selected_variable_type' not in st.session_state:
+        st.session_state['selected_variable_type'] = None
+    if 'selected_chargeable_on' not in st.session_state:
+        st.session_state['selected_chargeable_on'] = None
+    if 'selected_fee_nature' not in st.session_state:
+        st.session_state['selected_fee_nature'] = None
+    if 'user_input_3' not in st.session_state:
+        st.session_state['user_input_3'] = None
+    if 'usage_3' not in st.session_state:
+        st.session_state['usage_3'] = None
+    if 'Capping_or_Minimum_Guarantee_3' not in st.session_state:
+        st.session_state['Capping_or_Minimum_Guarantee_3'] = None
+    if 'threshold' not in st.session_state:
+        st.session_state['threshold'] = None
+    if 'expected_billing' not in st.session_state:
+        st.session_state['expected_billing'] = None
+    if 'selected_payment_method' not in st.session_state:
+        st.session_state['selected_payment_method'] = None
+    if 'selected_plan_validity' not in st.session_state:
+        st.session_state['selected_plan_validity'] = None
+    if 'final_billing' not in st.session_state:
+        st.session_state['final_billing'] = None
+    if 'fee_reversal' not in st.session_state:
+        st.session_state['fee_reversal'] = None
+    if 'reversal_per' not in st.session_state:
+        st.session_state['reversal_per'] = None
+
+
+    # Check if 'mapping_mode' exists in session state before using it
+    if 'mapping_mode' in st.session_state and st.session_state['mapping_mode']:
+        # 1st layer: Select ordering channel
+        selected_ordering_channel = st.sidebar.selectbox(
+            "Product lines",
+            [""] + list(st.session_state['new_mapping'].keys()),
+            help="Select a Product line",
+            key='ordering_channel'
+        )
+
+        if selected_ordering_channel:
+            selected_ordering_channel = str(selected_ordering_channel) or int(selected_ordering_channel)
+
+            # Determine fulfilling_location and application_id based on selected_ordering_channel
+            if any(channel in selected_ordering_channel for channel in ["TMS", "GMC", "Catalog Cloud"]):
+                st.session_state['fulfilling_location'] = None
+                st.session_state['application_id'] = None
+            else:
+                abc = ["Store", "Warehouse"]
+                st.session_state['fulfilling_location'] = st.sidebar.selectbox(
+                    "Fulfilling Location", [""] + abc, help="Select the fulfilling location"
+                )
+                st.session_state['application_id'] = st.sidebar.text_input(
+                    "Application ID", key="application_id_input", help="Enter the application ID"
+                )
 
-    # Layer 1: Mapping of Ordering channels with Fulfilling location
-    if "TMS" in selected_ordering_channel or "GMC" in selected_ordering_channel or "Catalog Cloud" in selected_ordering_channel:
-        fulfilling_location = None
-        application_id = None
-    else:
-        abc = ["Store", "Warehouse"]
-        fulfilling_location = st.sidebar.selectbox("Fulfilling Location", [""] + abc, help="Select the fulfilling location")
-        application_id = st.sidebar.text_input("Application ID", key="application_id_input", help="Enter the application ID")
-
-    # 2nd layer: Select fee type based on the selected ordering channel
-    fee_types_for_channel = []
-    if selected_ordering_channel:
-        details = new_mapping.get(selected_ordering_channel, {})
-        fee_types_for_channel = list(details.keys())
-
-    selected_fee_type = st.selectbox(
-        "Fee Type",
-        [""] + fee_types_for_channel,
-        help="Fee type is a revenue head for grouping the revenue"
-    )
+            # Use the session state variables in your app
+            fulfilling_location = st.session_state['fulfilling_location']
+            application_id = st.session_state['application_id']
 
-    # 3rd layer: Select variable type based on the selected fee type
-    variable_type_options = []
-    if selected_fee_type and selected_ordering_channel:
-        details = new_mapping.get(selected_ordering_channel, {})
-        variables = details.get(selected_fee_type, {})
-        variable_type_options = list(variables.keys())
-
-    selected_variable_type = st.selectbox(
-        "Variable Type",
-        [""] + variable_type_options,
-        help="Variable type is an attribute on which revenue calculation will be done"
-    )
 
-    # 4th layer: Select chargeable on based on the selected variable type
-    chargeable_on_options = []
-    if selected_variable_type and selected_fee_type and selected_ordering_channel:
-        details = new_mapping.get(selected_ordering_channel, {})
-        variables = details.get(selected_fee_type, {})
-        chargeable_on_options = variables.get(selected_variable_type, [])
-
-    selected_chargeable_on = st.selectbox(
-    "Chargeable on",
-    [""] + chargeable_on_options,
-    help="Chargeable on is a service status on which the above fee will be applicable"
-)
+
+            # 2nd layer: Select fee type based on the selected ordering channel
+
+            if 'selected_fee_type' not in st.session_state:
+                st.session_state['selected_fee_type'] = None
+
+            fee_types_for_channel = st.session_state['new_mapping'].get(selected_ordering_channel, {}).keys()
+            fee_types_for_channel = list(fee_types_for_channel)
+
+            if fee_types_for_channel:
+
+
+                # Define selected_fee_type based on user input
+
+                st.session_state['selected_fee_type'] = st.selectbox(
+                    "Fee Type",
+                    [""] + fee_types_for_channel,
+                    help="Fee type is a revenue head for grouping the revenue",
+                    key='fee_type'
+                )
+
+                # Access selected_fee_type for any session state operations
+                selected_fee_type = st.session_state['selected_fee_type']
+
+                # 3rd layer: Select variable type based on the selected fee type
+                variable_type_options = []
+                if selected_fee_type:
+                    details = st.session_state['new_mapping'].get(selected_ordering_channel, {})
+                    variables = details.get(selected_fee_type, {})
+                    variable_type_options = list(variables.keys())
+
+                st.session_state['selected_variable_type'] = st.selectbox(
+                    "Variable Type",
+                    [""] + variable_type_options,
+                    help="Variable type is an attribute on which revenue calculation will be done",
+                    key='variable_type'
+                )
+
+                selected_variable_type = st.session_state['selected_variable_type']
+
+            
+                # 4th layer: Select chargeable on based on the selected variable type
+                chargeable_on_options = []
+                if selected_variable_type:
+                    chargeable_on_options = variables.get(selected_variable_type, [])
+
+                st.session_state['selected_chargeable_on'] = st.selectbox(
+                    "Chargeable on",
+                    [""] + chargeable_on_options,
+                    help="Chargeable on is a service status on which the above fee will be applicable",
+                    key='chargeable_on'
+                )
+
+                selected_chargeable_on = st.session_state['selected_chargeable_on']
+
+            else:
+                st.write("No fee types available for the selected ordering channel.")
+        else:
+            st.write("")
+
+    
+    selected_variable_type = st.session_state['selected_variable_type']
 
     usage_1 = st.number_input(f"Usage limit for {selected_variable_type}", min_value=0.0, help="Enter the usage limit")
     usage_2 = float(usage_1)
@@ -1393,13 +1892,38 @@ elif bundle_by == "Single value":
 
     rule_id = None
     plan_id = None 
+    business_head = st.session_state['business_head']
+    company_name = st.session_state['company_name']
+    company_id = st.session_state['company_id']
+    currency = st.session_state['currency']
+    # bundle_by = st.session_state['bundle_by']
+    plan_name = st.session_state['plan_name']
+    plan_description = st.session_state['plan_description']
+    selected_ordering_channel = st.session_state['selected_ordering_channel']
+    fulfilling_location = st.session_state['fulfilling_location']
+    application_id = st.session_state['application_id']
+    selected_fee_type = st.session_state['selected_fee_type']
+    selected_variable_type = st.session_state['selected_variable_type']
+    selected_chargeable_on = st.session_state['selected_chargeable_on']
+    user_input_3 = st.session_state['user_input_3']
+    usage_3 = st.session_state['usage_3']
+    Capping_or_Minimum_Guarantee_3 = st.session_state['Capping_or_Minimum_Guarantee_3']
+    fee_reversal = st.session_state['fee_reversal']
+    reversal_per = st.session_state['reversal_per']
+    threshold = st.session_state['threshold']
+    expected_billing = st.session_state['expected_billing']
+    selected_payment_method = st.session_state['selected_payment_method']
+    selected_plan_validity = st.session_state['selected_plan_validity']
+    final_billing = int(st.session_state['final_billing']) if st.session_state['final_billing'] is not None else 0
+    
+    
 
 # Submit button
     if st.button("Submit"):
     # Save user data
         user_data = {
         "Business Head": business_head,
-        "Company ID": company_id,
+        "Company ID": company_id.strip() if company_id else None,
         "Company Name": company_name,
         "Currency": currency,
         "Bundle by": bundle_by,
@@ -1411,7 +1935,7 @@ elif bundle_by == "Single value":
         "Fee Type": selected_fee_type,
         "Variable Type": selected_variable_type,
         "Chargeable on": selected_chargeable_on,
-        "Fee Nature": "Flat currency",
+        "Fee Nature": " ",
         "Commercial value": user_input_3,
         "Fee reversal": 0,
         "Reversal %": 0,
@@ -1479,13 +2003,20 @@ elif bundle_by == "Single value":
 
             
 
-        # Your data and service account info here
         project_id = "fynd-db"
         dataset_id = "finance_dwh"
         table_id = "plan_v2"
+        fulfilling_location = st.session_state['fulfilling_location']
+        application_id = st.session_state['application_id']
+        # bundle_by = st.session_state['bundle_by']
+        
+
+        created_at = datetime.now().isoformat()
+
+
         user_data = {
             "Business_Head": business_head,
-            "Company_ID": company_id,
+            "Company_ID": company_id if company_id and company_id.strip() else None,
             "Company_Name": company_name,
             "Currency": currency,
             "Bundle_by": bundle_by,
@@ -1497,7 +2028,7 @@ elif bundle_by == "Single value":
             "Fee_Type": selected_fee_type,
             "Variable_Type": selected_variable_type,
             "Chargeable_on": selected_chargeable_on,
-            "Fee_Nature": "Flat currency",
+            "Fee_Nature": " ",
             "Commercial_value": user_input_3,
             "Fee_reversal": 0,
             "Reversal_%": 0,
@@ -1509,9 +2040,23 @@ elif bundle_by == "Single value":
             "Plan_Validity": selected_plan_validity,
             "rule_id": rule_id,
             "plan_id": plan_id,
+            "created_at": created_at,
             "Net_total_billing":int(final_billing)
         }
 
+        global_user_data.append((user_name, user_data))
+
+    # Generate PM_id and add it to user_data
+        rule_id = generate_rule_id(user_data)
+        user_data['rule_id'] = rule_id
+
+    # Generate Plan ID and add it to user_data
+        plan_id = generate_plan_id(user_data)
+        user_data['plan_id'] = plan_id
+
+
+        save_user_data(user_data)
+
         service_account_info = {
             "type": "service_account",
             "project_id": "fynd-db",
@@ -1562,8 +2107,8 @@ st.write(" ")
 # Start a tunnel to the correct port
 
 # Connect to a port (e.g., 8501) without a custom subdomain
-# try:
-#     ngrok_tunnel = ngrok.connect(8501)  # Replace 8501 with your port number
-#     print(f"ngrok tunnel URL: {ngrok_tunnel.public_url}")
-# except Exception as e:
-#     print(f"Failed to start ngrok tunnel: {e}")
+try:
+    ngrok_tunnel = ngrok.connect(8501)  # Replace 8501 with your port number
+    print(f"ngrok tunnel URL: {ngrok_tunnel.public_url}")
+except Exception as e:
+    print(f"Failed to start ngrok tunnel: {e}")