Update app.py

app.py

@@ -1,9 +1,20 @@
 import streamlit as st
 
 def main():
-    st.title("集合の操作")
+    st.title("集合操作アプリ")
 
-    # 集合の入力
+    # メニューの選択
+    option = st.sidebar.selectbox("機能を選択してください", ["集合の操作", "補集合の計算"])
+
+    if option == "集合の操作":
+        set_operations()
+    elif option == "補集合の計算":
+        complement_operation()
+
+def set_operations():
+    st.header("集合の操作")
+
+    # ユーザー入力
     set1 = st.text_input("集合1 (カンマ区切りで要素を入力してください)")
     set2 = st.text_input("集合2 (カンマ区切りで要素を入力してください)")
     set3 = st.text_input("集合3 (カンマ区切りで要素を入力してください)")
@@ -14,7 +25,7 @@ def main():
     set3 = set(map(int, set3.split(',')))
 
     # 集合の操作
-    st.header("集合の操作結果")
+    st.subheader("集合の操作結果")
 
     st.subheader("和集合")
     union = set1.union(set2, set3)
@@ -32,5 +43,26 @@ def main():
     symmetric_difference = set1.symmetric_difference(set2, set3)
     st.write(symmetric_difference)
 
+def complement_operation():
+    st.header("補集合の計算")
+
+    # 全体集合の指定
+    universe = st.text_input("全体集合 (カンマ区切りで要素を入力してください)")
+
+    # 入力された文字列を集合に変換
+    universe_set = set(map(int, universe.split(',')))
+
+    # ユーザー入力
+    initial_value = st.number_input("初期値", value=0)
+    step = st.number_input("ステップ", value=1)
+    num_steps = st.number_input("回数", value=5)
+
+    # 補集合の計算
+    st.subheader("補集合の計算結果")
+    for i in range(num_steps):
+        complement = universe_set.difference(range(initial_value, initial_value + step))
+        st.write(f"補集合({initial_value} - {initial_value + step - 1})", complement)
+        initial_value += step
+
 if __name__ == "__main__":
     main()