groshanr's picture
Upload folder using huggingface_hub
a8e9019 verified
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 += 0
## gradio
def nextQuestion(image, radio, scorebox):
global answer
global minPairGroup
global score
setRandAnswer()
setMinPairGroup()
image = gr.Image(f'./assets/{answer}.jpeg', label=answer, show_label= False)
radio = gr.Radio(choices=minPairGroup, label='Which word is pictured?')
scorebox = gr.Markdown(f"# Score: {score}")
output_column = gr.Column(visible=False)
input_column = gr.Column(visible=True)
return image, radio, scorebox, output_column, input_column
# submit button function. updates everything.
def onSubmit(radio, scorebox):
global answer
global score
updateScore(radio)
input_column = gr.Column(visible=False)
output_column = gr.Column(visible=True)
results = f"{answer} is correct. Good job!" if radio==answer else f"Incorrect. The correct answer was {answer}."
scorebox = gr.Markdown(f"# Score: {score}")
return scorebox, output_column, results, input_column
# 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)
with gr.Column(visible=True) as input_column:
radio = gr.Radio(choices=minPairGroup, label='Which word is pictured?', interactive=True)
submitButton = gr.Button("Submit")
with gr.Column(visible=False) as output_column:
results = gr.Textbox(label="Correct Answer")
nextButton = gr.Button("Next")
nextButton.click(nextQuestion, inputs= [image, radio, scorebox], outputs=[image, radio, scorebox, output_column, input_column])
submitButton.click(onSubmit, inputs= [radio, scorebox], outputs=[scorebox, output_column, results, input_column])
demo.launch()