Update app.py

app.py

@@ -1,12 +1,11 @@
-import streamlit as st
-from PIL import Image
-from components import GamePlay, Player, Dealer, Deck
 
-# PersistDataset -----
+
+import streamlit as st
 import os
 import csv
-
 import huggingface_hub
+from PIL import Image
+from components import GamePlay, Player, Dealer, Deck
 from huggingface_hub import Repository, hf_hub_download, upload_file
 from datetime import datetime
 
@@ -15,7 +14,8 @@ DATASET_REPO_ID = "awacke1/Carddata.csv"
 DATA_FILENAME = "Carddata.csv"
 DATA_FILE = os.path.join("data", DATA_FILENAME)
 HF_TOKEN = os.environ.get("HF_TOKEN")
-# overriding/appending to the gradio template
+number_of_decks = 6
+blackjack_multiplier = 1.5
 
 def generate_html() -> str:
     with open(DATA_FILE) as csvfile:
@@ -45,10 +45,6 @@ def store_message(name: str, message: str):
             )
         commit_url = repo.push_to_hub()
     return generate_html()
-    
-# Game settings
-number_of_decks = 6
-blackjack_multiplier = 1.5
 
 # Initialize player, dealer, deck and game play. Cache these variables
 @st.cache(allow_output_mutation=True, suppress_st_warning=True)
@@ -61,6 +57,42 @@ def start_game():
 
 game_deck, dealer, player, game_play = start_game()
 
+def display_pro_tip(player, dealer):
+    player_total = sum(card.rank for card in player.cards)
+    
+    if dealer.cards:
+        dealer_upcard = dealer.cards[0].rank
+    else:
+        dealer_upcard = 0
+
+    if player_total <= 11:
+        return "Pro Tip: With a total of 11 or less, it's generally advisable to hit."
+    elif player_total == 12 and dealer_upcard <= 3:
+        return "Pro Tip: With a total of 12 and the dealer showing 3 or less, consider hitting."
+    elif player_total >= 17:
+        return "Pro Tip: With a total of 17 or more, it's usually best to stand."
+    else:
+        return ""
+
+def display_betting_strategy(player, dealer):
+    player_total = sum(card.rank for card in player.cards)
+    
+    if dealer.cards:
+        dealer_upcard = dealer.cards[0].rank
+    else:
+        dealer_upcard = 0
+
+    if player_total <= 11:
+        return "Betting Strategy: Consider increasing your bet when your total is 11 or less."
+    elif player_total >= 17 and dealer_upcard <= 6:
+        return "Betting Strategy: Consider increasing your bet when you have a strong hand and the dealer has a weak upcard."
+    else:
+        return ""
+
+def calculate_hand_total(hand):
+    total = sum(card.rank for card in hand)
+    return total
+
 st.title('🃏Blackjack Simulator AI♠2️⃣1️⃣')
 
 if st.button('New hand?'):
@@ -68,39 +100,42 @@ if st.button('New hand?'):
 
 player_stats = st.empty()
 player_images = st.empty()
-player_hit_option = st.empty()
-player_double_down_option = st.empty()
-player_stand_option = st.empty()
+player_action = st.empty()
+done_button = st.empty()
 dealer_stats = st.empty()
 dealer_images = st.empty()
 result = st.empty()
+pro_tip = st.empty()
+betting_strategy = st.empty()
+
+while True:
+    game_play.update()
+    player_stats.write(player)
+    player_images.image([Image.open(card.image_location) for card in player.cards], width=100)
+    dealer_stats.write(dealer)
+    dealer_images.image([Image.open(card.image_location) for card in dealer.cards], width=100)
+    result.write(game_play)
 
+    pro_tip.write(display_pro_tip(player, dealer))
+    betting_strategy.write(display_betting_strategy(player, dealer))
 
-if 'Hit' in player.possible_actions:
-    if player_hit_option.button('Hit'):
-        player.player_hit(game_deck, game_play)
-        if 'Hit' not in player.possible_actions:
-            player_hit_option.empty()
-if 'Double Down' in player.possible_actions:
-    if player_double_down_option.button('Double Down'):
-        player.double_down(game_deck, game_play)
-        player_double_down_option.empty()
-        player_hit_option.empty()
-        player_stand_option.empty()
-if 'Stand' in player.possible_actions:
-    if player_stand_option.button('Stand'):
+    player_total = calculate_hand_total(player.cards)
+    dealer_total = calculate_hand_total(dealer.cards)
+
+    if player_total == 21 or dealer_total == 21 or not player.possible_actions:
+        break
+
+    if 'Hit' in player.possible_actions or 'Double Down' in player.possible_actions:
+        action = player_action.radio("Choose your action", ("👋 Hit", "⏬ Double Down"), key=f"action_{len(player.cards)}")
+
+        if action == "👋 Hit":
+            player.player_hit(game_deck, game_play)
+        elif action == "⏬ Double Down":
+            player.double_down(game_deck, game_play)
+
+        if done_button.button("✅ Done", key=f"done_{len(player.cards)}"):
+            player.stand(game_play)
+    else:
         player.stand(game_play)
-        player_hit_option.empty()
-        player_double_down_option.empty()
-        player_stand_option.empty()
-
-
-game_play.update()
-player_stats.write(player)
-player_images.image([Image.open(card.image_location)
-                     for card in player.cards], width=100)
-dealer_stats.write(dealer)
-dealer_images.image([Image.open(card.image_location)
-                     for card in dealer.cards], width=100)
-                     
-result.write(game_play)
+
+