Update app.py

app.py

@@ -2,33 +2,11 @@ import streamlit as st
 import random
 from collections import Counter
 
-# Set up the Streamlit app
+# Set up 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()
 
@@ -41,50 +19,33 @@ 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])))
 
-# Opposite and adjacent faces (not used in Streamlit version)
-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à"]
-}
-
 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)]
+    results = []
+    shake_count = 0
+
+    while True:
+        shake_count += 1
+        results.append(random.choice(outcomes))
         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.")
 
-def reset_results():
-    result_label.text("Kết quả: ")
-    shake_info.text("Số lần lắc: 0")
+        if len(sorted_results) > 1:
+            rank_1, rank_2 = sorted_results[:2]
+            rank_1_pct = rank_1[1] / shake_count
+            rank_2_pct = rank_2[1] / shake_count
+
+            if (rank_1_pct - rank_2_pct) > 0.03:  # Chênh lệch > 3%
+                break
+
+    # Hiển thị kết quả
+    result_text = f"Số lần lắc: {shake_count}\n\nKết quả:\n"
+    for outcome, count in sorted_results:
+        result_text += f"{outcome}: {count} ({count/shake_count:.2%})\n"
+
+    result_label.text(result_text)
+    shake_info.text(f"Tổng số lần lắc: {shake_count}")
 
 # Buttons
 if st.button("Lắc"):
-    predict_result()
-
-if st.button("Reset"):
-    reset_results()
+    predict_result()