# Importing necessary libraries import os import streamlit as st import pandas as pd from html2image import Html2Image from io import BytesIO from PIL import Image from datetime import datetime from utilities import ( initialize_session_state, save_user_data, generate_rule_id, generate_plan_id, display_saved_records, load_theme_config, store_and_fetch_company_info, get_feature_user_data, default_button_click, custom_button_click, uploader, render_selectboxes, insert_into_bq ) from html_templates import ( app_title, plan_info_header, button_styles, rule_info_header, expected_billing_header, final_billing_header, side_logo ) from placeholders import ( fee_type_options, fee_nature_options, variable_options, chargeable_on_options, plan_validity_options, payment_method_options, fee_reversal_options, threshold_options ) from mapping import new_mapping from slab import generate_slab_data initialize_session_state() theme_settings = load_theme_config() # Streamlit UI elements st.sidebar.markdown(side_logo, unsafe_allow_html=True) st.markdown(app_title, unsafe_allow_html=True) st.write(" ") # User input section user_name = st.sidebar.text_input("Enter your name:") selected_business_head = st.sidebar.selectbox( "Business Head", ["", "Commerce India", "Reliance", "Commerce Global", "Government Projects", "Individual BH"], help="Business Head is used for cost centre allocation" ) company_id = st.sidebar.text_input("Company ID", help="Enter the company ID") store_and_fetch_company_info(company_id) st.sidebar.text_input("Company Name", value=st.session_state.get('company_name', ''), disabled=False) currency = st.sidebar.radio("Currency", options=["INR", "USD"], help="Select the type of currency") bundle_by = st.sidebar.radio("Bundle by", options=["Single value", "Feature specific"], help="Select the bundle") # Feature specific options if bundle_by == "Feature specific": plan_name = st.sidebar.text_input("Plan Name", help="Enter the name of the plan") plan_description = st.sidebar.text_area("Plan Description", max_chars=250, help="Describe the plan") plan_start_date = st.sidebar.date_input("Plan Start Date", value=datetime.now(), help="Select the start date of the plan") 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") if default_clicked: default_button_click() elif custom_clicked: custom_button_click() st.markdown(button_styles, unsafe_allow_html=True) if st.session_state.get('button_clicked') == 'default': st.session_state['new_mapping'] = new_mapping st.session_state['mapping_mode'] = 'default' st.sidebar.markdown(rule_info_header, unsafe_allow_html=True) elif st.session_state.get('button_clicked') == 'custom': st.markdown(button_styles, unsafe_allow_html=True) uploaded_file = st.sidebar.file_uploader("Upload a file:", type=['csv', 'xlsx'], key='file_uploader') st.sidebar.markdown(rule_info_header, unsafe_allow_html=True) if uploaded_file is not None: uploader(uploaded_file) if st.session_state.get('mapping_mode'): render_selectboxes(st.session_state, st.session_state.get('new_mapping')) selected_fee_nature = st.selectbox("Fee Nature:", [""] + fee_nature_options, key="fee_nature_select", help="Fee nature is an option to consume the price in flat / % / slab") ######################################################## Slab based ########################################################### if selected_fee_nature == "Slab based": slab_data = generate_slab_data() 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") os.environ['selected_payment_method'] = selected_payment_method rule_id = None plan_id = None user_data = None project_id = "fynd-db" dataset_id = "finance_dwh" table_id = "plan_v2" service_account_info = { "type": "service_account", "project_id": "fynd-db", "private_key_id": "48954327ef172acc3091a91881840dc353ff0413", "private_key": "-----BEGIN PRIVATE KEY-----\nMIIEvQIBADANBgkqhkiG9w0BAQEFAASCBKcwggSjAgEAAoIBAQCmIs0vgXflZmHD\n0IeHCiDEXavRb2GvRQqkCZGLy/EHFO7XLmpNrFzZHTSQach6HErARWTMsd2AMIA5\nHbmjEyt9deTOMUmICszkwNwpn6pzuoF2v4+fL2PQ4VBElLfCv4qbP9JGljvLsj5b\nC9UWe9rWgRrw2Z88LlVmFy3k4Om80Q57gW3hSYZpG4rBhC9+YRXjZ+Iq5SFEJQ+U\nj8ej7q6DnI8Prbvw/e13JH1pxm9AUQDm+sTsBmvjh4Ye02vKbfUP+pFOM0eC3Eoa\nOyG7unZnDuAaK9g87xVjd9HtD6MfaM0EHAEF1eyC4UqeiXTbVMwi7+0IGA2evEDZ\n3WRR2i+lAgMBAAECggEAF7Vis7NVsVqFfCS/moFTAMLfWLF87r67EIK/DwSpHloZ\n5YJdsDz3ORKmZC98aRSthDfC9UUyt270dXItAj0jmTFgWB6HgE8OQ1zUbmo3MHG+\ntPwMWmqwY2gsBMV2Xefot1QJxYH+AYkrxeFv5NgC+FaPSiy8QSHZlQqcxYtlP8kY\nzv8QFMgZ3b0sixK+YcSjCRXHu/DznnS/ZU7XlTALxANa50hT11rVLq6zYDF40fqC\nOWCOZvupjMOnNimXCWn7q6D1F8S/IMYlnHArJpEb7SdbJnJzxUvJC2rrMSoY896r\nyFEVhlElWp5GfTopu0WnL7iaw3pukM1OtpanFAk29wKBgQDW+yMzL7RiQ5LdElVP\n7Ml5hs+t6cjrUJTgBI8Uf9qGxE+tMe7Ye1fyJfbYslbOYO9xIUUT1jJmU6jadNWX\nA7UT59/IjvAOBIYyMEdh6Xo+H6ggW8DeU1HSdvixnQdgi0c03E6OCbOKi9E/UzK/\nTK2RBJJLN4MCiNL0FzrhTuxyEwKBgQDF1cz9Sb2UbrL5hr8LdcEH0ctatNlhJcSN\nnMd0Fn+e1EAkip+Hs9HmTvNdty4gQE+yEq/Tj0cRHXmjzK2AYx0WYRAhWGNSTlTS\n4OJdjE6vsoPu7Ouz1R/i/egKniUifEXYSgyL+oPBUrqqsnk9lgGBkJipN3u1f9+K\niuUJJP/OZwKBgF6ZvqCcom0HPU5I7f+wu+vdVfA6yy45lHmLqAamSFw7cLBPI8Jh\nbI7jA9/Rgn9oipUmxcX34M/Eiq4u8Xp1qC4tP/16YMpaVU8qjY7ZdfB2b75lgdaT\npZLOxZsq9X8Xauso8uxv+nDCG/8YtmEV9d61u0acE+t+mA3PVxqkZ0m/AoGBAI5c\nY76AqeN+JVxaEm/0tIsj9Om46hR2URJ2lzB6YCuzINUqy9GjHJBWj9oITzD2FmNV\n/yCGIeW3CClOyCtzJyNLhYf5Sr+XjoKRQVN/+7+C/l2YL6Sg4Ok/PRMm6iH+u2QB\nJTY1d0pOdfUPqR8gKsVJgBGE04iwE/RmLpp9/XZRAoGAFtaa9JeHk9dmZiST3lev\ns5syeOK1cCj1+nJer/GPEjqWMKxbCULvoEYBp7RaoEXjfGi7P4Mc6YHnbljHc+sL\niJzHhxrv/HYZdlCdTU3IRWT0sNHzqZemfYnsVTSp+4y61sGDxioDYL4CLnMq5T2F\nqzfc2qp3s8ApVloY3giuqnI=\n-----END PRIVATE KEY-----\n", "client_email": "plan-maker@fynd-db.iam.gserviceaccount.com", "client_id": "114883060650442183907", "auth_uri": "https://accounts.google.com/o/oauth2/auth", "token_uri": "https://oauth2.googleapis.com/token", "auth_provider_x509_cert_url": "https://www.googleapis.com/oauth2/v1/certs", "client_x509_cert_url": "https://www.googleapis.com/robot/v1/metadata/x509/plan-maker%40fynd-db.iam.gserviceaccount.com", "universe_domain": "googleapis.com" } if st.button("Submit", key='submit_button'): user_data = { "Business_Head": selected_business_head, "Company_ID": company_id.strip(), "Company_Name": st.session_state.get('company_name', ''), "Currency": currency, "Bundle_by": bundle_by, "Plan_Name": plan_name, "Plan_Description": plan_description, "Product_lines": st.session_state.get('ordering_channel', ''), "Fulfilling_Location": st.session_state.get('fulfilling_location', ''), "Application_ID": st.session_state.get('application_id', ''), "Fee_Type": st.session_state.get('selected_fee_type', ''), "Variable_Type": st.session_state.get('selected_variable_type', ''), "Chargeable_on": st.session_state.get('selected_chargeable_on', ''), "Fee_Nature": selected_fee_nature, "Commercial_value": 0.0, "Fee_reversal": os.getenv('fee_reversal',''), "Reversal_%": os.getenv('reversal_per',''), "Usage_limit": 0.0, "Threshold_value": 0.0, "Threshold_option": '', "Plan_validity": '', "Payment_Method": os.getenv('selected_payment_method',''), "Expected_Billing": 0.0, "Net_total_billing": 0, "slab_data": slab_data if selected_fee_nature == "Slab based" else "" } # Save and display user data save_user_data(user_data) st.session_state.user_data = user_data st.session_state.df = pd.DataFrame([user_data]) st.write(st.session_state.df) if st.button("Export to BQ"): if project_id and dataset_id and table_id and service_account_info: user_data = st.session_state.user_data # Generate rule_id and plan_id rule_id = generate_rule_id(user_data) plan_id = generate_plan_id(user_data) # Update user_data with generated IDs user_data['rule_id'] = rule_id user_data['plan_id'] = plan_id st.session_state['created_at'] = datetime.now().isoformat() user_data = { "Business_Head": selected_business_head, "Company_ID": company_id.strip(), "Company_Name": st.session_state.get('company_name', ''), "Currency": currency, "Bundle_by": bundle_by, "Plan_Name": plan_name, "Plan_Description": plan_description, "Product_lines": st.session_state.get('ordering_channel', ''), "Fulfilling_Location": st.session_state.get('fulfilling_location', ''), "Application_ID": st.session_state.get('application_id', ''), "Fee_Type": st.session_state.get('selected_fee_type', ''), "Variable_Type": st.session_state.get('selected_variable_type', ''), "Chargeable_on": st.session_state.get('selected_chargeable_on', ''), "Fee_Nature": selected_fee_nature, # Ensure environment values are safely converted to float, handling empty strings "Commercial_value": 0.0, "Fee_reversal": os.getenv('fee_reversal', ''), "Reversal_%": float(os.getenv('reversal_per', '0.0') or 0.0), "Usage_limit": 0.0, "Threshold_value": 0.0, "Threshold_option": '', "Plan_validity": '', "Payment_Method": os.getenv('selected_payment_method', ''), "Expected_Billing": 0.0, "Net_total_billing": 0, "rule_id": rule_id, "plan_id": plan_id, "created_at": st.session_state.get('created_at', ''), # Set slab_data based on the selected fee nature "slab_data": slab_data if selected_fee_nature == "Slab based" else "" } # Save the user data locally and to session state save_user_data(user_data) # Display saved records display_saved_records() # Store created_at, rule_id, and plan_id in session state st.session_state['created_at'] = datetime.now().isoformat() st.session_state['rule_id'] = rule_id st.session_state['plan_id'] = plan_id insert_into_bq(user_data, project_id, dataset_id, table_id, service_account_info) else: st.error("One or more required variables are not defined.") ############################################################ Fixed or Flat ##################################################### elif selected_fee_nature in ["Fixed %", "Flat currency"]: user_input_1 = st.number_input("Enter Commercial value:", min_value=0.0, help="Enter the Commercial value") user_input_2 = float(user_input_1) if isinstance(user_input_1, (int, float)) else 0.0 user_input_3 = "{:.2f}".format(user_input_2) os.environ['user_input_3'] = user_input_3 selected_fee_type = st.session_state.get('selected_fee_type', '') selected_variable_type = st.session_state.get('selected_variable_type', '') 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 fee_reversal = "/".join(fee_reversal) if fee_reversal else "" os.environ['fee_reversal'] = str(fee_reversal) if fee_reversal is not None else '' 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 "" os.environ['reversal_per'] = str(reversal_per) if reversal_per is not None else '' 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) usage_3 = "{:.2f}".format(usage_2) os.environ['usage_3'] = usage_3 product_1 = usage_1 * user_input_1 if selected_fee_nature in ["Fixed %", "Flat currency"] else 0 product_2 = float(product_1) product_3 = "{:.2f}".format(product_2) threshold = st.selectbox("Threshold option:", [""] + threshold_options, help="If Minimum guarantee is set as threshold option then Expected billing will be the higher value b/w 'Commercial*Usage' & 'Threshold' and if it is set to Capping then Expected billing will be the lower value b/w 'Commercial*Usage' & 'Threshold'") os.environ['threshold'] = threshold capping_or_minimum_guarantee_1 = st.number_input(f"Threshold value for {selected_variable_type}", min_value=0.0, help="Enter the Capping/Minimum Guarantee value") capping_or_minimum_guarantee_2 = float(capping_or_minimum_guarantee_1) capping_or_minimum_guarantee_3 = "{:.2f}".format(capping_or_minimum_guarantee_2) os.environ['capping_or_minimum_guarantee_3'] = capping_or_minimum_guarantee_3 initial_expected_billing = product_1 if threshold == 'Capping value': expected_billing = min(product_1, capping_or_minimum_guarantee_1) elif threshold == 'Minimum Guarantee': expected_billing = max(product_1, capping_or_minimum_guarantee_1) else: expected_billing = initial_expected_billing os.environ['expected_billing'] = str(expected_billing) if expected_billing is not None else '' 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") os.environ['selected_plan_validity'] = selected_plan_validity expected_billing_header = f"""

