File size: 4,636 Bytes
3589a12 6c5b8f2 3589a12 6881175 3589a12 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 |
import streamlit as st
from PIL import Image
from components import GamePlay, Player, Dealer, Deck
# PersistDataset -----
import os
import csv
import gradio as gr
from gradio import inputs, outputs
import huggingface_hub
from huggingface_hub import Repository, hf_hub_download, upload_file
from datetime import datetime
DATASET_REPO_URL = "https://huggingface.co/datasets/awacke1/Carddata.csv"
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
SCRIPT = """
<script>
if (!window.hasBeenRun) {
window.hasBeenRun = true;
console.log("should only happen once");
document.querySelector("button.submit").click();
}
</script>
"""
with open(os.path.join(gr.networking.STATIC_TEMPLATE_LIB, "frontend", "index.html"), "a") as f:
f.write(SCRIPT)
try:
hf_hub_download(
repo_id=DATASET_REPO_ID,
filename=DATA_FILENAME,
cache_dir=DATA_DIRNAME,
force_filename=DATA_FILENAME
)
except:
print("file not found")
repo = Repository(
local_dir="data", clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN
)
def generate_html() -> str:
with open(DATA_FILE) as csvfile:
reader = csv.DictReader(csvfile)
rows = []
for row in reader:
rows.append(row)
rows.reverse()
if len(rows) == 0:
return "no messages yet"
else:
html = "<div class='chatbot'>"
for row in rows:
html += "<div>"
html += f"<span>{row['name']}</span>"
html += f"<span class='message'>{row['message']}</span>"
html += "</div>"
html += "</div>"
return html
def store_message(name: str, message: str):
if name and message:
with open(DATA_FILE, "a") as csvfile:
writer = csv.DictWriter(csvfile, fieldnames=["name", "message", "time"])
writer.writerow(
{"name": name, "message": message, "time": str(datetime.now())}
)
commit_url = repo.push_to_hub()
return generate_html()
iface = gr.Interface(
store_message,
[
inputs.Textbox(placeholder="Your name"),
inputs.Textbox(placeholder="Your message", lines=2),
],
"html",
css="""
.message {background-color:cornflowerblue;color:white; padding:4px;margin:4px;border-radius:4px; }
""",
title="Reading/writing to a HuggingFace dataset repo from Spaces",
description=f"This is a demo of how to do simple *shared data persistence* in a Gradio Space, backed by a dataset repo.",
article=f"The dataset repo is [{DATASET_REPO_URL}]({DATASET_REPO_URL})",
)
iface.launch()
# -------
# 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)
def start_game():
game_deck = Deck(number_of_decks)
dealer = Dealer()
player = Player()
game_play = GamePlay(player, dealer, game_deck, blackjack_multiplier)
return game_deck, dealer, player, game_play
game_deck, dealer, player, game_play = start_game()
st.title('🃏Blackjack Simulator AI♠2️⃣1️⃣')
if st.button('New hand?'):
game_play.deal_in()
player_stats = st.empty()
player_images = st.empty()
player_hit_option = st.empty()
player_double_down_option = st.empty()
player_stand_option = st.empty()
dealer_stats = st.empty()
dealer_images = st.empty()
result = st.empty()
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.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)
|