Update app.py

app.py

@@ -25,9 +25,13 @@ def apply_kmeans(data, k):
     labels = kmeans.labels_
     return centroids, labels
 
+
 def main():
     st.title("K-means Clustering Simulator")
     
+    # Global variables declaration
+    global traffic_df, nature_df, population_df
+
     if st.button("Initialize Datasets"):
         traffic_data = np.random.uniform(0, 100, (num_samples * len(traffic_centers), 2))
         nature_data = np.random.uniform(0, 100, (num_samples * len(nature_centers), 2))
@@ -45,7 +49,7 @@ def main():
         "자연환경": nature_df,
         "인구밀집도": population_df
     }
-    
+   
     if datasets:
         combined_data = pd.concat([dataset_mapping[dataset_name] for dataset_name in datasets])