Update app.py

app.py

@@ -840,9 +840,6 @@ try:
 
     elif main_option == "Segmentation Analysis":
         st.header("Segmentation Analysis")
-
-        cols = edit_strings(df.columns)
-        cols = sorted(list(set(cols)))
         
         st.sidebar.header("Selection of questions")
         single_list = st.sidebar.multiselect(
@@ -850,34 +847,37 @@ try:
             cols,
             default=[]
         )
-
+    
         multi_list = st.sidebar.multiselect(
             'Multi answer questions', 
             cols,
             default=[]
         )
-
+    
         score_list = st.sidebar.multiselect(
             'Score answer questions', 
             cols,
             default=[]
         )
-
+    
         matching_cols1 = []
         for i in multi_list:
             matching_cols1 += [col for col in df.columns if is_matching_pattern(col, i)]
-
+    
         df_clean = process_dataframe(df[single_list + matching_cols1])
         st.subheader("Selected Table")
         st.dataframe(df_clean)
 
-        linkage_method = st.sidebar.selectbox("Select the Linkage Method of Segmentation Analysis:", ["Hierarchical Clustering"])
-        
-        df_cluster = hierarchical_clustering_with_plotly(df_clean, linkage_method)
+        selected_method = st.sidebar.selectbox("Select the Linkage Method of Segmentation Analysis:", ['Hierarchical Clustering', 'K-means Clustering'])
+        if selected_method = 'Hierarchical Clustering':
+            linkage_method = st.sidebar.selectbox("Select the Linkage Method of Segmentation Analysis:", ['average', 'single', 'complete', 'weighted', 'centroid', 'median', 'ward'])
+            df_cluster = hierarchical_clustering_with_plotly(df_clean, linkage_method)
+        if selected_method = 'K-means Clustering':
+            k = int(st.text_input("Enter the desired number of clusters"))
+            df_clean = kmeans_clustering(df, k)
 
         st.subheader("Cluster Table")
         st.dataframe(df_clean)
-
     
     elif main_option == "Hypothesis test":
         st.header("Hypothesis Testing")