Update app.py

app.py

@@ -4,12 +4,14 @@ def main():
     st.title("集合操作アプリ")
 
     # メニューの選択
-    option = st.sidebar.selectbox("機能を選択してください", ["集合の操作", "全体集合の生成と補集合の計算"])
+    option = st.sidebar.selectbox("機能を選択してください", ["集合の操作", "全体集合の生成", "補集合の計算"])
 
     if option == "集合の操作":
         set_operations()
-    elif option == "全体集合の生成と補集合の計算":
-        generate_universe_and_complement()
+    elif option == "全体集合の生成":
+        generate_universe()
+    elif option == "補集合の計算":
+        complement_operation()
 
 def parse_set(input_str):
     if input_str:
@@ -63,34 +65,48 @@ def set_operations():
             symmetric_difference = set1.symmetric_difference(set2)
             st.write(symmetric_difference)
 
-def generate_universe_and_complement():
-    st.header("全体集合の生成と補集合の計算")
+def generate_universe():
+    st.header("全体集合の生成")
 
     with st.form(key='generate_universe_form'):
-        # 全体集合の生成用ユーザー入力
+        # ユーザー入力
         initial_value = st.number_input("初期値", value=0)
         step = st.number_input("ステップ", value=1)
         num_steps = st.number_input("回数", value=10, min_value=1)
         
         # フォーム送信ボタン
-        generate_button = st.form_submit_button(label='全体集合の生成')
+        submit_button = st.form_submit_button(label='生成')
 
-    if generate_button:
+    if submit_button:
         # 全体集合の生成
         universe_set = set(range(initial_value, initial_value + step * num_steps, step))
         
         st.subheader("生成された全体集合")
         st.write(universe_set)
+
+def complement_operation():
+    st.header("補集合の計算")
+
+    with st.form(key='complement_operation_form'):
+        # 全体集合の指定
+        universe = st.text_input("全体集合 (カンマ区切りで要素を入力してください)", "1,2,3,4,5,6,7,8,9,10")
+        # 対象集合の指定
+        target_set = st.text_input("対象集合 (カンマ区切りで要素を入力してください)", "1,3,5")
         
-        # 補集合の計算用ユーザー入力
-        subset_input = st.text_input("対象となる集合 (カンマ区切りで要素を入力してください)", "1,2,3")
-        
-        if subset_input:
-            subset = parse_set(subset_input)
-            if subset is not None:
-                complement = universe_set.difference(subset)
-                st.subheader("補集合の計算結果")
-                st.write(complement)
+        # フォーム送信ボタン
+        submit_button = st.form_submit_button(label='計算')
+
+    if submit_button:
+        # 入力された文字列を集合に変換
+        universe_set = parse_set(universe)
+        target_set = parse_set(target_set)
+
+        if universe_set is not None and target_set is not None:
+            # 補集合の計算
+            complement = universe_set.difference(target_set)
+            
+            st.subheader("補集合の計算結果")
+            st.write(complement)
 
 if __name__ == "__main__":
     main()