historyの修正

kasumi_ui.py

@@ -1,20 +1,20 @@
 import streamlit as st
-from simulator.kasumi import kasumi
+from simulator.kasumi import Kasumi_sim
 
-history = []
 
+    
 def kasumi_ui():
     # アプリのタイトル
     st.title("カスミ")
+    ks = Kasumi_sim()
 
     if st.button("カードを使う"):
-        num_energy = kasumi()
+        num_energy = ks.use_card()
         st.write(f"エネルギー数: {num_energy}")
-        history.append(num_energy)
-    
+        
     st.write("履歴")
-    st.write(history)
-    return history
+    st.write(ks.history)
+
 
 
 

simulator/kasumi.py

@@ -1,14 +1,24 @@
 
 import numpy as np
+import streamlit as st
+
+class Kasumi_sim:
+    def __init__(self):
+        if "ksm_history" not in st.session_state:
+            st.session_state.ksm_history = []
+        self.history = st.session_state.ksm_history
+
+    def use_card(self):
+        is_omote = True
+        num_energy = 0
+        while is_omote:
+            if np.random.rand() < 0.5:
+                is_omote = False
+            else:
+                num_energy += 1
+        st.session_state.ksm_history.append(num_energy)
+        self.history = st.session_state.ksm_history
+        return num_energy
+
 
-def kasumi():
-    is_omote = True
-    num_energy = 0
-    while is_omote:
-        if np.random.rand() < 0.5:
-            is_omote = False
-        else:
-            num_energy += 1
-    
-    return num_energy