Expected Monthly Billing (excluding GST): {expected_billing}

""" st.markdown(expected_billing_header, unsafe_allow_html=True) if selected_plan_validity == 'One time': final_billing = 1 * expected_billing elif selected_plan_validity == 'Monthly': final_billing = 1 * expected_billing elif selected_plan_validity == 'Quarterly': final_billing = 3 * expected_billing elif selected_plan_validity == 'Bi-Annually': final_billing = 6 * expected_billing elif selected_plan_validity == 'Annually': final_billing = 12 * expected_billing else: final_billing = 0 os.environ['final_billing']= str(final_billing) if final_billing is not None else '' final_billing_header = f"""

Net total billing (excluding GST): {final_billing}

""" st.markdown(final_billing_header, unsafe_allow_html=True) 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") os.environ['selected_payment_method'] = selected_payment_method st.markdown(button_styles, unsafe_allow_html=True) rule_id = None plan_id = None user_data = None user_input_3 = float(os.getenv('user_input_3')) usage_3 = float(os.getenv('usage_3')) capping_or_minimum_guarantee_3 = float(os.getenv('capping_or_minimum_guarantee_3')) threshold = (os.getenv('threshold')) selected_plan_validity= os.getenv('selected_plan_validity') expected_billing= os.getenv('expected_billing') final_billing = os.getenv('final_billing') project_id = "fynd-db" dataset_id = "finance_dwh" table_id = "plan_v2" service_account_info = { "type": "service_account", "project_id": "fynd-db", "private_key_id": "48954327ef172acc3091a91881840dc353ff0413", "private_key": "-----BEGIN PRIVATE KEY-----\nMIIEvQIBADANBgkqhkiG9w0BAQEFAASCBKcwggSjAgEAAoIBAQCmIs0vgXflZmHD\n0IeHCiDEXavRb2GvRQqkCZGLy/EHFO7XLmpNrFzZHTSQach6HErARWTMsd2AMIA5\nHbmjEyt9deTOMUmICszkwNwpn6pzuoF2v4+fL2PQ4VBElLfCv4qbP9JGljvLsj5b\nC9UWe9rWgRrw2Z88LlVmFy3k4Om80Q57gW3hSYZpG4rBhC9+YRXjZ+Iq5SFEJQ+U\nj8ej7q6DnI8Prbvw/e13JH1pxm9AUQDm+sTsBmvjh4Ye02vKbfUP+pFOM0eC3Eoa\nOyG7unZnDuAaK9g87xVjd9HtD6MfaM0EHAEF1eyC4UqeiXTbVMwi7+0IGA2evEDZ\n3WRR2i+lAgMBAAECggEAF7Vis7NVsVqFfCS/moFTAMLfWLF87r67EIK/DwSpHloZ\n5YJdsDz3ORKmZC98aRSthDfC9UUyt270dXItAj0jmTFgWB6HgE8OQ1zUbmo3MHG+\ntPwMWmqwY2gsBMV2Xefot1QJxYH+AYkrxeFv5NgC+FaPSiy8QSHZlQqcxYtlP8kY\nzv8QFMgZ3b0sixK+YcSjCRXHu/DznnS/ZU7XlTALxANa50hT11rVLq6zYDF40fqC\nOWCOZvupjMOnNimXCWn7q6D1F8S/IMYlnHArJpEb7SdbJnJzxUvJC2rrMSoY896r\nyFEVhlElWp5GfTopu0WnL7iaw3pukM1OtpanFAk29wKBgQDW+yMzL7RiQ5LdElVP\n7Ml5hs+t6cjrUJTgBI8Uf9qGxE+tMe7Ye1fyJfbYslbOYO9xIUUT1jJmU6jadNWX\nA7UT59/IjvAOBIYyMEdh6Xo+H6ggW8DeU1HSdvixnQdgi0c03E6OCbOKi9E/UzK/\nTK2RBJJLN4MCiNL0FzrhTuxyEwKBgQDF1cz9Sb2UbrL5hr8LdcEH0ctatNlhJcSN\nnMd0Fn+e1EAkip+Hs9HmTvNdty4gQE+yEq/Tj0cRHXmjzK2AYx0WYRAhWGNSTlTS\n4OJdjE6vsoPu7Ouz1R/i/egKniUifEXYSgyL+oPBUrqqsnk9lgGBkJipN3u1f9+K\niuUJJP/OZwKBgF6ZvqCcom0HPU5I7f+wu+vdVfA6yy45lHmLqAamSFw7cLBPI8Jh\nbI7jA9/Rgn9oipUmxcX34M/Eiq4u8Xp1qC4tP/16YMpaVU8qjY7ZdfB2b75lgdaT\npZLOxZsq9X8Xauso8uxv+nDCG/8YtmEV9d61u0acE+t+mA3PVxqkZ0m/AoGBAI5c\nY76AqeN+JVxaEm/0tIsj9Om46hR2URJ2lzB6YCuzINUqy9GjHJBWj9oITzD2FmNV\n/yCGIeW3CClOyCtzJyNLhYf5Sr+XjoKRQVN/+7+C/l2YL6Sg4Ok/PRMm6iH+u2QB\nJTY1d0pOdfUPqR8gKsVJgBGE04iwE/RmLpp9/XZRAoGAFtaa9JeHk9dmZiST3lev\ns5syeOK1cCj1+nJer/GPEjqWMKxbCULvoEYBp7RaoEXjfGi7P4Mc6YHnbljHc+sL\niJzHhxrv/HYZdlCdTU3IRWT0sNHzqZemfYnsVTSp+4y61sGDxioDYL4CLnMq5T2F\nqzfc2qp3s8ApVloY3giuqnI=\n-----END PRIVATE KEY-----\n", "client_email": "plan-maker@fynd-db.iam.gserviceaccount.com", "client_id": "114883060650442183907", "auth_uri": "https://accounts.google.com/o/oauth2/auth", "token_uri": "https://oauth2.googleapis.com/token", "auth_provider_x509_cert_url": "https://www.googleapis.com/oauth2/v1/certs", "client_x509_cert_url": "https://www.googleapis.com/robot/v1/metadata/x509/plan-maker%40fynd-db.iam.gserviceaccount.com", "universe_domain": "googleapis.com" } if st.button("Submit", key='submit_button'): user_data = { "Business_Head": selected_business_head, "Company_ID": company_id.strip(), "Company_Name": st.session_state.get('company_name', ''), "Currency": currency, "Bundle_by": bundle_by, "Plan_Name": plan_name, "Plan_Description": plan_description, "Product_lines": st.session_state.get('ordering_channel', ''), "Fulfilling_Location": st.session_state.get('fulfilling_location', ''), "Application_ID": st.session_state.get('application_id', ''), "Fee_Type": st.session_state.get('selected_fee_type', ''), "Variable_Type": st.session_state.get('selected_variable_type', ''), "Chargeable_on": st.session_state.get('selected_chargeable_on', ''), "Fee_Nature": selected_fee_nature, "Commercial_value": float(os.getenv('user_input_3', '0.0')) if user_input_3 else 0.0, "Fee_reversal": os.getenv('fee_reversal',''), "Reversal_%": os.getenv('reversal_per',''), "Usage_limit": os.getenv('usage_3','') if usage_3 else 0.0, "Threshold_value": os.getenv('capping_or_minimum_guarantee_3','') if capping_or_minimum_guarantee_3 else 0.0, "Threshold_option": os.getenv('threshold','') if threshold else '', "Plan_validity": os.getenv('selected_plan_validity','') if selected_plan_validity else '', "Payment_Method": os.getenv('selected_payment_method',''), "Expected_Billing": os.getenv('expected_billing','') if expected_billing else 0.0, "Net_total_billing": os.getenv('final_billing','') if final_billing else 0, "slab_data": slab_data if selected_fee_nature == "Slab based" else "" } # Save and display user data save_user_data(user_data) st.session_state.user_data = user_data st.session_state.df = pd.DataFrame([user_data]) st.write(st.session_state.df) # Display saved records # display_saved_records() # Insert into BigQuery if st.button("Export to BQ"): if project_id and dataset_id and table_id and service_account_info: user_data = st.session_state.user_data # Generate rule_id and plan_id rule_id = generate_rule_id(user_data) plan_id = generate_plan_id(user_data) # Update user_data with generated IDs user_data['rule_id'] = rule_id user_data['plan_id'] = plan_id st.session_state['created_at'] = datetime.now().isoformat() user_data = { "Business_Head": selected_business_head, "Company_ID": company_id.strip(), "Company_Name": st.session_state.get('company_name', ''), "Currency": currency, "Bundle_by": bundle_by, "Plan_Name": plan_name, "Plan_Description": plan_description, "Product_lines": st.session_state.get('ordering_channel', ''), "Fulfilling_Location": st.session_state.get('fulfilling_location', ''), "Application_ID": st.session_state.get('application_id', ''), "Fee_Type": st.session_state.get('selected_fee_type', ''), "Variable_Type": st.session_state.get('selected_variable_type', ''), "Chargeable_on": st.session_state.get('selected_chargeable_on', ''), "Fee_Nature": selected_fee_nature, # Ensure environment values are safely converted to float, handling empty strings "Commercial_value": float(os.getenv('user_input_3', '0.0') or 0.0) if user_input_3 else 0.0, "Fee_reversal": os.getenv('fee_reversal', ''), "Reversal_%": float(os.getenv('reversal_per', '0.0') or 0.0), "Usage_limit": float(os.getenv('usage_3', '0.0') or 0.0) if usage_3 else 0.0, "Threshold_value": float(os.getenv('capping_or_minimum_guarantee_3', '0.0') or 0.0) if capping_or_minimum_guarantee_3 else 0.0, "Threshold_option": os.getenv('threshold', '') if threshold else '', "Plan_validity": os.getenv('selected_plan_validity', '') if selected_plan_validity else '', "Payment_Method": os.getenv('selected_payment_method', ''), # Convert expected_billing to float if it's a valid string or 0.0 if empty "Expected_Billing": float(os.getenv('expected_billing', '0.0') or 0.0) if expected_billing else 0.0, # Convert final_billing to int (default to 0 if empty) # Safely convert 'final_billing' to a float first, then to an integer "Net_total_billing": int(float(os.getenv('final_billing', '0.0') or 0.0)) if final_billing else 0, "rule_id": rule_id, "plan_id": plan_id, "created_at": st.session_state.get('created_at', ''), # Set slab_data based on the selected fee nature "slab_data": slab_data if selected_fee_nature == "Slab based" else "" } # Save the user data locally and to session state save_user_data(user_data) # Display saved records display_saved_records() # Store created_at, rule_id, and plan_id in session state st.session_state['created_at'] = datetime.now().isoformat() st.session_state['rule_id'] = rule_id st.session_state['plan_id'] = plan_id insert_into_bq(user_data, project_id, dataset_id, table_id, service_account_info) else: st.error("One or more required variables are not defined.") elif bundle_by == "Single value": user_input_1 = st.sidebar.number_input("Enter Commercial value:") user_input_2 = float(user_input_1) st.session_state['user_input_3'] = "{:.2f}".format(user_input_2) plan_name = st.sidebar.text_input("Plan Name", help="Enter the name of the plan") plan_description = st.sidebar.text_area("Plan Description", max_chars=250, help="Describe the plan") plan_start_date = st.sidebar.date_input("Plan Start Date", value=datetime.now(), help="Select the start date of the plan") # 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() st.markdown(button_styles, unsafe_allow_html=True) # Default button mapping if st.session_state.button_clicked == 'default': st.session_state['new_mapping'] = new_mapping st.session_state['mapping_mode'] = 'default' st.sidebar.markdown(rule_info_header, unsafe_allow_html=True) # Handle Custom Mapping elif st.session_state.button_clicked == 'custom': st.markdown(button_styles, unsafe_allow_html=True) uploaded_file = st.sidebar.file_uploader("Upload a file:", type=['csv', 'xlsx'], key='file_uploader') st.sidebar.markdown(rule_info_header, unsafe_allow_html=True) if uploaded_file is not None: uploader(uploaded_file) if st.session_state.get('mapping_mode'): render_selectboxes(st.session_state, st.session_state.get('new_mapping')) 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) usage_3 = "{:.2f}".format(usage_2) expected_billing = user_input_1 expected_billing_header = f"""

