Upload app.py

app.py

@@ -6,6 +6,9 @@ import configparser  # For activating the config
 from PIL import Image, ImageFilter 
 import io
 import base64
+from google.cloud import bigquery
+from google.oauth2 import service_account
+
 
 global_user_data = []  # For appending user_data
 global_data_rows = [] # For appending data rows
@@ -556,29 +559,29 @@ if bundle_by == "Feature specific":
     if st.button("Submit", key ='submit_button'):
     # Save user data
         user_data = {
-        "Business Head": business_head,
-        "Company ID": company_id,
-        "Company Name": company_name,
+        "Business_Head": business_head,
+        "Company_ID": company_id,
+        "Company_Name": company_name,
         "Currency": currency,
-        "Bundle by": bundle_by,
-        "Plan Name": plan_name,
-        "Plan Description": plan_description,
-        "Product lines": ordering_channels,
-        "Fulfilling Location": fulfilling_location,
-        "Application ID": application_id,
-        "Fee Type": selected_fee_type,
-        "Variable Type": selected_variable_type,
-        "Chargeable on": selected_chargeable_on,
-        "Fee Nature": selected_fee_nature,
-        "Commercial value": user_input_3,
-        "Fee reversal": 0,
-        "Reversal %": 0,
-        "Usage limit": usage_3,
-        "Threshold value": Capping_or_Minimum_Guarantee_3,
-        "Threshold option": threshold,
-        "Expected Billing": expected_billing,
-        "Plan Validity": selected_plan_validity,
-        "Payment Method": selected_payment_method,
+        "Bundle_by": bundle_by,
+        "Plan_Name": plan_name,
+        "Plan_Description": plan_description,
+        "Product_lines": ordering_channels,
+        "Fulfilling_Location": fulfilling_location,
+        "Application_ID": application_id,
+        "Fee_Type": selected_fee_type,
+        "Variable_Type": selected_variable_type,
+        "Chargeable_on": selected_chargeable_on,
+        "Fee_Nature": selected_fee_nature,
+        "Commercial_value": user_input_3,
+        "Fee_reversal": 0,
+        "Reversal_%": 0,
+        "Usage_limit": usage_3,
+        "Threshold_value": Capping_or_Minimum_Guarantee_3,
+        "Threshold_option": threshold,
+        "Expected_Billing": expected_billing,
+        "Payment_Method": selected_payment_method,
+        "Plan_Validity": selected_plan_validity,
         }
 
         global_user_data.append((user_name, user_data))
@@ -596,6 +599,76 @@ if bundle_by == "Feature specific":
     # Display saved records
         display_saved_records()
 
+        if st.button("Export data to BQ"):
+            from google.oauth2 import service_account
+            from google.cloud import bigquery
+    
+            def insert_into_bq(user_data, project_id, dataset_id, table_id, service_account_info):
+                # Use from_service_account_info instead of from_service_account_file
+                credentials = service_account.Credentials.from_service_account_info(service_account_info)
+                
+                # Initialize a BigQuery client
+                client = bigquery.Client(credentials=credentials, project=project_id)
+                
+                # Specify the table reference
+                table_ref = f"{project_id}.{dataset_id}.{table_id}"
+                table = client.get_table(table_ref)
+                
+                # Insert form data into the table
+                rows_to_insert = [user_data]
+                errors = client.insert_rows_json(table, rows_to_insert)
+                
+                if errors:
+                    print(f"Errors occurred while inserting rows: {errors}")
+                else:
+                    print("Insert successful")
+            
+            project_id = "fynd-db"
+            dataset_id = "finance_dwh"
+            table_id = "plan_v2"
+            user_data = {
+                "Business_Head": business_head,
+                "Company_ID": company_id,
+                "Company_Name": company_name,
+                "Currency": currency,
+                "Bundle_by": bundle_by,
+                "Plan_Name": plan_name,
+                "Plan_Description": plan_description,
+                "Product_lines": ordering_channels,
+                "Fulfilling_Location": fulfilling_location,
+                "Application_ID": application_id,
+                "Fee_Type": selected_fee_type,
+                "Variable_Type": selected_variable_type,
+                "Chargeable_on": selected_chargeable_on,
+                "Fee_Nature": "Flat currency",
+                "Commercial_value": user_input_3,
+                "Fee_reversal": 0,
+                "Reversal_%": 0,
+                "Usage_limit": usage_3,
+                "Threshold_value": Capping_or_Minimum_Guarantee_3,
+                "Threshold_option": threshold,
+                "Expected_Billing": expected_billing,
+                "Payment_Method": selected_payment_method,
+                "Plan_Validity": selected_plan_validity
+            }
+            
+            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"
+            }
+    
+            insert_into_bq(user_data, project_id, dataset_id, table_id, service_account_info)
+
+
 
 
 elif bundle_by == "Single value":
@@ -809,6 +882,7 @@ elif bundle_by == "Single value":
         "Expected Billing": user_input_3,
         "Plan Validity": selected_plan_validity,
         "Payment Method": selected_payment_method,
+    
         }
 
         global_user_data.append((user_name, user_data))
@@ -826,4 +900,74 @@ elif bundle_by == "Single value":
     # Display saved records
         display_saved_records()
 
-    # Exporting the data to BQ
+
+    if st.button("Export data to BQ"):
+
+        from google.oauth2 import service_account
+        from google.cloud import bigquery
+    
+        def insert_into_bq(user_data, project_id, dataset_id, table_id, service_account_info):
+            # Use from_service_account_info instead of from_service_account_file
+            credentials = service_account.Credentials.from_service_account_info(service_account_info)
+            
+            # Initialize a BigQuery client
+            client = bigquery.Client(credentials=credentials, project=project_id)
+            
+            # Specify the table reference
+            table_ref = f"{project_id}.{dataset_id}.{table_id}"
+            table = client.get_table(table_ref)
+            
+            # Insert form data into the table
+            rows_to_insert = [user_data]
+            errors = client.insert_rows_json(table, rows_to_insert)
+            
+            if errors:
+                print(f"Errors occurred while inserting rows: {errors}")
+            else:
+                print("Insert successful")
+    
+        project_id = "fynd-db"
+        dataset_id = "finance_dwh"
+        table_id = "plan_v2"
+        user_data = {
+            "Business_Head": business_head,
+            "Company_ID": company_id,
+            "Company_Name": company_name,
+            "Currency": currency,
+            "Bundle_by": bundle_by,
+            "Plan_Name": plan_name,
+            "Plan_Description": plan_description,
+            "Product_lines": ordering_channels,
+            "Fulfilling_Location": fulfilling_location,
+            "Application_ID": application_id,
+            "Fee_Type": selected_fee_type,
+            "Variable_Type": selected_variable_type,
+            "Chargeable_on": selected_chargeable_on,
+            "Fee_Nature": "Flat currency",
+            "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,
+            "Payment_Method": selected_payment_method,
+            "Plan_Validity": selected_plan_validity
+            
+        }
+    
+        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"
+        }
+        
+        insert_into_bq(user_data, project_id, dataset_id, table_id, service_account_info)