Update app.py

app.py

@@ -4,10 +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()
 
@@ -63,33 +65,54 @@ def set_operations():
             symmetric_difference = set1.symmetric_difference(set2)
             st.write(symmetric_difference)
 
+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)
+        
+        # フォーム送信ボタン
+        submit_button = st.form_submit_button(label='生成')
+
+    if submit_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'):
-        # 全体集合の指定
-        universe = st.text_input("全体集合 (カンマ区切りで要素を入力してください)", "1,2,3,4,5,6,7,8,9,10")
-        
         # ユーザー入力
-        initial_value = st.number_input("初期値", value=0)
-        step = st.number_input("ステップ", value=1)
-        num_steps = st.number_input("回数", value=5, min_value=1)
+        target_set_input = st.text_input("対象集合 (カンマ区切りで要素を入力してください)", "1,2,3")
         
         # フォーム送信ボタン
-        submit_button = st.form_submit_button(label='実行')
+        submit_button = st.form_submit_button(label='計算')
 
     if submit_button:
         # 入力された文字列を集合に変換
-        universe_set = parse_set(universe)
+        target_set = parse_set(target_set_input)
+
+        if target_set is not None:
+            # 全体集合をセッション状態から取得
+            universe_set = st.session_state['universe_set']
 
-        if universe_set is not None:
             # 補集合の計算
+            complement = universe_set.difference(target_set)
+            
             st.subheader("補集合の計算結果")
-            for i in range(num_steps):
-                current_range = range(initial_value, initial_value + step)
-                complement = universe_set.difference(current_range)
-                st.write(f"補集合({initial_value} - {initial_value + step - 1})", complement)
-                initial_value += step
+            st.write(complement)
 
 if __name__ == "__main__":
     main()