Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 import random
 import streamlit as st
+import base64
 
 # Define the game rules
 NUM_ROUNDS = 26
@@ -53,6 +54,21 @@ def determine_winner(player_card, ai_card):
     else:
         return compare_cards(player_card, ai_card)
 
+def create_download_link(filename):
+    with open(filename, 'r') as f:
+        text = f.read()
+    b64 = base64.b64encode(text.encode()).decode()
+    href = f'<a href="data:file/txt;base64,{b64}" download="{filename}">Download {filename}</a>'
+    return href
+
+def start_game():
+    """Initializes the game state and starts the game."""
+    game_state = {'player_cards': [], 'ai_cards': [], 'player_score': 0, 'ai_score': 0, 'rounds_played': 0}
+    deck = shuffle_deck()
+    game_state['player_cards'] = deck[:26]
+    game_state['ai_cards'] = deck[26:]
+    return game_state
+
 # Define the game UI
 def game_ui():
     """Displays the game UI and updates the game state."""
@@ -71,23 +87,49 @@ def game_ui():
 
     st.write('**Dealer**')
     st.write('Cards: ', ' '.join([f"🂠" if len(ai_cards) == 1 else f"{card[0]}{card[1]}" for card in ai_cards]))
+    st.write
+    st.write('Score: ', game_state['ai_score'])
+    st.write('---')
+
+    if st.button('Play'):
+        if player_card is None:
+            st.write('Out of cards!')
+            return
+
+        winner = determine_winner(player_card, ai_card)
+
+        if winner == 'player':
+            st.write('Player wins!')
+            game_state['player_cards'].extend([player_card, ai_card])
+            game_state['player_score'] += 2
+        elif winner == 'ai':
+            st.write('Dealer wins!')
+            game_state['ai_cards'].extend([player_card, ai_card])
+            game_state['ai_score'] += 2
+        else:
+            st.write('Tie!')
+            game_state['player_cards'].append(player_card)
+            game_state['ai_cards'].append(ai_card)
+
+        game_state['rounds_played']
     st.write('Score: ', game_state['ai_score'])
     st.write('---')
 
-    if st.button('Draw'):
-        st.write('---')
-        st.write('You drew:', f"{player_card[0]}{player_card[1]}")
-        st.write('Dealer drew:', f"🂠" if len(ai_cards) == 1 else f"{ai_card[0]}{ai_card[1]}")
+    if st.button('Play'):
+        if player_card is None:
+            st.write('Out of cards!')
+            return
+
         winner = determine_winner(player_card, ai_card)
+
         if winner == 'player':
-            st.write('You won this round!')
+            st.write('Player wins!')
             game_state['player_cards'].extend([player_card, ai_card])
             game_state['player_score'] += 2
         elif winner == 'ai':
-            st.write('Dealer won this round!')
+            st.write('Dealer wins!')
             game_state['ai_cards'].extend([player_card, ai_card])
             game_state['ai_score'] += 2
-    
         else:
             st.write('Tie!')
             game_state['player_cards'].append(player_card)
@@ -104,14 +146,20 @@ def game_ui():
     st.sidebar.write('---')
     if st.sidebar.button('New Game'):
         # Reset game state
-        game_state['player_cards'] = []
-        game_state['ai_cards'] = []
-        game_state['player_score'] = 0
-        game_state['ai_score'] = 0
-        game_state['rounds_played'] = 0
-        deck = shuffle_deck()
-        game_state['player_cards'] = deck[:26]
-        game_state['ai_cards'] = deck[26:]
+        game_state = start_game()
+
+        # Save game state to file
+        with open('game_state.txt', 'w') as f:
+            f.write('player_cards,ai_cards,player_score,ai_score,rounds_played\n')
+            f.write(','.join([str(game_state[key]) for key in game_state.keys()]) + '\n')
+
+    if st.sidebar.button('Reset Game'):
+        # Reset game state
+        game_state = start_game()
+
+        # Truncate game_state.txt file by deleting it and reloading it
+        os.remove('game_state.txt')
+        open('game_state.txt', 'w').close()
 
         # Save game state to file
         with open('game_state.txt', 'w') as f:
@@ -143,44 +191,15 @@ def game_ui():
     with open('game_state.txt', 'r') as f:
         lines = f.readlines()
         headers = [header.strip() for header in lines[0].strip().split(',')]
-        data = [[cell.strip() for cell in line.strip().split(',')] for line in lines[1:]]
-        st.write(st.dataframe(data, columns=headers))
-
-# Play the game
-game_state = {'player_cards': [], 'ai_cards': [], 'player_score': 0, 'ai_score': 0, 'rounds_played': 0}
-while game_state['rounds_played'] < NUM_ROUNDS and len(game_state['player_cards']) + len(game_state['ai_cards']) == 52:
-    game_ui()
-
-    player_card = draw_card(game_state['player_cards'])
-    ai_card = draw_card(game_state['ai_cards'])
-    winner = determine_winner(player_card, ai_card)
-
-    if winner == 'player':
-        game_state['player_cards'].extend([player_card, ai_card])
-        game_state['player_score'] += 2
-    else:
-        game_state['player_cards'].append(player_card)
-        game_state['ai_cards'].append(ai_card)
-
-    game_state['rounds_played'] += 1
-
-    # Save game state to file
-    with open('game_state.txt', 'w') as f:
-        if not os.path.exists('game_state.txt'):
-            f.write('player_cards,ai_cards,player_score,ai_score,rounds_played\n')
-        f.write(','.join([str(game_state[key]) for key in game_state.keys()]) + '\n')
-
-import base64
-
-def create_download_link(filename):
-    with open(filename, 'r') as f:
-        text = f.read()
-    b64 = base64.b64encode(text.encode()).decode()
-    href = f'<a href="data:file/txt;base64,{b64}" download="{filename}">Download {filename}</a>'
-    return href
-
-if st.sidebar.button('Download Game State'):
-    st.sidebar.markdown(create_download_link('game_state.txt'), unsafe_allow_html=True)
-
-    
-
+        data =
+        [
+            [eval(cell) if cell.isdigit() else cell for cell in line.strip().split(',')]
+            for line in lines[1:]
+        ]
+    st.dataframe(data, columns=headers)
+
+    # Add download button for game history
+    if st.sidebar.button('Download Game History'):
+        st.sidebar.markdown(create_download_link('game_state.txt'), unsafe_allow_html=True)
+
+game_ui()