Update app.py

app.py

@@ -1,90 +1,66 @@
 import streamlit as st
-import random
+import secrets  # Use secrets instead of random for better randomness
 from collections import Counter
 
-# Set up the Streamlit app
-st.set_page_config(page_title="Bầu Cua Tôm Cá", layout="centered")
-st.title("Bầu Cua Tôm Cá")
-
-# Set font and colors
-st.markdown(
-    """
-    <style>
-    .stButton>button {
-        font-size: 16px;
-        background-color: #FF0000;
-        color: #FFFFFF;
-    }
-    .stSelectbox, .stRadio {
-        font-size: 16px;
-    }
-    .stText {
-        font-size: 16px;
-        color: #000000;
-    }
-    </style>
-    """,
-    unsafe_allow_html=True
-)
-
-# UI elements
-shake_count = st.selectbox("Số lần lắc", ["1000", "50", "100", "500"])
-shake_info = st.empty()
-result_label = st.empty()
-
-dice_faces = ["Bầu", "Cua", "Tôm", "Cá", "Gà", "Nai"]
-default_faces = ["Cá", "Gà", "Nai"]
-selected_faces = []
+# Streamlit app setup
+st.set_page_config(page_title="Bầu Cua Tôm Cá - Xác Suất Xúc Xắc 3", page_icon="dice_icon.png", layout="centered")
 
-cols = st.columns(3)
-for i, col in enumerate(cols):
-    with col:
-        selected_faces.append(st.radio(f"Chọn mặt {i+1}", dice_faces, index=dice_faces.index(default_faces[i])))
+# Title and instructions
+st.title("Bầu Cua Tôm Cá")
+st.subheader("Chọn 3 mặt xúc xắc:")
 
-# Opposite and adjacent faces (not used in Streamlit version)
+# Dropdown menus for dice faces
+valid_faces = ["Bầu", "Cua", "Tôm", "Cá", "Gà", "Nai"]
 opposite_faces = {
-    "Bầu": "Cua",
-    "Cua": "Bầu",
-    "Nai": "Tôm",
-    "Tôm": "Nai",
-    "Gà": "Cá",
-    "Cá": "Gà"
-}
-
-adjacent_faces = {
-    "Bầu": ["Tôm", "Gà", "Nai", "Cá"],
-    "Cá": ["Bầu", "Nai", "Cua", "Tôm"],
-    "Gà": ["Bầu", "Tôm", "Cua", "Nai"],
-    "Tôm": ["Bầu", "Cá", "Cua", "Gà"],
-    "Nai": ["Bầu", "Gà", "Cua", "Cá"],
-    "Cua": ["Tôm", "Cá", "Nai", "Gà"]
+    "Gà": "Nai",
+    "Nai": "Gà",
+    "Cua": "Tôm",
+    "Tôm": "Cua",
+    "Bầu": "Cá",
+    "Cá": "Bầu"
 }
 
-def predict_result():
-    outcomes = ["Bầu", "Cua", "Tôm", "Cá", "Gà", "Nai"]
-    shake_count_int = int(shake_count)
-    
-    if len(selected_faces) == 3:
-        results = [random.choice(outcomes) for _ in range(shake_count_int)]
-        result_counts = Counter(results)
-        sorted_results = sorted(result_counts.items(), key=lambda x: x[1], reverse=True)
-        
-        result_text = "Kết quả:\n"
-        for outcome, count in sorted_results:
-            result_text += f"{outcome}: {count} ({count/shake_count_int:.2%})\n"
-        
-        result_label.text(result_text)
-        shake_info.text(f"Số lần lắc: {shake_count}")
-    else:
-        result_label.text("Vui lòng chọn 3 mặt đầu tiên.")
+face_selectors = [st.selectbox(f"Chọn mặt xúc xắc {i+1}", valid_faces) for i in range(3)]
 
-def reset_results():
-    result_label.text("Kết quả: ")
-    shake_info.text("Số lần lắc: 0")
+# Roll times selector
+roll_times = st.selectbox("Số lần lắc:", ["1000", "5", "50", "100", "500", "1"])
 
-# Buttons
-if st.button("Lắc"):
-    predict_result()
+# Roll dice button
+if st.button("Lắc Xúc Xắc"):
+    faces = face_selectors
+    roll_times = int(roll_times)
+    
+    # Simulate rolling the dice
+    roll_history = []
+    for _ in range(roll_times):
+        rolled_faces = []
+        for face in faces:
+            valid_choices = [f for f in valid_faces if f != opposite_faces[face]]
+            rolled_faces.append(secrets.choice(valid_choices))  # Use secrets.choice for better randomness
+        roll_history.extend(rolled_faces)
+    
+    # Count the frequency of each face
+    face_counts = Counter(roll_history)
+    
+    # Calculate the probability of each face (%)
+    total_rolls = len(roll_history)
+    probabilities = {face: (count / total_rolls) * 100 for face, count in face_counts.items()}
+    
+    # Sort probabilities
+    sorted_probabilities = sorted(probabilities.items(), key=lambda x: x[1], reverse=True)
+    
+    # Sort face counts by the number of rolls in descending order
+    sorted_face_counts = sorted(face_counts.items(), key=lambda x: x[1], reverse=True)
+    
+    # Display results
+    st.write(f"Kết quả lắc xúc xắc (tổng cộng {total_rolls // 3} lần):")
+    for face, count in sorted_face_counts:
+        st.write(f"{face}: {count} lần")
+    
+    st.write("\nXác suất tổng hợp:")
+    for rank, (face, prob) in enumerate(sorted_probabilities, 1):
+        st.write(f"{rank}. {face}: {prob:.2f}%")
 
+# Reset button
 if st.button("Reset"):
-    reset_results()
+    st.experimental_rerun()