groshanr's picture
Upload folder using huggingface_hub
48cb310 verified
raw
history blame
No virus
2.28 kB
import gspread
import random
import os
import unicodedata
import gradio as gr
import keyfiledict
from oauth2client.service_account import ServiceAccountCredentials
from dotenv import load_dotenv
load_dotenv()
import json
def getData():
gc = gspread.service_account_from_dict(keyfiledict.create_keyfile_dict())
spreadsheet = gc.open_by_key(os.environ['OPEN_BY_KEY'])
data = spreadsheet.worksheet('main').get_all_values()
data = [[unicodedata.normalize('NFKC', row[0]).strip()] + row[1:] for row in data][1:] # normalize chars and remove header row
return data
##### game #####
## globals
data = getData()
possibleAnswers = [unicodedata.normalize('NFKC', filename.rsplit('.', 1)[0]).strip() for filename in os.listdir('./assets')]
answer = ""
minPairGroup = []
score = 0
## setters
def setRandAnswer():
global answer
global possibleAnswers
answer = random.choice(possibleAnswers)
def setMinPairGroup():
global answer
global minPairGroup
global data
answerRowIndex = next((i for i, sublist in enumerate(data) if sublist and sublist[0] == answer), -1)
minPairGroup = [sublist[0] for sublist in data if sublist[1] == data[answerRowIndex][1]]
def updateScore(radio):
global score
if radio == answer:
score += 5
else:
score -= 1
## gradio
# submit button function. updates everything.
def onSubmit(image, radio, scorebox):
global answer
global minPairGroup
global score
updateScore(radio)
setRandAnswer()
setMinPairGroup()
image = gr.Image(f'./assets/{answer}.jpeg', label=answer, show_label= False)
radio = gr.Radio(choices=minPairGroup, label='Which word is pictured?', value=None)
scorebox = gr.Markdown(f"# Score: {score}")
return image, radio, scorebox
# the interface
with gr.Blocks() as demo:
setRandAnswer()
setMinPairGroup()
with gr.Column():
scorebox = gr.Markdown(f"# Score: {score}")
image = gr.Image(f'./assets/{answer}.jpeg', label=answer, show_label= False)
radio = gr.Radio(choices=minPairGroup, label='Which word is pictured?')
submitButton = gr.Button("Submit")
submitButton.click(onSubmit, inputs= [image, radio, scorebox], outputs=[image, radio, scorebox])
demo.launch()