Update app.py

app.py

@@ -4,14 +4,12 @@ def main():
     st.title("集合操作アプリ")
 
     # メニューの選択
-    option = st.sidebar.selectbox("機能を選択してください", ["集合の操作", "全体集合の生成", "補集合の計算"])
+    option = st.sidebar.selectbox("機能を選択してください", ["集合の操作", "全体集合の生成と補集合の計算"])
 
     if option == "集合の操作":
         set_operations()
-    elif option == "全体集合の生成":
-        generate_universe()
-    elif option == "補集合の計算":
-        complement_operation()
+    elif option == "全体集合の生成と補集合の計算":
+        generate_universe_and_complement()
 
 def parse_set(input_str):
     if input_str:
@@ -65,54 +63,34 @@ def set_operations():
             symmetric_difference = set1.symmetric_difference(set2)
             st.write(symmetric_difference)
 
-def generate_universe():
-    st.header("全体集合の生成")
+def generate_universe_and_complement():
+    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)
         
         # フォーム送信ボタン
-        submit_button = st.form_submit_button(label='生成')
+        generate_button = st.form_submit_button(label='全体集合の生成')
 
-    if submit_button:
+    if generate_button:
         # 全体集合の生成
         universe_set = set(range(initial_value, initial_value + step * num_steps, step))
         
         st.subheader("生成された全体集合")
         st.write(universe_set)
         
-        st.session_state['universe_set'] = universe_set  # 全体集合をセッション状態に保存
-
-def complement_operation():
-    st.header("補集合の計算")
-
-    if 'universe_set' not in st.session_state:
-        st.warning("全体集合が設定されていません。先に全体集合を生成してください。")
-        return
-
-    with st.form(key='complement_operation_form'):
-        # ユーザー入力
-        target_set_input = st.text_input("対象集合 (カンマ区切りで要素を入力してください)", "1,2,3")
+        # 補集合の計算用ユーザー入力
+        subset_input = st.text_input("対象となる集合 (カンマ区切りで要素を入力してください)", "1,2,3")
         
-        # フォーム送信ボタン
-        submit_button = st.form_submit_button(label='計算')
-
-    if submit_button:
-        # 入力された文字列を集合に変換
-        target_set = parse_set(target_set_input)
-
-        if target_set is not None:
-            # 全体集合をセッション状態から取得
-            universe_set = st.session_state['universe_set']
-
-            # 補集合の計算
-            complement = universe_set.difference(target_set)
-            
-            st.subheader("補集合の計算結果")
-            st.write(complement)
+        if subset_input:
+            subset = parse_set(subset_input)
+            if subset is not None:
+                complement = universe_set.difference(subset)
+                st.subheader("補集合の計算結果")
+                st.write(complement)
 
 if __name__ == "__main__":
     main()