Upload app.py

app.py

@@ -171,6 +171,8 @@ else:
     st.success('Running in Production mode!', icon="✅")
 
 st.subheader("Tell us about your Databricks and Labelbox environments", divider='grey')
+cloud = "GCP"
+#cloud = st.selectbox('Which cloud environment does your Databricks Workspace run in?', ['AWS', 'Azure', 'GCP'], index=None)
 title = st.text_input('Enter Databricks Domain (e.g., <instance>.<cloud>.databricks.com)', '')
 databricks_api_key = st.text_input('Databricks API Key', type='password')
 labelbox_api_key = st.text_input('Labelbox API Key', type='password')
@@ -477,15 +479,21 @@ if new_dataset_name or selected_dataset_name:
                             # Schema Map
                             row_data_input = selected_row_data
                             global_key_input = selected_global_key
+
+                            # Create the initial dictionary
                             schema_map_dict = {'row_data': row_data_input}
                             if global_key_input:
                                 schema_map_dict['global_key'] = global_key_input
 
-                            # Convert the dict to a stringified JSON
-                            schema_map_str = json.dumps(schema_map_dict)
-                            
+                            # Swap keys and values
+                            reversed_schema_map_dict = {v: k for k, v in schema_map_dict.items()}
+
+                            # Convert the reversed dictionary to a stringified JSON
+                            reversed_schema_map_str = json.dumps(reversed_schema_map_dict)
+                                                        
                     
                             data = {
+                                "cloud": cloud,
                                 "mode": mode,
                                 "databricks_instance": databricks_instance,
                                 "databricks_api_key": databricks_api_key,
@@ -496,7 +504,7 @@ if new_dataset_name or selected_dataset_name:
                                 "frequency": frequency,
                                 "new_cluster": 0,
                                 "cluster_id": cluster_id,
-                                "schema_map": schema_map_str
+                                "schema_map": reversed_schema_map_str
                             }
                             
 
@@ -504,7 +512,7 @@ if new_dataset_name or selected_dataset_name:
                                 # Ensure all fields are filled out
                                 required_fields = [
                                     mode, databricks_instance, databricks_api_key, new_dataset, dataset_id,
-                                    table_path, labelbox_api_key, frequency, cluster_id, schema_map_str
+                                    table_path, labelbox_api_key, frequency, cluster_id, reversed_schema_map_str
                                 ]