Expected Monthly Billing (excluding GST): {expected_billing}

""" st.markdown(expected_billing_header, unsafe_allow_html=True) 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") 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") initial_net_billing = 0 user_input_3 = st.session_state['user_input_3'] if selected_plan_validity == 'One time': final_billing = 1 * expected_billing elif selected_plan_validity == 'Monthly': final_billing = 1 * expected_billing elif selected_plan_validity == 'Quarterly': final_billing = 3 * expected_billing elif selected_plan_validity == 'Bi-Annually': final_billing = 6 * expected_billing elif selected_plan_validity == 'Annually': final_billing = 12 * expected_billing else: final_billing = initial_net_billing final_billing_header = f"""

Net total billing (excluding GST): {final_billing}

""" st.markdown(final_billing_header, unsafe_allow_html=True) st.markdown(button_styles, unsafe_allow_html=True) rule_id = None plan_id = None user_data = None project_id = "fynd-db" dataset_id = "finance_dwh" table_id = "plan_v2" service_account_info = { "type": "service_account", "project_id": "fynd-db", "private_key_id": "48954327ef172acc3091a91881840dc353ff0413", "private_key": "-----BEGIN PRIVATE KEY-----\nMIIEvQIBADANBgkqhkiG9w0BAQEFAASCBKcwggSjAgEAAoIBAQCmIs0vgXflZmHD\n0IeHCiDEXavRb2GvRQqkCZGLy/EHFO7XLmpNrFzZHTSQach6HErARWTMsd2AMIA5\nHbmjEyt9deTOMUmICszkwNwpn6pzuoF2v4+fL2PQ4VBElLfCv4qbP9JGljvLsj5b\nC9UWe9rWgRrw2Z88LlVmFy3k4Om80Q57gW3hSYZpG4rBhC9+YRXjZ+Iq5SFEJQ+U\nj8ej7q6DnI8Prbvw/e13JH1pxm9AUQDm+sTsBmvjh4Ye02vKbfUP+pFOM0eC3Eoa\nOyG7unZnDuAaK9g87xVjd9HtD6MfaM0EHAEF1eyC4UqeiXTbVMwi7+0IGA2evEDZ\n3WRR2i+lAgMBAAECggEAF7Vis7NVsVqFfCS/moFTAMLfWLF87r67EIK/DwSpHloZ\n5YJdsDz3ORKmZC98aRSthDfC9UUyt270dXItAj0jmTFgWB6HgE8OQ1zUbmo3MHG+\ntPwMWmqwY2gsBMV2Xefot1QJxYH+AYkrxeFv5NgC+FaPSiy8QSHZlQqcxYtlP8kY\nzv8QFMgZ3b0sixK+YcSjCRXHu/DznnS/ZU7XlTALxANa50hT11rVLq6zYDF40fqC\nOWCOZvupjMOnNimXCWn7q6D1F8S/IMYlnHArJpEb7SdbJnJzxUvJC2rrMSoY896r\nyFEVhlElWp5GfTopu0WnL7iaw3pukM1OtpanFAk29wKBgQDW+yMzL7RiQ5LdElVP\n7Ml5hs+t6cjrUJTgBI8Uf9qGxE+tMe7Ye1fyJfbYslbOYO9xIUUT1jJmU6jadNWX\nA7UT59/IjvAOBIYyMEdh6Xo+H6ggW8DeU1HSdvixnQdgi0c03E6OCbOKi9E/UzK/\nTK2RBJJLN4MCiNL0FzrhTuxyEwKBgQDF1cz9Sb2UbrL5hr8LdcEH0ctatNlhJcSN\nnMd0Fn+e1EAkip+Hs9HmTvNdty4gQE+yEq/Tj0cRHXmjzK2AYx0WYRAhWGNSTlTS\n4OJdjE6vsoPu7Ouz1R/i/egKniUifEXYSgyL+oPBUrqqsnk9lgGBkJipN3u1f9+K\niuUJJP/OZwKBgF6ZvqCcom0HPU5I7f+wu+vdVfA6yy45lHmLqAamSFw7cLBPI8Jh\nbI7jA9/Rgn9oipUmxcX34M/Eiq4u8Xp1qC4tP/16YMpaVU8qjY7ZdfB2b75lgdaT\npZLOxZsq9X8Xauso8uxv+nDCG/8YtmEV9d61u0acE+t+mA3PVxqkZ0m/AoGBAI5c\nY76AqeN+JVxaEm/0tIsj9Om46hR2URJ2lzB6YCuzINUqy9GjHJBWj9oITzD2FmNV\n/yCGIeW3CClOyCtzJyNLhYf5Sr+XjoKRQVN/+7+C/l2YL6Sg4Ok/PRMm6iH+u2QB\nJTY1d0pOdfUPqR8gKsVJgBGE04iwE/RmLpp9/XZRAoGAFtaa9JeHk9dmZiST3lev\ns5syeOK1cCj1+nJer/GPEjqWMKxbCULvoEYBp7RaoEXjfGi7P4Mc6YHnbljHc+sL\niJzHhxrv/HYZdlCdTU3IRWT0sNHzqZemfYnsVTSp+4y61sGDxioDYL4CLnMq5T2F\nqzfc2qp3s8ApVloY3giuqnI=\n-----END PRIVATE KEY-----\n", "client_email": "plan-maker@fynd-db.iam.gserviceaccount.com", "client_id": "114883060650442183907", "auth_uri": "https://accounts.google.com/o/oauth2/auth", "token_uri": "https://oauth2.googleapis.com/token", "auth_provider_x509_cert_url": "https://www.googleapis.com/oauth2/v1/certs", "client_x509_cert_url": "https://www.googleapis.com/robot/v1/metadata/x509/plan-maker%40fynd-db.iam.gserviceaccount.com", "universe_domain": "googleapis.com" } # Submit button if st.button("Submit"): # Save user data user_data = { "Business_Head": selected_business_head, "Company_ID": company_id.strip() if company_id else None, "Company_Name": st.session_state.get('company_name', ''), "Currency": currency, "Bundle_by": bundle_by, "Plan_Name": plan_name, "Plan_Description": plan_description, "Product_lines": st.session_state.get('ordering_channel', ''), "Fulfilling_Location": st.session_state.get('fulfilling_location', ''), "Application_ID": st.session_state.get('application_id', ''), "Fee_Type": st.session_state.get('selected_fee_type', ''), "Variable_Type": selected_variable_type, "Chargeable_on": st.session_state.get('selected_chargeable_on', ''), "Fee_Nature": 0, "Commercial_value": user_input_3, "Fee_reversal": 0, "Reversal_%": 0, "Usage_limit": usage_3, "Threshold_value": 0, "Threshold_option": 0, "Expected_Billing": user_input_3, "Plan_Validity": selected_plan_validity, "Payment_Method": selected_payment_method, "Net_total_billing":int(final_billing), "slab_data": "" } # Save and display user data save_user_data(user_data) st.session_state.user_data = user_data st.session_state.df = pd.DataFrame([user_data]) st.write(st.session_state.df) # Display saved records # display_saved_records() # Insert into BigQuery if st.button("Export to BQ"): if project_id and dataset_id and table_id and service_account_info: user_data = st.session_state.user_data # Generate rule_id and plan_id rule_id = generate_rule_id(user_data) plan_id = generate_plan_id(user_data) # Update user_data with generated IDs user_data['rule_id'] = rule_id user_data['plan_id'] = plan_id st.session_state['created_at'] = datetime.now().isoformat() user_data = { "Business_Head": selected_business_head, "Company_ID": company_id.strip(), "Company_Name": st.session_state.get('company_name', ''), "Currency": currency, "Bundle_by": bundle_by, "Plan_Name": plan_name, "Plan_Description": plan_description, "Product_lines": st.session_state.get('ordering_channel', ''), "Fulfilling_Location": st.session_state.get('fulfilling_location', ''), "Application_ID": st.session_state.get('application_id', ''), "Fee_Type": st.session_state.get('selected_fee_type', ''), "Variable_Type": selected_variable_type, "Chargeable_on": st.session_state.get('selected_chargeable_on', ''), "Fee_Nature": 0, "Commercial_value": user_input_3, "Fee_reversal": 0, "Reversal_%": 0, "Usage_limit": usage_3, "Threshold_value": 0, "Threshold_option": 0, "Expected_Billing": user_input_3, "Plan_Validity": selected_plan_validity, "Payment_Method": selected_payment_method, "Net_total_billing":int(final_billing), "rule_id": rule_id, "plan_id": plan_id, "created_at": st.session_state.get('created_at',''), "slab_data": "" } # Save the user data locally and to session state save_user_data(user_data) # Display saved records display_saved_records() # Store created_at, rule_id, and plan_id in session state st.session_state['created_at'] = datetime.now().isoformat() st.session_state['rule_id'] = rule_id st.session_state['plan_id'] = plan_id insert_into_bq(user_data, project_id, dataset_id, table_id, service_account_info) else: st.error("One or more required variables are not defined.")