Update app.py

app.py

@@ -18,7 +18,7 @@ def plot_clusters(data, k):
     plt.title(f'K-means Clustering with k={k}')
     plt.xlabel('Feature 1')
     plt.ylabel('Feature 2')
-    plt_fig = plt  # 지역 변수명 변경
+    plt_fig = plt
     return plt_fig
 
 def main():
@@ -27,28 +27,28 @@ def main():
 
     # 데이터 생성 및 추가
     if 'data' not in st.session_state:
-        st.session_state.data = np.array([]).reshape(0, 2)  # 초기 데이터는 비어 있게 설정 -> 오류수정
+        st.session_state.data = np.array([]).reshape(0, 2)
 
     add_point = st.button("Add Random Data Point")
     if add_point:
-        new_point, _ = make_blobs(n_samples=1, centers=1, cluster_std=1.0)
+        new_point = np.array([[np.random.uniform(0, 100), np.random.uniform(0, 100)]])
         st.session_state.data = np.vstack([st.session_state.data, new_point])
 
-    x_coord = st.number_input("X Coordinate", value=0.0)
-    y_coord = st.number_input("Y Coordinate", value=0.0)
+    x_coord = st.number_input("X Coordinate", min_value=0.0, max_value=100.0, value=50.0)
+    y_coord = st.number_input("Y Coordinate", min_value=0.0, max_value=100.0, value=50.0)
     add_custom_point = st.button("Add Custom Data Point")
     if add_custom_point:
         st.session_state.data = np.vstack([st.session_state.data, [x_coord, y_coord]])
 
     st.write("Here is the data:")
     plt.scatter(st.session_state.data[:, 0], st.session_state.data[:, 1], s=50, cmap='viridis')
-    st.pyplot()  # plt.show() 대신 이를 사용
+    st.pyplot()
 
     k = st.slider("Select the number of clusters (k)", 1, 10, 4)
     st.write(f"You selected k={k}")
 
     plt_fig = plot_clusters(st.session_state.data, k)
-    st.pyplot(plt_fig)  # plt.show() 대신 이를 사용
+    st.pyplot(plt_fig)
 
 if __name__ == '__main__':
     main()