Spaces:
Runtime error
Runtime error
File size: 2,274 Bytes
a2961dd 817fdfc a2961dd c64eaf4 a2961dd c64eaf4 a2961dd 7349ddb a2961dd 817fdfc a2961dd a35ad72 a2961dd 817fdfc a2961dd f081d7c a2961dd 8392718 a2961dd |
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 |
import os
import csv
import gradio as gr
import huggingface_hub
from huggingface_hub import Repository
from datetime import datetime
DATASET_REPO_URL = "https://huggingface.co/datasets/oskarastrom/cake-data"
DATA_FILENAME = "data.csv"
DATA_FILE = os.path.join("data", DATA_FILENAME)
HF_TOKEN = os.environ.get("HF_TOKEN")
print("is none?", HF_TOKEN is None)
print("hfh", huggingface_hub.__version__)
repo = Repository(
local_dir="data", clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN
)
repo.git_pull()
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:
print(rows)
html = "<div class='chatbot'>"
for row in rows:
print(row)
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):
global repo
if name and message:
repo.git_pull()
with open(DATA_FILE, "a") as csvfile:
writer = csv.DictWriter(csvfile, fieldnames=["name", "message", "time"])
writer.writerow(
{'winner': name, 'looser': message, 'time': str(datetime.now())}
)
commit_url = repo.push_to_hub()
print(commit_url)
repo.git_pull()
return generate_html()
iface = gr.Interface(
store_message,
[
gr.Textbox(placeholder="Your name"),
gr.Textbox(placeholder="Your message", lines=2),
],
gr.HTML(generate_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}) (open in new tab)",
)
iface.launch()
|